mirror of
https://github.com/migatu/conjurer.git
synced 2026-07-15 05:48:35 +00:00
Compare commits
20 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| fe81f5e8fc | |||
| 6d51412ea3 | |||
| 382cf5d501 | |||
| e121c6dc39 | |||
| 548496e0a6 | |||
| 99409f7c0b | |||
| fd1107cd1b | |||
| f08ce02e85 | |||
| 78346d8448 | |||
| c780308541 | |||
| cc0af0ce32 | |||
| 99e3dd7dca | |||
| 16ead67cf3 | |||
| 5ca73d6ac9 | |||
| f3e690f395 | |||
| 8293b3a69c | |||
| 0ac9006665 | |||
| d912d9bc1b | |||
| b08642dd88 | |||
| 8b1950823b |
@@ -46,7 +46,12 @@ from spotify_dl import youtube as youtube_download
|
|||||||
|
|
||||||
from communication_subroutine import comm_subroutine, IN_COMM_Q, OUT_COMM_Q
|
from communication_subroutine import comm_subroutine, IN_COMM_Q, OUT_COMM_Q
|
||||||
|
|
||||||
|
# *=========================== Classes definition
|
||||||
|
|
||||||
|
# The above code is defining a TypedDict called `Music_Config` in Python. This TypedDict has three
|
||||||
|
# keys: "ctx" which is an optional string, "queue" which is a list of strings, and "requester" which
|
||||||
|
# is a list of strings. This TypedDict is used to define the structure of a dictionary that should
|
||||||
|
# have these specific keys and value types.
|
||||||
Music_Config = TypedDict(
|
Music_Config = TypedDict(
|
||||||
"Music_Config",
|
"Music_Config",
|
||||||
{
|
{
|
||||||
@@ -58,6 +63,10 @@ Music_Config = TypedDict(
|
|||||||
|
|
||||||
|
|
||||||
class QueryControl:
|
class QueryControl:
|
||||||
|
"""
|
||||||
|
This class `QueryControl` is used to manage queries with information about the author, UUID,
|
||||||
|
content, logger, context, and replies.
|
||||||
|
"""
|
||||||
def __init__(self, query_author, query_uuid, query_content, uplogger, ctx) -> None:
|
def __init__(self, query_author, query_uuid, query_content, uplogger, ctx) -> None:
|
||||||
self.author = query_author
|
self.author = query_author
|
||||||
self.uuid = query_uuid
|
self.uuid = query_uuid
|
||||||
@@ -70,6 +79,95 @@ class QueryControl:
|
|||||||
)
|
)
|
||||||
self.replies = []
|
self.replies = []
|
||||||
|
|
||||||
|
class MusicFileList(object):
|
||||||
|
"""
|
||||||
|
The `MusicFileList` class manages a list of music files, with the ability to refresh the list from a
|
||||||
|
file service or local directory and update the list with new items.
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __init__(self, uplogger) -> None:
|
||||||
|
self.music_file_list = []
|
||||||
|
self.file_service_active = False
|
||||||
|
self.logger = uplogger
|
||||||
|
self.logger.info("Created Playlist organizer class")
|
||||||
|
|
||||||
|
def refresh_file_list(self):
|
||||||
|
"""
|
||||||
|
The `refresh_file_list` function attempts to connect to a file service to retrieve a list of music
|
||||||
|
files, and if unsuccessful, it populates the list by scanning a local directory for .mp3 files.
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
self.logger.info("Attempt to connect to file service")
|
||||||
|
response = requests.get(f"{FILE_SERVICE_ADDRESS}{GET_MP3}", timeout=360)
|
||||||
|
self.music_file_list = response.json()["music_file_list"]
|
||||||
|
self.file_service_active = True
|
||||||
|
except requests.exceptions.RequestException as e:
|
||||||
|
for mp3_item in Path.glob(Path(MUSIC_FOLDER), "**/*.mp3"):
|
||||||
|
temp_music_file = mp3_item.as_posix()
|
||||||
|
if platform == "win32":
|
||||||
|
temp_music_file = temp_music_file.replace("/", "\\")
|
||||||
|
self.music_file_list.append(temp_music_file)
|
||||||
|
self.file_service_active = False
|
||||||
|
self.logger.error(e.strerror)
|
||||||
|
self.logger.error("Service Unavailable")
|
||||||
|
|
||||||
|
def get_file_list(self):
|
||||||
|
"""
|
||||||
|
The `get_file_list` function returns the list of music files stored in the object.
|
||||||
|
:return: The `music_file_list` attribute is being returned.
|
||||||
|
"""
|
||||||
|
return self.music_file_list
|
||||||
|
|
||||||
|
def update_file_list(self, item):
|
||||||
|
"""
|
||||||
|
The `update_file_list` function appends an item to a music file list and sends a POST request with
|
||||||
|
the item data to a file service address.
|
||||||
|
|
||||||
|
:param item: The `item` parameter in the `update_file_list` method is the file that you want to add
|
||||||
|
to the `music_file_list` attribute of the class instance. It is then sent as a JSON payload in a
|
||||||
|
POST request to a file service address along with the endpoint `SEND_MP3`
|
||||||
|
"""
|
||||||
|
self.music_file_list.append(item)
|
||||||
|
post_data = {"item": str(item)}
|
||||||
|
requests.post(f"{FILE_SERVICE_ADDRESS}{SEND_MP3}", json=post_data, timeout=360)
|
||||||
|
|
||||||
|
class DoSearchView(discord.ui.View):
|
||||||
|
def __init__(self, _uuid, result_list, *args, **kwargs) -> None:
|
||||||
|
super().__init__(*args, **kwargs)
|
||||||
|
self.uuid = _uuid
|
||||||
|
self.result_list = result_list
|
||||||
|
|
||||||
|
@discord.ui.button(
|
||||||
|
label="Nastepna strona", style=discord.ButtonStyle.primary, row=0
|
||||||
|
)
|
||||||
|
async def next_callback(self, interaction, button):
|
||||||
|
# emoji = await ctx.guild.fetch_emoji(id)
|
||||||
|
# button.emoji=1102174413887111189
|
||||||
|
logger.info("Pressed")
|
||||||
|
await interaction.response.edit_message(
|
||||||
|
content="You clicked the button!"
|
||||||
|
) # Send a message when the button is clicked
|
||||||
|
|
||||||
|
@discord.ui.button(label="Wyszukaj wiecej", style=discord.ButtonStyle.danger, row=0)
|
||||||
|
async def reset_callback(self, interaction, button):
|
||||||
|
# emoji = await ctx.guild.fetch_emoji(id)
|
||||||
|
# button.emoji=1102174413887111189
|
||||||
|
logger.info("Pressed")
|
||||||
|
await interaction.response.edit_message(
|
||||||
|
content="You clicked the button!"
|
||||||
|
) # Send a message when the button is clicked
|
||||||
|
|
||||||
|
@discord.ui.button(
|
||||||
|
label="Poprzednia Strona", style=discord.ButtonStyle.primary, row=0
|
||||||
|
)
|
||||||
|
async def perv_callback(self, interaction, button):
|
||||||
|
# emoji = await ctx.guild.fetch_emoji(id)
|
||||||
|
# button.emoji=1102174413887111189
|
||||||
|
logger.info("Pressed")
|
||||||
|
await interaction.response.edit_message(
|
||||||
|
content="You clicked the button!"
|
||||||
|
) # Send a message when the button is clicked
|
||||||
|
|
||||||
|
|
||||||
# *=========================================== Predefines
|
# *=========================================== Predefines
|
||||||
formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
|
formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
|
||||||
@@ -93,6 +191,8 @@ FILE_SERVICE_ADDRESS = "http://192.168.1.15:5000"
|
|||||||
GET_MP3 = "/mp3"
|
GET_MP3 = "/mp3"
|
||||||
SEND_MP3 = "/update_mp3"
|
SEND_MP3 = "/update_mp3"
|
||||||
GET_PLAYLIST = "/get_music"
|
GET_PLAYLIST = "/get_music"
|
||||||
|
ADD_TO_PRIO_PLAYLIST = "/add_to_priority"
|
||||||
|
CLEAR_PRIO = "/clear_pr_pls"
|
||||||
LIBRARIAN_SERVICE_ADDRESS = "http://192.168.1.192:5001"
|
LIBRARIAN_SERVICE_ADDRESS = "http://192.168.1.192:5001"
|
||||||
SEND_QUERY = "/query"
|
SEND_QUERY = "/query"
|
||||||
TIME_BETWEEN_CALLS = 100000
|
TIME_BETWEEN_CALLS = 100000
|
||||||
@@ -193,60 +293,6 @@ logger.info("Playlist generation")
|
|||||||
|
|
||||||
# *=========================================== Load Data files
|
# *=========================================== Load Data files
|
||||||
|
|
||||||
|
|
||||||
class MusicFileList(object):
|
|
||||||
"""
|
|
||||||
The `MusicFileList` class manages a list of music files, with the ability to refresh the list from a
|
|
||||||
file service or local directory and update the list with new items.
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(self, uplogger) -> None:
|
|
||||||
self.music_file_list = []
|
|
||||||
self.file_service_active = False
|
|
||||||
self.logger = uplogger
|
|
||||||
self.logger.info("Created Playlist organizer class")
|
|
||||||
|
|
||||||
def refresh_file_list(self):
|
|
||||||
"""
|
|
||||||
The `refresh_file_list` function attempts to connect to a file service to retrieve a list of music
|
|
||||||
files, and if unsuccessful, it populates the list by scanning a local directory for .mp3 files.
|
|
||||||
"""
|
|
||||||
try:
|
|
||||||
self.logger.info("Attempt to connect to file service")
|
|
||||||
response = requests.get(f"{FILE_SERVICE_ADDRESS}{GET_MP3}", timeout=360)
|
|
||||||
self.music_file_list = response.json()["music_file_list"]
|
|
||||||
self.file_service_active = True
|
|
||||||
except requests.exceptions.RequestException as e:
|
|
||||||
for mp3_item in Path.glob(Path(MUSIC_FOLDER), "**/*.mp3"):
|
|
||||||
temp_music_file = mp3_item.as_posix()
|
|
||||||
if platform == "win32":
|
|
||||||
temp_music_file = temp_music_file.replace("/", "\\")
|
|
||||||
self.music_file_list.append(temp_music_file)
|
|
||||||
self.file_service_active = False
|
|
||||||
self.logger.error(e.strerror)
|
|
||||||
self.logger.error("Service Unavailable")
|
|
||||||
|
|
||||||
def get_file_list(self):
|
|
||||||
"""
|
|
||||||
The `get_file_list` function returns the list of music files stored in the object.
|
|
||||||
:return: The `music_file_list` attribute is being returned.
|
|
||||||
"""
|
|
||||||
return self.music_file_list
|
|
||||||
|
|
||||||
def update_file_list(self, item):
|
|
||||||
"""
|
|
||||||
The `update_file_list` function appends an item to a music file list and sends a POST request with
|
|
||||||
the item data to a file service address.
|
|
||||||
|
|
||||||
:param item: The `item` parameter in the `update_file_list` method is the file that you want to add
|
|
||||||
to the `music_file_list` attribute of the class instance. It is then sent as a JSON payload in a
|
|
||||||
POST request to a file service address along with the endpoint `SEND_MP3`
|
|
||||||
"""
|
|
||||||
self.music_file_list.append(item)
|
|
||||||
post_data = {"item": str(item)}
|
|
||||||
requests.post(f"{FILE_SERVICE_ADDRESS}{SEND_MP3}", json=post_data, timeout=360)
|
|
||||||
|
|
||||||
|
|
||||||
music_file_list = []
|
music_file_list = []
|
||||||
music_file_list = MusicFileList(logger)
|
music_file_list = MusicFileList(logger)
|
||||||
music_file_list.refresh_file_list()
|
music_file_list.refresh_file_list()
|
||||||
@@ -283,6 +329,8 @@ for key in word_reactions:
|
|||||||
word_reactions[key][2] = datetime.now()
|
word_reactions[key][2] = datetime.now()
|
||||||
logger.info("Done")
|
logger.info("Done")
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
# *=========================================== Create Client
|
# *=========================================== Create Client
|
||||||
logger.info("Creating discord bot")
|
logger.info("Creating discord bot")
|
||||||
client = commands.Bot(intents=intents, command_prefix="$")
|
client = commands.Bot(intents=intents, command_prefix="$")
|
||||||
@@ -342,8 +390,8 @@ async def on_message(message):
|
|||||||
if isinstance(message.channel, discord.DMChannel):
|
if isinstance(message.channel, discord.DMChannel):
|
||||||
channel = client.get_channel(1064888712565100614)
|
channel = client.get_channel(1064888712565100614)
|
||||||
await channel.send("Słyszałem ja żem że: " + message.content)
|
await channel.send("Słyszałem ja żem że: " + message.content)
|
||||||
else:
|
return
|
||||||
channel = message.channel
|
channel = message.channel
|
||||||
await client.process_commands(message)
|
await client.process_commands(message)
|
||||||
|
|
||||||
message.content = message.content.lower()
|
message.content = message.content.lower()
|
||||||
@@ -682,10 +730,40 @@ async def check_music():
|
|||||||
@tasks.loop(seconds=3)
|
@tasks.loop(seconds=3)
|
||||||
# trunk-ignore(pylint/R0915)
|
# trunk-ignore(pylint/R0915)
|
||||||
async def check_data_q():
|
async def check_data_q():
|
||||||
channel = client.get_channel(1062047367337095268)
|
|
||||||
try:
|
try:
|
||||||
data = IN_COMM_Q.get(block=False)
|
data = IN_COMM_Q.get(block=False)
|
||||||
await channel.send(f"O. A tak będzie wyglądało coś ciekawego w przyszłości: {data}")
|
entries = []
|
||||||
|
if data.stop:
|
||||||
|
searcher = data.author
|
||||||
|
ctx = data.ctx
|
||||||
|
query = data.content
|
||||||
|
l_p = 1
|
||||||
|
for doi in data.entries:
|
||||||
|
logger.info(doi)
|
||||||
|
desc = data.entries[doi]
|
||||||
|
title = desc['Title'][0]
|
||||||
|
entries.append(f"{l_p}. {title} pod linkiem https://www.sci-hub.se/{doi} i jest to {desc['type']}\n")
|
||||||
|
l_p+=1
|
||||||
|
message = '*Z podłogi wysuwa się winda na książki*'
|
||||||
|
message += f' Dzielny poszukiwaczu @{searcher} w odpowiedzi na twoje zapytanie {query}'
|
||||||
|
message += 'nasza biblioteka służy Ci odpowiedzią. Przy dźwięku fanfar winda się otwiera a w środku '
|
||||||
|
if len(entries)<1:
|
||||||
|
message += "niestety nie ma nic"
|
||||||
|
elif len(entries)>=1 and len(entries)<5:
|
||||||
|
message += " znajduje się coś:\n"
|
||||||
|
for item in entries:
|
||||||
|
message+=item
|
||||||
|
else:
|
||||||
|
message+= " znajduje się cholernie dużo:\n"
|
||||||
|
for item in entries:
|
||||||
|
message+=item
|
||||||
|
if len(message) > 1500:
|
||||||
|
await ctx.send(message)
|
||||||
|
message = ""
|
||||||
|
|
||||||
|
await ctx.send(message)
|
||||||
|
#Kept for sentimental reasons
|
||||||
|
#await ctx.send(f"O. A tak będzie wyglądało coś ciekawego w przyszłości: {data}")
|
||||||
except Empty:
|
except Empty:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
@@ -1047,8 +1125,10 @@ async def get_file(ctx, source, link):
|
|||||||
dir_path = None
|
dir_path = None
|
||||||
item_type, item_id = spotify.parse_spotify_url(link)
|
item_type, item_id = spotify.parse_spotify_url(link)
|
||||||
if item_id:
|
if item_id:
|
||||||
file_list = spotify.fetch_tracks(spotify_ctrl, item_type, link)
|
coro = asyncio.to_thread(spotify.fetch_tracks, args=(spotify_ctrl, item_type, link))
|
||||||
directory_name = spotify.get_item_name(spotify_ctrl, item_type, item_id)
|
file_list = await coro
|
||||||
|
coro = asyncio.to_thread(spotify.get_item_name, args=(spotify_ctrl, item_type, item_id))
|
||||||
|
directory_name = await coro
|
||||||
url_data = {"urls": []}
|
url_data = {"urls": []}
|
||||||
url_dict = {}
|
url_dict = {}
|
||||||
url_dict["save_path"] = Path(
|
url_dict["save_path"] = Path(
|
||||||
@@ -1275,6 +1355,7 @@ async def wyszukaj(ctx, how_many=0, slowa_kluczowe=None):
|
|||||||
json=jrequest,
|
json=jrequest,
|
||||||
timeout=360,
|
timeout=360,
|
||||||
)
|
)
|
||||||
|
|
||||||
return_data = await coroutine
|
return_data = await coroutine
|
||||||
if not return_data.status_code == 200:
|
if not return_data.status_code == 200:
|
||||||
await ctx.send("Wołaj szefa - coś się wyjebało")
|
await ctx.send("Wołaj szefa - coś się wyjebało")
|
||||||
@@ -1700,6 +1781,63 @@ async def zagraj_mi_kawalek(ctx):
|
|||||||
"Obawiam się że nie mogę nic znaleźć u siebie. Spróbuj komendy '$dej_co_ze_spotifaja' jeśli masz link"
|
"Obawiam się że nie mogę nic znaleźć u siebie. Spróbuj komendy '$dej_co_ze_spotifaja' jeśli masz link"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@client.hybrid_command(
|
||||||
|
name="dodaj_do_ulubionych",
|
||||||
|
description="Dodaje do listy ulubionych w radiu",
|
||||||
|
guild=discord.Object(id=664789470779932693),
|
||||||
|
)
|
||||||
|
async def dodaj_do_ulubionych(ctx):
|
||||||
|
"""
|
||||||
|
Generate a playlist in queue. First word in this command shall be int defining length of the playlist.
|
||||||
|
Rest of the line are search terms.
|
||||||
|
|
||||||
|
:param ctx: ctx stands for "context" and is a parameter commonly used in Discord.py commands. It
|
||||||
|
represents the context in which the command was invoked, including information such as the message,
|
||||||
|
the channel, the server, and the user who invoked the command. This parameter is required in all
|
||||||
|
Discord.py commands
|
||||||
|
"""
|
||||||
|
async with ctx.typing():
|
||||||
|
logger.info("Zaczynam szukać timestamp %s", datetime.now())
|
||||||
|
|
||||||
|
dlugosc_playlisty = ctx.message.content.split()[1]
|
||||||
|
word_list = ctx.message.content.split()
|
||||||
|
jrequest = {
|
||||||
|
"lista_slow": word_list,
|
||||||
|
"dlugosc_plejlisty": dlugosc_playlisty,
|
||||||
|
"UUID": str(uuid.uuid4()),
|
||||||
|
}
|
||||||
|
coroutine = asyncio.to_thread(
|
||||||
|
requests.post,
|
||||||
|
f"{FILE_SERVICE_ADDRESS}{ADD_TO_PRIO_PLAYLIST}",
|
||||||
|
json=jrequest,
|
||||||
|
timeout=360,
|
||||||
|
)
|
||||||
|
_ = await coroutine
|
||||||
|
|
||||||
|
@client.hybrid_command(
|
||||||
|
name="wyczysc_ulubione",
|
||||||
|
description="Czysci liste ulubionych w radiu",
|
||||||
|
guild=discord.Object(id=664789470779932693),
|
||||||
|
)
|
||||||
|
async def wyczysc_ulubione(ctx):
|
||||||
|
"""
|
||||||
|
Generate a playlist in queue. First word in this command shall be int defining length of the playlist.
|
||||||
|
Rest of the line are search terms.
|
||||||
|
|
||||||
|
:param ctx: ctx stands for "context" and is a parameter commonly used in Discord.py commands. It
|
||||||
|
represents the context in which the command was invoked, including information such as the message,
|
||||||
|
the channel, the server, and the user who invoked the command. This parameter is required in all
|
||||||
|
Discord.py commands
|
||||||
|
"""
|
||||||
|
async with ctx.typing():
|
||||||
|
coroutine = asyncio.to_thread(
|
||||||
|
requests.post,
|
||||||
|
f"{FILE_SERVICE_ADDRESS}{CLEAR_PRIO}",
|
||||||
|
json={},
|
||||||
|
timeout=360,
|
||||||
|
)
|
||||||
|
_ = await coroutine
|
||||||
|
|
||||||
|
|
||||||
@client.hybrid_command(
|
@client.hybrid_command(
|
||||||
name="zrob_mi_plejliste",
|
name="zrob_mi_plejliste",
|
||||||
@@ -2065,44 +2203,6 @@ async def parametry_muzyki_mego_ludu(
|
|||||||
logger.info(exce)
|
logger.info(exce)
|
||||||
|
|
||||||
|
|
||||||
class DoSearchView(discord.ui.View):
|
|
||||||
def __init__(self, uuid, result_list, *args, **kwargs) -> None:
|
|
||||||
super().__init__(*args, **kwargs)
|
|
||||||
self.uuid = uuid
|
|
||||||
self.result_list = result_list
|
|
||||||
|
|
||||||
@discord.ui.button(
|
|
||||||
label="Nastepna strona", style=discord.ButtonStyle.primary, row=0
|
|
||||||
)
|
|
||||||
async def next_callback(self, interaction, button):
|
|
||||||
# emoji = await ctx.guild.fetch_emoji(id)
|
|
||||||
# button.emoji=1102174413887111189
|
|
||||||
logger.info("Pressed")
|
|
||||||
await interaction.response.edit_message(
|
|
||||||
content="You clicked the button!"
|
|
||||||
) # Send a message when the button is clicked
|
|
||||||
|
|
||||||
@discord.ui.button(label="Wyszukaj wiecej", style=discord.ButtonStyle.danger, row=0)
|
|
||||||
async def reset_callback(self, interaction, button):
|
|
||||||
# emoji = await ctx.guild.fetch_emoji(id)
|
|
||||||
# button.emoji=1102174413887111189
|
|
||||||
logger.info("Pressed")
|
|
||||||
await interaction.response.edit_message(
|
|
||||||
content="You clicked the button!"
|
|
||||||
) # Send a message when the button is clicked
|
|
||||||
|
|
||||||
@discord.ui.button(
|
|
||||||
label="Poprzednia Strona", style=discord.ButtonStyle.primary, row=0
|
|
||||||
)
|
|
||||||
async def perv_callback(self, interaction, button):
|
|
||||||
# emoji = await ctx.guild.fetch_emoji(id)
|
|
||||||
# button.emoji=1102174413887111189
|
|
||||||
logger.info("Pressed")
|
|
||||||
await interaction.response.edit_message(
|
|
||||||
content="You clicked the button!"
|
|
||||||
) # Send a message when the button is clicked
|
|
||||||
|
|
||||||
|
|
||||||
@client.hybrid_command(
|
@client.hybrid_command(
|
||||||
name="wyszukaj_linki_do_dokumentow",
|
name="wyszukaj_linki_do_dokumentow",
|
||||||
description="Szuka linkow doi w bazie crossref i podaje linki do scihuba",
|
description="Szuka linkow doi w bazie crossref i podaje linki do scihuba",
|
||||||
@@ -2111,8 +2211,8 @@ class DoSearchView(discord.ui.View):
|
|||||||
async def wyszukaj_linki_do_dokumentow(ctx):
|
async def wyszukaj_linki_do_dokumentow(ctx):
|
||||||
query = ctx.message.content
|
query = ctx.message.content
|
||||||
query_uuid = uuid.uuid4()
|
query_uuid = uuid.uuid4()
|
||||||
# query_uuid = "test"
|
#TODO: TESTING ONLY!!
|
||||||
# 3 wyslij Librariana
|
#query_uuid = uuid.UUID('{12345678-1234-5678-1234-567812345678}')
|
||||||
json_query = {"UUID": str(query_uuid), "query": str(query)[30:], "page": 1}
|
json_query = {"UUID": str(query_uuid), "query": str(query)[30:], "page": 1}
|
||||||
coroutine = asyncio.to_thread(
|
coroutine = asyncio.to_thread(
|
||||||
requests.post,
|
requests.post,
|
||||||
@@ -2145,46 +2245,6 @@ async def wyszukaj_linki_do_dokumentow(ctx):
|
|||||||
await ctx.send(f"No dobra poszło. Zapamiętaj proszę {query_uuid} - to identyfikator twojego zapytania. Jesteś {queue_size} w kolejce."+
|
await ctx.send(f"No dobra poszło. Zapamiętaj proszę {query_uuid} - to identyfikator twojego zapytania. Jesteś {queue_size} w kolejce."+
|
||||||
" Zapytania obsługuje algorytm zasilany czterema chomikami zapierdalającymi w kołowrotku - więc wyniki najwcześniej za kilka godzi - ale mogą być też dni.")
|
" Zapytania obsługuje algorytm zasilany czterema chomikami zapierdalającymi w kołowrotku - więc wyniki najwcześniej za kilka godzi - ale mogą być też dni.")
|
||||||
|
|
||||||
'''
|
|
||||||
async def wyswietl_wyniki_wyszukiwania(ctx):
|
|
||||||
|
|
||||||
# 4 zwroc 4 wyniki od librariana
|
|
||||||
# 5 przyciski przy każdym wierszu pozwalają sprawdzić czy plik istnieje na scihubie - jeśli tak podaje link do sciagniecia
|
|
||||||
# 6 ostatni wiersz to dalej, wróć, skasuj, zaciagnij kolejne wyniki (jeśli jest ich więcej niż 1000)
|
|
||||||
result_view = DoSearchView(
|
|
||||||
uuid=str(query_uuid), result_list=query_response.json()["data"]
|
|
||||||
)
|
|
||||||
resulttab = []
|
|
||||||
resultlink = []
|
|
||||||
hit, fetched, total, _ = query_response.json()["data"][0]
|
|
||||||
answer = query_response.json()["data"][1]
|
|
||||||
|
|
||||||
for item in answer:
|
|
||||||
resulttab.append(str(item["title"]))
|
|
||||||
item_link = "https://sci-hub.se/" + str(item["DOI"])
|
|
||||||
resultlink.append(item_link)
|
|
||||||
logger.info(len(resulttab))
|
|
||||||
for i in range(0, min(4, len(resulttab))):
|
|
||||||
logger.info(i)
|
|
||||||
logger.info(resulttab[i])
|
|
||||||
logger.info(resultlink[i])
|
|
||||||
url = resultlink[i]
|
|
||||||
logger.info(url)
|
|
||||||
temp_label = str(resulttab[i])
|
|
||||||
temp_label = temp_label.strip("[']")
|
|
||||||
result_view.add_item(
|
|
||||||
discord.ui.Button(label=str(temp_label[0:79]), row=i + 1, url=url)
|
|
||||||
)
|
|
||||||
|
|
||||||
labelka = (
|
|
||||||
"Wyniki wyszukiwania dla: "
|
|
||||||
+ str(query)[30:]
|
|
||||||
+ f"\nWszystkich {total} sprawdzonych: {fetched} znalezionych: {hit} "
|
|
||||||
) # ile znaleziono artykułów w ogóle, ile zostało po refine
|
|
||||||
await ctx.send(
|
|
||||||
labelka, view=result_view
|
|
||||||
) # Send a message with our View class that contains the button
|
|
||||||
'''
|
|
||||||
|
|
||||||
@client.hybrid_command(
|
@client.hybrid_command(
|
||||||
name="krecimy_pornola",
|
name="krecimy_pornola",
|
||||||
@@ -2223,7 +2283,6 @@ async def krecimy_pornola(ctx):
|
|||||||
|
|
||||||
# *================================== Run
|
# *================================== Run
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
# TODO: wpakowac kondzia w wątki odpalic asynchronicznie rownolegle z flaskiem do sterowania nim
|
|
||||||
logger.info("Starting discord bot")
|
logger.info("Starting discord bot")
|
||||||
threads = []
|
threads = []
|
||||||
logger.info("Starting discord bot: Creating threads")
|
logger.info("Starting discord bot: Creating threads")
|
||||||
|
|||||||
+61
-15
@@ -1,6 +1,7 @@
|
|||||||
import logging
|
import logging
|
||||||
import threading
|
import threading
|
||||||
import json
|
import json
|
||||||
|
import time
|
||||||
from queue import Queue, Empty
|
from queue import Queue, Empty
|
||||||
from flask import Flask, jsonify, request
|
from flask import Flask, jsonify, request
|
||||||
from waitress import serve
|
from waitress import serve
|
||||||
@@ -9,28 +10,34 @@ HOST_ADDRESS = "192.168.1.191"
|
|||||||
PORT_ADDRESS = 5000
|
PORT_ADDRESS = 5000
|
||||||
OUT_COMM_Q = Queue()
|
OUT_COMM_Q = Queue()
|
||||||
IN_COMM_Q = Queue()
|
IN_COMM_Q = Queue()
|
||||||
internal_q = Queue()
|
awaiting_q = []
|
||||||
|
incoming_q = Queue()
|
||||||
app = Flask(__name__)
|
app = Flask(__name__)
|
||||||
|
|
||||||
@app.route("/conjurer", methods=["POST"])
|
@app.route("/conjurer", methods=["POST"])
|
||||||
def answer_external_command():
|
def answer_external_command():
|
||||||
|
"""
|
||||||
|
The function `answer_external_command` logs the request data, loads the data as JSON, logs the
|
||||||
|
record, and then puts the record into an incoming queue before returning a success message.
|
||||||
|
:return: The function `answer_external_command()` is returning a JSON response with the message
|
||||||
|
"SUCCESS".
|
||||||
|
"""
|
||||||
|
app.logger.info(request)
|
||||||
record = json.loads(request.data)
|
record = json.loads(request.data)
|
||||||
app.logger.info(record)
|
app.logger.info(record)
|
||||||
app.logger.info("DATA RECEIVED")
|
app.logger.info("DATA RECEIVED")
|
||||||
|
incoming_q.put(record)
|
||||||
|
|
||||||
while True:
|
|
||||||
try:
|
|
||||||
q_data = internal_q.get(block=False)
|
|
||||||
except Empty:
|
|
||||||
break
|
|
||||||
if q_data.uuid in record:
|
|
||||||
pass
|
|
||||||
|
|
||||||
return jsonify("SUCCESS")
|
return jsonify("SUCCESS")
|
||||||
|
|
||||||
@app.route("/conjurer", methods=["GET"])
|
@app.route("/conjurer", methods=["GET"])
|
||||||
def check_alive():
|
def check_alive():
|
||||||
|
"""
|
||||||
|
The function "check_alive" returns a JSON response with the message "ALIVE" when the "/conjurer"
|
||||||
|
route is accessed via a GET request.
|
||||||
|
:return: The code snippet is a Flask route that listens for GET requests to the "/conjurer"
|
||||||
|
endpoint. When a GET request is made to this endpoint, the function check_alive() is called, which
|
||||||
|
returns a JSON response with the message "ALIVE".
|
||||||
|
"""
|
||||||
return jsonify("ALIVE")
|
return jsonify("ALIVE")
|
||||||
|
|
||||||
def flask_debug(_logger):
|
def flask_debug(_logger):
|
||||||
@@ -52,19 +59,58 @@ def waitress_run(_logger):
|
|||||||
serve(app, host=HOST_ADDRESS, port=PORT_ADDRESS)
|
serve(app, host=HOST_ADDRESS, port=PORT_ADDRESS)
|
||||||
|
|
||||||
def scan_queue(_logger):
|
def scan_queue(_logger):
|
||||||
|
"""
|
||||||
|
The function `scan_queue` reads data from a queue, logs it, and appends it to another queue.
|
||||||
|
|
||||||
|
:param _logger: The `_logger` parameter is typically an instance of a logging object that is used to
|
||||||
|
record and store log messages. It is commonly used to track the flow of the program, record errors,
|
||||||
|
and provide information for debugging purposes. In this code snippet, the `_logger` object is used
|
||||||
|
to log the
|
||||||
|
"""
|
||||||
while True:
|
while True:
|
||||||
data = OUT_COMM_Q.get()
|
data = OUT_COMM_Q.get()
|
||||||
_logger.info(data)
|
_logger.info(data)
|
||||||
internal_q.put(data)
|
awaiting_q.append(data)
|
||||||
query_list.append(data)
|
|
||||||
|
def scan_incoming(_logger):
|
||||||
|
"""
|
||||||
|
The `scan_incoming` function continuously checks for incoming data, processes it, and logs when data
|
||||||
|
is found.
|
||||||
|
|
||||||
|
:param _logger: The `_logger` parameter in the `scan_incoming` function is a logger object that is
|
||||||
|
used to log messages or information during the execution of the function. It is typically used for
|
||||||
|
debugging, monitoring, or tracking the flow of the program. In this case, the `_logger` is being
|
||||||
|
used
|
||||||
|
"""
|
||||||
|
while True:
|
||||||
|
try:
|
||||||
|
answer = incoming_q.get(block=False)
|
||||||
|
_logger.info("DATA FOUND")
|
||||||
|
for record in awaiting_q:
|
||||||
|
if record.uuid in answer.keys():
|
||||||
|
record.stop = True
|
||||||
|
record.entries = answer[record.uuid]
|
||||||
|
IN_COMM_Q.put(record)
|
||||||
|
except Empty:
|
||||||
|
time.sleep(1)
|
||||||
|
|
||||||
|
|
||||||
def comm_subroutine(logger):
|
def comm_subroutine(logger):
|
||||||
|
"""
|
||||||
|
The `comm_subroutine` function starts multiple threads to run different tasks concurrently.
|
||||||
|
|
||||||
|
:param logger: The `logger` parameter in the `comm_subroutine` function is an instance of a logger
|
||||||
|
object that is used for logging messages at various levels (e.g., debug, info, warning, error). In
|
||||||
|
the provided code snippet, the logger is used to log messages at the "info" level
|
||||||
|
"""
|
||||||
#logger.setLevel(logging.DEBUG)
|
#logger.setLevel(logging.DEBUG)
|
||||||
logger.info("Started")
|
logger.info("Started")
|
||||||
threads = []
|
threads = []
|
||||||
threads.append(threading.Thread(target=flask_debug, args=(logger,)))
|
#threads.append(threading.Thread(target=flask_debug, args=(logger,)))
|
||||||
#threads.append(threading.Thread(target=waitress_run, args=(logger,)))
|
threads.append(threading.Thread(target=waitress_run, args=(logger,)))
|
||||||
threads.append(threading.Thread(target=scan_queue,args=(logger,)))
|
threads.append(threading.Thread(target=scan_queue,args=(logger,)))
|
||||||
|
threads.append(threading.Thread(target=scan_incoming,args=(logger,)))
|
||||||
|
|
||||||
|
|
||||||
for worker in threads:
|
for worker in threads:
|
||||||
worker.start()
|
worker.start()
|
||||||
|
|||||||
@@ -1,3 +1,4 @@
|
|||||||
|
from logging import handlers
|
||||||
import re
|
import re
|
||||||
import logging
|
import logging
|
||||||
import threading
|
import threading
|
||||||
@@ -12,9 +13,8 @@ from habanero import Crossref
|
|||||||
from waitress import serve
|
from waitress import serve
|
||||||
from queue import Queue
|
from queue import Queue
|
||||||
import search_bot
|
import search_bot
|
||||||
|
import scrape_bot
|
||||||
import requests
|
import requests
|
||||||
from typing import Callable
|
|
||||||
from uuid import UUID
|
|
||||||
NETRC_FILE = r"C:\Users\Activcom.pl\.netrc"
|
NETRC_FILE = r"C:\Users\Activcom.pl\.netrc"
|
||||||
|
|
||||||
HOST_ADDRESS = "192.168.1.192"
|
HOST_ADDRESS = "192.168.1.192"
|
||||||
@@ -22,7 +22,7 @@ PORT_ADDRESS = 5001
|
|||||||
MAIN_BOT_ADDRESS = "http://192.168.1.191:5000"
|
MAIN_BOT_ADDRESS = "http://192.168.1.191:5000"
|
||||||
SEND_RESULTS = "/conjurer"
|
SEND_RESULTS = "/conjurer"
|
||||||
BDSM_UUID_TEST = '96b7f85a-1142-4908-8986-62a2ea25a147'
|
BDSM_UUID_TEST = '96b7f85a-1142-4908-8986-62a2ea25a147'
|
||||||
|
MAX_CR_RESULTS = 150000
|
||||||
ENCODING = "utf-8"
|
ENCODING = "utf-8"
|
||||||
app = Flask(__name__)
|
app = Flask(__name__)
|
||||||
|
|
||||||
@@ -37,13 +37,14 @@ class Librarian(object):
|
|||||||
self.cr = Crossref(mailto=authTokens[0])
|
self.cr = Crossref(mailto=authTokens[0])
|
||||||
self.query = query
|
self.query = query
|
||||||
self.uuid = str(uuid)
|
self.uuid = str(uuid)
|
||||||
self.limit = 3000
|
self.limit = MAX_CR_RESULTS
|
||||||
self.fetched = 0
|
self.fetched = 0
|
||||||
self.hit = 0
|
self.hit = 0
|
||||||
self.total = 0
|
self.total = 0
|
||||||
self.app = app
|
self.app = app
|
||||||
self.logger = app.logger
|
|
||||||
self.live_results = []
|
self.live_results = []
|
||||||
|
self.final_result = []
|
||||||
|
self.not_in_db = []
|
||||||
self.search_result_from_cr = {}
|
self.search_result_from_cr = {}
|
||||||
self.done = False
|
self.done = False
|
||||||
|
|
||||||
@@ -53,17 +54,16 @@ class Librarian(object):
|
|||||||
self.search_result_from_cr.update(result)
|
self.search_result_from_cr.update(result)
|
||||||
self.total = result['message']['total-results']
|
self.total = result['message']['total-results']
|
||||||
self.fetched += len(result['message']['items'])
|
self.fetched += len(result['message']['items'])
|
||||||
self.logger.info (self.total)
|
self.app.logger.info (self.total)
|
||||||
self.logger.info(self.fetched)
|
self.app.logger.info(self.fetched)
|
||||||
while self.total > self.fetched and self.limit > self.fetched:
|
while self.total > self.fetched and self.limit > self.fetched:
|
||||||
result = self.cr.works(query = query, limit = 1000)
|
result = self.cr.works(query = query, limit = 1000)
|
||||||
self.search_result_from_cr.update(result)
|
self.search_result_from_cr.update(result)
|
||||||
self.total = result['message']['total-results']
|
self.total = result['message']['total-results']
|
||||||
self.fetched += len(result['message']['items'])
|
self.fetched += len(result['message']['items'])
|
||||||
self.logger.info (self.total)
|
self.app.logger.info (self.total)
|
||||||
self.logger.info(self.fetched)
|
self.app.logger.info(self.fetched)
|
||||||
|
time.sleep(0.1)
|
||||||
time.sleep(1)
|
|
||||||
self.app.logger.info("CROSSREF DONE")
|
self.app.logger.info("CROSSREF DONE")
|
||||||
#TODO: add to sql with UUID
|
#TODO: add to sql with UUID
|
||||||
#sql row - title, doi, author, uuid, last retrieved, is_available
|
#sql row - title, doi, author, uuid, last retrieved, is_available
|
||||||
@@ -131,8 +131,8 @@ class Librarian(object):
|
|||||||
temp = []
|
temp = []
|
||||||
refined_result = []
|
refined_result = []
|
||||||
for key in unrefined_reqult:
|
for key in unrefined_reqult:
|
||||||
self.logger.info("KEY:")
|
self.app.logger.info("KEY:")
|
||||||
self.logger.info(key)
|
self.app.logger.info(key)
|
||||||
|
|
||||||
for item in unrefined_reqult[self.uuid]["summary"]:
|
for item in unrefined_reqult[self.uuid]["summary"]:
|
||||||
if item["container"] == "NO" and item["title"]:
|
if item["container"] == "NO" and item["title"]:
|
||||||
@@ -142,20 +142,20 @@ class Librarian(object):
|
|||||||
temp.append([item["DOI"], item])
|
temp.append([item["DOI"], item])
|
||||||
|
|
||||||
refined_result = await self.check_if_exists(temp)
|
refined_result = await self.check_if_exists(temp)
|
||||||
temp = []
|
|
||||||
for item in refined_result:
|
for item in refined_result:
|
||||||
if item["exists"]:
|
if item["exists"]:
|
||||||
temp.append(item["data"])
|
self.final_result.append(item["data"])
|
||||||
refined_result = temp
|
else:
|
||||||
self.hit = len(refined_result)
|
self.not_in_db.append(item["data"])
|
||||||
return refined_result
|
self.hit = len(self.final_result)
|
||||||
|
return self.final_result
|
||||||
#TODO: DOłożyć sprawdzenie czy w rafinowanym pliku już nie mamy częściowego wyniku
|
#TODO: DOłożyć sprawdzenie czy w rafinowanym pliku już nie mamy częściowego wyniku
|
||||||
async def answer_query(self):
|
async def answer_query(self):
|
||||||
#TODO: DOłożyć sprawdzenie czy w rafinowanym pliku już nie mamy częściowego wyniku tutaj bo zanim w ogóle otworzymy nierafinowany
|
#TODO: DOłożyć sprawdzenie czy w rafinowanym pliku już nie mamy częściowego wyniku tutaj bo zanim w ogóle otworzymy nierafinowany
|
||||||
#Search sql for UUID
|
#Search sql for UUID
|
||||||
#SQL SELECT * FROM RESULTS WHERE UUID = uuid
|
#SQL SELECT * FROM RESULTS WHERE UUID = uuid
|
||||||
self.logger.info(f"Search started {self.uuid}")
|
self.app.logger.info(f"Search started {self.uuid}")
|
||||||
uuid_found = False
|
|
||||||
database = None
|
database = None
|
||||||
query = self.query
|
query = self.query
|
||||||
with open("cr_results.json", "r+", encoding="utf-8") as cr_file:
|
with open("cr_results.json", "r+", encoding="utf-8") as cr_file:
|
||||||
@@ -165,11 +165,19 @@ class Librarian(object):
|
|||||||
database = json.load(cr_file)
|
database = json.load(cr_file)
|
||||||
except JSONDecodeError:
|
except JSONDecodeError:
|
||||||
pass
|
pass
|
||||||
|
#TODO: Zwracać i final_result i not_in_db. Not in db_do pliku - i do ręcznego przeglądania potem.
|
||||||
if database:
|
if database:
|
||||||
for item in database.keys():
|
for item in database.keys():
|
||||||
if item in self.uuid:
|
if item in self.uuid:
|
||||||
self.result = await self.refine_query(database)
|
self.result = await self.refine_query(database)
|
||||||
|
temp = {}
|
||||||
|
for item in self.result:
|
||||||
|
temp[item['DOI']] = {"Title" : item['title'], "type": item['type']}
|
||||||
|
self.result = temp
|
||||||
|
temp2 = {}
|
||||||
|
for item in self.not_in_db:
|
||||||
|
temp2[item['DOI']] = {"Title" : item['title'], "type": item['type']}
|
||||||
|
self.not_in_db = temp2
|
||||||
return self.result
|
return self.result
|
||||||
answer = await self.search_crossref(query=query)
|
answer = await self.search_crossref(query=query)
|
||||||
database.update(answer)
|
database.update(answer)
|
||||||
@@ -177,7 +185,6 @@ class Librarian(object):
|
|||||||
cr_file.seek(0)
|
cr_file.seek(0)
|
||||||
json.dump(database,cr_file)
|
json.dump(database,cr_file)
|
||||||
self.result = await self.refine_query(answer)
|
self.result = await self.refine_query(answer)
|
||||||
return self.result
|
|
||||||
else:
|
else:
|
||||||
answer = await self.search_crossref(query=query)
|
answer = await self.search_crossref(query=query)
|
||||||
database = answer
|
database = answer
|
||||||
@@ -187,7 +194,15 @@ class Librarian(object):
|
|||||||
self.result = await self.refine_query(answer)
|
self.result = await self.refine_query(answer)
|
||||||
self.app.logger.info("Refined")
|
self.app.logger.info("Refined")
|
||||||
self.done = True
|
self.done = True
|
||||||
return self.result
|
temp = {}
|
||||||
|
for item in self.result:
|
||||||
|
temp[item['DOI']] = {"Title" : item['title'], "type": item['type']}
|
||||||
|
self.result = temp
|
||||||
|
temp2 = {}
|
||||||
|
for item in self.not_in_db:
|
||||||
|
temp2[item['DOI']] = {"Title" : item['title'], "type": item['type']}
|
||||||
|
self.not_in_db = temp2
|
||||||
|
return self.result
|
||||||
|
|
||||||
#============================= FLASK INTERNALS===============================
|
#============================= FLASK INTERNALS===============================
|
||||||
def flask_debug():
|
def flask_debug():
|
||||||
@@ -214,18 +229,34 @@ class BackgroundTaskSearch(threading.Thread):
|
|||||||
async def _run(self):
|
async def _run(self):
|
||||||
while True:
|
while True:
|
||||||
database = None
|
database = None
|
||||||
|
ndb_database = None
|
||||||
librarian = librarian_queue.get()
|
librarian = librarian_queue.get()
|
||||||
self.app.logger.info("STARTED")
|
self.app.logger.info("STARTED")
|
||||||
result = await librarian.answer_query()
|
result = await librarian.answer_query()
|
||||||
result = {librarian.uuid: result}
|
result = {librarian.uuid: result}
|
||||||
self.app.logger.info("Saving to file")
|
self.app.logger.info("Saving to file")
|
||||||
|
with open("not_in_db.json", "r+", encoding="utf-8") as ndb_file:
|
||||||
|
try:
|
||||||
|
ndb_database = json.load(ndb_file)
|
||||||
|
except JSONDecodeError:
|
||||||
|
pass
|
||||||
|
if ndb_database:
|
||||||
|
ndb_database.update(librarian.not_in_db)
|
||||||
|
else:
|
||||||
|
ndb_database = librarian.not_in_db
|
||||||
|
ndb_file.truncate(0)
|
||||||
|
ndb_file.seek(0)
|
||||||
|
json.dump(ndb_database, ndb_file)
|
||||||
|
|
||||||
with open("s_results.json", "r+", encoding="utf-8") as s_file:
|
with open("s_results.json", "r+", encoding="utf-8") as s_file:
|
||||||
try:
|
try:
|
||||||
database = json.load(s_file)
|
database = json.load(s_file)
|
||||||
except JSONDecodeError:
|
except JSONDecodeError:
|
||||||
pass
|
pass
|
||||||
if database:
|
if database:
|
||||||
database.extend(result)
|
self.app.logger.info(database)
|
||||||
|
self.app.logger.info(result)
|
||||||
|
database.update(result)
|
||||||
else:
|
else:
|
||||||
database = result
|
database = result
|
||||||
self.app.logger.info("DUMPING DATA")
|
self.app.logger.info("DUMPING DATA")
|
||||||
@@ -242,7 +273,8 @@ class BackgroundTaskSearch(threading.Thread):
|
|||||||
timeout=360,
|
timeout=360,
|
||||||
)
|
)
|
||||||
self.app.logger.info("SENT")
|
self.app.logger.info("SENT")
|
||||||
_ = await coroutine
|
result = await coroutine
|
||||||
|
self.app.logger.info(result.status_code)
|
||||||
self.app.logger.info("SEND CONFIRMED")
|
self.app.logger.info("SEND CONFIRMED")
|
||||||
time.sleep(1)
|
time.sleep(1)
|
||||||
|
|
||||||
@@ -254,9 +286,6 @@ async def query_database():
|
|||||||
app.logger.info(record["query"])
|
app.logger.info(record["query"])
|
||||||
app.logger.info(record["UUID"])
|
app.logger.info(record["UUID"])
|
||||||
uuid = record["UUID"]
|
uuid = record["UUID"]
|
||||||
#TODO: TESTING ONLY======================
|
|
||||||
uuid = UUID('{12345678-1234-5678-1234-567812345678}')
|
|
||||||
|
|
||||||
cl = Librarian(app, record["query"], uuid)
|
cl = Librarian(app, record["query"], uuid)
|
||||||
librarian_queue.put(cl)
|
librarian_queue.put(cl)
|
||||||
librarian_list.append(cl)
|
librarian_list.append(cl)
|
||||||
@@ -284,22 +313,31 @@ async def get_partial():
|
|||||||
|
|
||||||
#=======================================MAIN===================================================
|
#=======================================MAIN===================================================
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
logger = logging.getLogger('conjurer_musician')
|
app.logger.setLevel(logging.DEBUG)
|
||||||
logger.setLevel(logging.DEBUG)
|
h1 = handlers.RotatingFileHandler(
|
||||||
|
filename="E:\\logs\\librarian.log",
|
||||||
|
encoding="utf-8",
|
||||||
|
mode="a",
|
||||||
|
maxBytes=6 * 1024 * 1024,
|
||||||
|
backupCount=6,
|
||||||
|
)
|
||||||
|
|
||||||
|
app.logger.addHandler(h1)
|
||||||
threads = []
|
threads = []
|
||||||
#threads.append(threading.Thread(target=waitress_run))
|
threads.append(threading.Thread(target=waitress_run))
|
||||||
threads.append(threading.Thread(target=flask_debug))
|
#threads.append(threading.Thread(target=flask_debug))
|
||||||
bgtask = BackgroundTaskSearch()
|
bgtask = BackgroundTaskSearch()
|
||||||
bgtask.app = app
|
bgtask.app = app
|
||||||
threads.append(bgtask)
|
threads.append(bgtask)
|
||||||
|
threads.append(threading.Thread(target = scrape_bot.scraper, args=(app.logger,)))
|
||||||
i = 0
|
i = 0
|
||||||
for worker in threads:
|
for worker in threads:
|
||||||
try:
|
try:
|
||||||
logger.info(f"App number: {i}")
|
app.logger.info(f"App number: {i}")
|
||||||
i+=1
|
i+=1
|
||||||
worker.start()
|
worker.start()
|
||||||
except RuntimeError as e:
|
except RuntimeError as e:
|
||||||
logger.error("Exploded")
|
app.logger.error("Exploded")
|
||||||
print(str(e))
|
print(str(e))
|
||||||
for worker in threads:
|
for worker in threads:
|
||||||
worker.join()
|
worker.join()
|
||||||
|
|||||||
@@ -0,0 +1 @@
|
|||||||
|
c:\conjurer\env\Scripts\activate.bat && c:\conjurer\conjurer_librarian\conjurer_librarian.py
|
||||||
@@ -0,0 +1,93 @@
|
|||||||
|
import json
|
||||||
|
from json.decoder import JSONDecodeError
|
||||||
|
from urllib.request import urlopen
|
||||||
|
from requests import Timeout,ConnectTimeout, ConnectionError
|
||||||
|
from queue import Queue
|
||||||
|
import re
|
||||||
|
import time
|
||||||
|
import random
|
||||||
|
import logging
|
||||||
|
from threading import Thread
|
||||||
|
SCR_DATABASE_PATH = r'C:\\Database\\chunks\\'
|
||||||
|
SCR_FILENAME = "40_chunk.txt"
|
||||||
|
SCR_ENCODING = "utf-8"
|
||||||
|
|
||||||
|
WORK_Q = Queue()
|
||||||
|
random.seed()
|
||||||
|
def load_ndb_to_q(logger):
|
||||||
|
logger.info("Loader started")
|
||||||
|
while True:
|
||||||
|
with open("not_in_db.json", "r+", encoding="utf-8") as ndb_file:
|
||||||
|
try:
|
||||||
|
ndb_database = json.load(ndb_file)
|
||||||
|
for item in ndb_database.keys():
|
||||||
|
logger.info(item)
|
||||||
|
url = f"https://sci-hub.se/{item}"
|
||||||
|
WORK_Q.put([item, url, False])
|
||||||
|
ndb_file.truncate(0)
|
||||||
|
except JSONDecodeError:
|
||||||
|
time.sleep(60*60*2)
|
||||||
|
pass
|
||||||
|
time.sleep(180)
|
||||||
|
|
||||||
|
def check_if_exists_brute_force(logger):
|
||||||
|
while True:
|
||||||
|
logger.info("Scraper tick")
|
||||||
|
item = WORK_Q.get()
|
||||||
|
page_url =item[1]
|
||||||
|
logger.error("REFINE: Brute force search!")
|
||||||
|
if not page_url.startswith(("http:", "https:")):
|
||||||
|
raise ValueError("URL must start with 'http:' or 'https:'")
|
||||||
|
blocked = True
|
||||||
|
try:
|
||||||
|
# trunk-ignore(bandit/B310)
|
||||||
|
with urlopen(page_url) as response:
|
||||||
|
data = response.read()
|
||||||
|
text = data.decode("utf-8")
|
||||||
|
if item[2]:
|
||||||
|
logger.info("Already found")
|
||||||
|
time.sleep(180)
|
||||||
|
blocked = False
|
||||||
|
else:
|
||||||
|
for line in text.splitlines():
|
||||||
|
if re.match(r".*Unfortunately, Sci-Hub doesn't have the requested document.*", line):
|
||||||
|
blocked = False
|
||||||
|
logger.info("Not found")
|
||||||
|
item[2] = False
|
||||||
|
if m := re.match(r".*<embed type=\"application.pdf\"\s*src=\"(.*\.pdf)",line):
|
||||||
|
blocked = False
|
||||||
|
direct_download_link = "https:" + str(m.group(1))
|
||||||
|
logger.info (direct_download_link)
|
||||||
|
with open(SCR_DATABASE_PATH + SCR_FILENAME, "a", encoding=SCR_ENCODING) as operated_file:
|
||||||
|
operated_file.write("\n")
|
||||||
|
operated_file.write(item[0])
|
||||||
|
item[2] = True
|
||||||
|
except (Timeout,ConnectTimeout, ConnectionRefusedError, ConnectionError):
|
||||||
|
pass
|
||||||
|
if blocked:
|
||||||
|
logger.info(item)
|
||||||
|
logger.info(text)
|
||||||
|
logger.error("Got blocked. Fuck.")
|
||||||
|
time.sleep(60*60*72)
|
||||||
|
# trunk-ignore(bandit/B311)
|
||||||
|
rand = random.randint(1,100)
|
||||||
|
logger.info(f"Sleeping for {3*rand} minutes")
|
||||||
|
time.sleep(180*rand)
|
||||||
|
|
||||||
|
def scraper(logger = None):
|
||||||
|
if not logger:
|
||||||
|
logger = logging.getLogger()
|
||||||
|
logger.setLevel("DEBUG")
|
||||||
|
h1 = logging.StreamHandler()
|
||||||
|
logger.addHandler(h1)
|
||||||
|
logger.info("No logger set. Test run")
|
||||||
|
|
||||||
|
loader = Thread(target = load_ndb_to_q, args = (logger,))
|
||||||
|
scraper = Thread(target = check_if_exists_brute_force, args = (logger,))
|
||||||
|
loader.start()
|
||||||
|
scraper.start()
|
||||||
|
loader.join()
|
||||||
|
scraper.join()
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
scraper()
|
||||||
@@ -1,25 +1,30 @@
|
|||||||
|
#TODO: Wpiemdolić to wszystko w klasę z loggerem przysłanym z góry
|
||||||
|
|
||||||
from queue import Queue, Empty
|
from queue import Queue, Empty
|
||||||
from threading import Thread
|
from threading import Thread
|
||||||
q = Queue()
|
q = Queue()
|
||||||
|
|
||||||
#TODO: TO w klasie jako parametry wybierane czy robimy pelen run czy testowy
|
#TODO: DATA FOR TEST ONLY
|
||||||
#MAXTHREADS = 3
|
#MAXTHREADS = 5
|
||||||
MAXTHREADS = 1
|
|
||||||
|
|
||||||
#DATABASE_PATH = r'C:\\Database\\chunks_1\\'
|
#DATABASE_PATH = r'C:\\Database\\chunks_1\\'
|
||||||
|
|
||||||
|
#TODO: DEPLOYMENT DATA
|
||||||
|
MAXTHREADS = 41
|
||||||
DATABASE_PATH = r'C:\\Database\\chunks\\'
|
DATABASE_PATH = r'C:\\Database\\chunks\\'
|
||||||
|
|
||||||
ENCODING = "utf-8"
|
ENCODING = "utf-8"
|
||||||
CHUNk = "_chunk.txt"
|
CHUNk = "_chunk.txt"
|
||||||
_sentinel = object()
|
_sentinel = object()
|
||||||
result_list = []
|
result_list = []
|
||||||
|
WORK_Q_SIZE = 15500000
|
||||||
|
|
||||||
#TODO: Wpiemdolić to wszystko w klasę z loggerem przysłanym z góry
|
def producer(out_q, control_q, filename, logger):
|
||||||
|
|
||||||
def producer(out_q, control_q, filename):
|
|
||||||
with open(DATABASE_PATH + filename, "r+", encoding=ENCODING) as operated_file:
|
with open(DATABASE_PATH + filename, "r+", encoding=ENCODING) as operated_file:
|
||||||
lines = operated_file.readlines()
|
logger.info(f"Worker {filename} ")
|
||||||
for line in lines:
|
while True:
|
||||||
|
line = operated_file.readline()
|
||||||
|
if not line:
|
||||||
|
break
|
||||||
out_q.put(line)
|
out_q.put(line)
|
||||||
try:
|
try:
|
||||||
check = control_q.get(block=False)
|
check = control_q.get(block=False)
|
||||||
@@ -27,11 +32,11 @@ def producer(out_q, control_q, filename):
|
|||||||
check = False
|
check = False
|
||||||
pass
|
pass
|
||||||
if check is _sentinel:
|
if check is _sentinel:
|
||||||
print("TERM signal received")
|
logger.info("TERM signal received")
|
||||||
control_q.put(check)
|
control_q.put(check)
|
||||||
break
|
break
|
||||||
print(f"Worker finished: {filename}")
|
logger.info(f"Worker finished: {filename}")
|
||||||
out_q.put(_sentinel)
|
out_q.put(_sentinel)
|
||||||
|
|
||||||
#IMPORTANT!!! ONLY ONE CONSUMER THREAD AS WE ARE NOT PUTTING SENTINELS BACK
|
#IMPORTANT!!! ONLY ONE CONSUMER THREAD AS WE ARE NOT PUTTING SENTINELS BACK
|
||||||
#TODO: Change doi to tuple list and only put term when all doi are found => (doi, True)
|
#TODO: Change doi to tuple list and only put term when all doi are found => (doi, True)
|
||||||
@@ -40,10 +45,6 @@ def consumer(in_q, control_q, doi, live_results, logger):
|
|||||||
for item in doi:
|
for item in doi:
|
||||||
result_list.append({"DOI" : item[0], "exists" : False, "data" : item[1]})
|
result_list.append({"DOI" : item[0], "exists" : False, "data" : item[1]})
|
||||||
while True:
|
while True:
|
||||||
#if in_q.empty():
|
|
||||||
# print("Queue empty!")
|
|
||||||
#if in_q.full():
|
|
||||||
# print("Queue full!")
|
|
||||||
done_check = True
|
done_check = True
|
||||||
data = in_q.get()
|
data = in_q.get()
|
||||||
if data is _sentinel:
|
if data is _sentinel:
|
||||||
@@ -51,7 +52,7 @@ def consumer(in_q, control_q, doi, live_results, logger):
|
|||||||
sentinels += 1
|
sentinels += 1
|
||||||
else:
|
else:
|
||||||
for item in result_list:
|
for item in result_list:
|
||||||
if item["DOI"] in data:
|
if item["DOI"] in data and not item["exists"]:
|
||||||
logger.info(data)
|
logger.info(data)
|
||||||
logger.info("HIT")
|
logger.info("HIT")
|
||||||
item["exists"] = True
|
item["exists"] = True
|
||||||
@@ -66,25 +67,24 @@ def consumer(in_q, control_q, doi, live_results, logger):
|
|||||||
|
|
||||||
def search_for_doi(doi, live_results, logger):
|
def search_for_doi(doi, live_results, logger):
|
||||||
threads = []
|
threads = []
|
||||||
work_q = Queue(maxsize=200000)
|
work_q = Queue(maxsize=WORK_Q_SIZE)
|
||||||
control_q = Queue()
|
control_q = Queue()
|
||||||
t_cons = Thread(target = consumer, args = (work_q, control_q, doi, live_results, logger))
|
t_cons = Thread(target = consumer, args = (work_q, control_q, doi, live_results, logger))
|
||||||
logger.info("Consumer thread created")
|
logger.info("Consumer thread created")
|
||||||
threads.append(t_cons)
|
threads.append(t_cons)
|
||||||
for i in range(0,MAXTHREADS):
|
for i in range(0,MAXTHREADS):
|
||||||
|
#TODO: TEST DATA
|
||||||
#filename = "test" + str(i) + CHUNk
|
#filename = "test" + str(i) + CHUNk
|
||||||
filename = "40" + CHUNk
|
#TODO: DEPLOYMENT DATA
|
||||||
#filename = str(i) + CHUNk
|
filename = str(i) + CHUNk
|
||||||
logger.info(f"Creating worker thread no: {i}")
|
logger.info(f"Creating worker thread no: {i}")
|
||||||
threads.append(Thread(target = producer, args = (work_q, control_q, filename)))
|
threads.append(Thread(target = producer, args = (work_q, control_q, filename,logger)))
|
||||||
for worker in threads:
|
for worker in threads:
|
||||||
worker.start()
|
worker.start()
|
||||||
for worker in threads:
|
for worker in threads:
|
||||||
worker.join()
|
worker.join()
|
||||||
return result_list
|
return result_list
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
import logging
|
import logging
|
||||||
logger = logging.getLogger()
|
logger = logging.getLogger()
|
||||||
|
|||||||
@@ -1,20 +1,22 @@
|
|||||||
|
# This Python file uses the following encoding: utf-8
|
||||||
"""
|
"""
|
||||||
The provided Python script sets up a Flask web server to manage a list of music files, with
|
The provided Python script sets up a Flask web server to manage a list of music files, with
|
||||||
functions for rescanning the music folder, updating the music list, and serving the music list via
|
functions for rescanning the music folder, updating the music list, and serving the music list via
|
||||||
API endpoints.
|
API endpoints.
|
||||||
"""
|
"""
|
||||||
import json
|
import json
|
||||||
|
import logging
|
||||||
import re
|
import re
|
||||||
from pathlib import Path
|
|
||||||
from platform import uname
|
|
||||||
import threading
|
import threading
|
||||||
import time
|
import time
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
|
from logging import handlers
|
||||||
|
from pathlib import Path
|
||||||
|
from platform import uname
|
||||||
from sys import platform
|
from sys import platform
|
||||||
import logging
|
|
||||||
from flask import Flask, jsonify, request
|
|
||||||
from waitress import serve
|
|
||||||
|
|
||||||
|
from flask import Flask, jsonify, redirect, request, send_from_directory
|
||||||
|
from waitress import serve
|
||||||
|
|
||||||
HOST_ADDRESS = "192.168.1.15"
|
HOST_ADDRESS = "192.168.1.15"
|
||||||
PORT_ADDRESS = 5000
|
PORT_ADDRESS = 5000
|
||||||
@@ -26,24 +28,42 @@ if platform in ("linux", "linux2"):
|
|||||||
NETRC_FILE = "/home/mtuszowski/.netrc"
|
NETRC_FILE = "/home/mtuszowski/.netrc"
|
||||||
LOGSTORE = "/home/mtuszowski/conjurer/logs/"
|
LOGSTORE = "/home/mtuszowski/conjurer/logs/"
|
||||||
ENCODING = "utf-8"
|
ENCODING = "utf-8"
|
||||||
GRAPHICS_PATH = "/home/mtuszowski/conjurer/Conjurer_graphics/"
|
|
||||||
|
|
||||||
else:
|
else:
|
||||||
LOGFILE = "/home/pi/Conjurer/discord_mus_service.log"
|
LOGFILE = "/home/pi/Conjurer/discord_mus_service.log"
|
||||||
MEMORY_FIVE_SIARA = "/home/pi/Conjurer/pamiec.json"
|
|
||||||
SYSTEM_GPT_SETTINGS = "/home/pi/Conjurer/system_gpt_settings.json"
|
|
||||||
MEMORY_FIVE_MUZYKA = "/home/pi/Conjurer/pamiec_muzyki.json"
|
|
||||||
MUSIC_FOLDER = "/home/pi/RetroPie/mp3/"
|
|
||||||
SETTINGS_FILE = "/home/pi/Conjurer/settings.json"
|
|
||||||
NETRC_FILE = "/home/pi/.netrc"
|
NETRC_FILE = "/home/pi/.netrc"
|
||||||
LOGSTORE = "/home/pi/RetroPie/logs/"
|
LOGSTORE = "/home/pi/RetroPie/logs/"
|
||||||
ACCIDENT_LOG = "/home/pi/Conjurer/accident_log.json"
|
|
||||||
ENCODING = "utf-8"
|
ENCODING = "utf-8"
|
||||||
GRAPHICS_PATH = "/home/pi/RetroPie/Conjurer_graphics/"
|
MUSIC_FOLDER = "/home/pi/RetroPie/mp3/"
|
||||||
DIR_PATH_SADOX = "/home/pi/RetroPie/Fansadox/"
|
|
||||||
|
|
||||||
|
|
||||||
music_file_list = []
|
music_file_list = []
|
||||||
|
|
||||||
|
|
||||||
|
def create_playlist():
|
||||||
|
"""
|
||||||
|
Reads a JSON file containing a playlist and writes the playlist items to a text file.
|
||||||
|
|
||||||
|
The JSON file path is '/home/pi/Conjurer/playlist.json',
|
||||||
|
and the text file path is '/home/pi/Conjurer/all_playlist.playlist'.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
JSONDecodeError: If the JSON file cannot be decoded.
|
||||||
|
|
||||||
|
"""
|
||||||
|
with open("/home/pi/Conjurer/playlist.json", "r+", encoding="utf-8") as r_file:
|
||||||
|
with open(
|
||||||
|
"/home/pi/Conjurer/all_playlist.playlist", "w", encoding="utf-8"
|
||||||
|
) as w_file:
|
||||||
|
try:
|
||||||
|
playlist = json.load(r_file)
|
||||||
|
for item in playlist:
|
||||||
|
w_file.write(item)
|
||||||
|
w_file.write("\n")
|
||||||
|
except json.JSONDecodeError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
def rescan():
|
def rescan():
|
||||||
"""
|
"""
|
||||||
The `rescan` function logs a message, scans for mp3 files in a specified folder,
|
The `rescan` function logs a message, scans for mp3 files in a specified folder,
|
||||||
@@ -55,6 +75,14 @@ def rescan():
|
|||||||
if platform == "win32":
|
if platform == "win32":
|
||||||
temp_music_file = temp_music_file.replace("/", "\\")
|
temp_music_file = temp_music_file.replace("/", "\\")
|
||||||
music_file_list.append(temp_music_file)
|
music_file_list.append(temp_music_file)
|
||||||
|
with open("/home/pi/Conjurer/playlist.json", "r+", encoding="utf-8") as s_file:
|
||||||
|
database = music_file_list
|
||||||
|
# self.app.logger.info("DUMPING DATA")
|
||||||
|
s_file.truncate(0)
|
||||||
|
s_file.seek(0)
|
||||||
|
json.dump(database, s_file)
|
||||||
|
create_playlist()
|
||||||
|
|
||||||
|
|
||||||
def thread_rescan():
|
def thread_rescan():
|
||||||
"""
|
"""
|
||||||
@@ -63,14 +91,31 @@ def thread_rescan():
|
|||||||
"""
|
"""
|
||||||
logging.info("Starting filesystemupdater")
|
logging.info("Starting filesystemupdater")
|
||||||
while True:
|
while True:
|
||||||
time.sleep(60*60*24)
|
time.sleep(60 * 60 * 24)
|
||||||
logging.info("Rescan triggered")
|
logging.info("Rescan triggered")
|
||||||
rescan()
|
rescan()
|
||||||
|
|
||||||
|
|
||||||
app = Flask(__name__)
|
app = Flask(__name__)
|
||||||
|
|
||||||
def wyszukaj(word_list, how_many):
|
|
||||||
logger = logging.getLogger()
|
# TODO: Odpalić wyszukiwarki w wątkach i dopiero po wszystkim zsumować wyszukiwanie.
|
||||||
|
def wyszukaj(word_list, how_many, _logger=None, return_to_bot=True):
|
||||||
|
"""
|
||||||
|
Searches for music files based on a list of keywords and returns a list of matching files.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
word_list (list): A list of keywords to search for.
|
||||||
|
how_many (int): The maximum number of matching files to return.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
list: A list of matching files, sorted by their search weight.
|
||||||
|
|
||||||
|
"""
|
||||||
|
if _logger is None:
|
||||||
|
fun_logger = logging.getLogger()
|
||||||
|
else:
|
||||||
|
fun_logger = _logger
|
||||||
return_list = []
|
return_list = []
|
||||||
search_weight = []
|
search_weight = []
|
||||||
|
|
||||||
@@ -84,7 +129,7 @@ def wyszukaj(word_list, how_many):
|
|||||||
skip_start = 1
|
skip_start = 1
|
||||||
for word in word_list[skip_start:]:
|
for word in word_list[skip_start:]:
|
||||||
token_weight = len(word)
|
token_weight = len(word)
|
||||||
logger.info("Słowo kluczowe: %s", word)
|
fun_logger.info("Słowo kluczowe: %s", word)
|
||||||
itr = 0
|
itr = 0
|
||||||
for file in music_file_list:
|
for file in music_file_list:
|
||||||
file = file.split(SEPARATOR_FILE_PATH)
|
file = file.split(SEPARATOR_FILE_PATH)
|
||||||
@@ -124,6 +169,7 @@ def wyszukaj(word_list, how_many):
|
|||||||
else:
|
else:
|
||||||
skip = 4
|
skip = 4
|
||||||
matched_times = 1
|
matched_times = 1
|
||||||
|
# TODO: To do wątków albo for asynchroniczny
|
||||||
for itm in all_words[skip:]:
|
for itm in all_words[skip:]:
|
||||||
pingu += 1
|
pingu += 1
|
||||||
pattern = ".*" + word + ".*"
|
pattern = ".*" + word + ".*"
|
||||||
@@ -139,26 +185,41 @@ def wyszukaj(word_list, how_many):
|
|||||||
matched_times += 1
|
matched_times += 1
|
||||||
itr += 1
|
itr += 1
|
||||||
|
|
||||||
logger.info("Stworzylem tablice wag zajęło mi to %s", datetime.now() - time_start)
|
fun_logger.info(
|
||||||
|
"Stworzylem tablice wag zajęło mi to %s", datetime.now() - time_start
|
||||||
|
)
|
||||||
time_start = datetime.now()
|
time_start = datetime.now()
|
||||||
item_to_search = max_weight(search_weight)
|
item_to_search = max_weight(search_weight)
|
||||||
itr = 0
|
itr = 0
|
||||||
if item_to_search == 0:
|
if item_to_search == 0:
|
||||||
return None
|
return []
|
||||||
not_found = True
|
not_found = True
|
||||||
return_list = []
|
return_list = []
|
||||||
if int(how_many) <= 0:
|
if int(how_many) <= 0:
|
||||||
logger.info("Jeden plik do zagrania")
|
fun_logger.info("Jeden plik do zagrania")
|
||||||
while not_found:
|
while not_found:
|
||||||
if search_weight[itr][0] == item_to_search:
|
if search_weight[itr][0] == item_to_search:
|
||||||
return_list.append(search_weight[itr])
|
return_list.append(search_weight[itr])
|
||||||
return return_list
|
if not return_to_bot:
|
||||||
|
with open(
|
||||||
|
"/home/pi/Conjurer/priority_queue.playlist", "r+", encoding="utf-8"
|
||||||
|
) as s_file:
|
||||||
|
s_file.write(search_weight[itr][1])
|
||||||
|
break
|
||||||
itr += 1
|
itr += 1
|
||||||
else:
|
else:
|
||||||
logger.info("Wiele plików do zagrania")
|
fun_logger.info("Wiele plików do zagrania")
|
||||||
search_weight.sort(key=lambda x: x[0], reverse=True)
|
search_weight.sort(key=lambda x: x[0], reverse=True)
|
||||||
return_list.extend(search_weight[: int(how_many)])
|
return_list.extend(search_weight[: int(how_many)])
|
||||||
return return_list
|
if not return_to_bot:
|
||||||
|
|
||||||
|
with open(
|
||||||
|
"/home/pi/Conjurer/priority_queue.playlist", "a", encoding="utf-8"
|
||||||
|
) as s_file:
|
||||||
|
for item in return_list:
|
||||||
|
s_file.write(item[1] + "\n")
|
||||||
|
return return_list
|
||||||
|
|
||||||
|
|
||||||
def max_weight(lista):
|
def max_weight(lista):
|
||||||
"""
|
"""
|
||||||
@@ -188,6 +249,47 @@ def remove_characters(string, character):
|
|||||||
return string.replace(character, "")
|
return string.replace(character, "")
|
||||||
|
|
||||||
|
|
||||||
|
@app.route("/stream", methods=["GET"])
|
||||||
|
def stream_music():
|
||||||
|
"""
|
||||||
|
The function `get_music_list` returns a JSON object containing a list of music files.
|
||||||
|
:return: A JSON response containing a key "music_file_list" with the value of the variable
|
||||||
|
`music_file_list`.
|
||||||
|
"""
|
||||||
|
return send_from_directory("/tmp/hls", "stream.m3u8")
|
||||||
|
|
||||||
|
@app.route('/<string:file_name>')
|
||||||
|
def stream(file_name):
|
||||||
|
video_dir = '/tmp/hls'
|
||||||
|
return send_from_directory(video_dir, file_name)
|
||||||
|
|
||||||
|
|
||||||
|
@app.route("/stream_mp3", methods=["GET"])
|
||||||
|
def stream_music_mp3():
|
||||||
|
"""
|
||||||
|
The function `get_music_list` returns a JSON object containing a list of music files.
|
||||||
|
:return: A JSON response containing a key "music_file_list" with the value of the variable
|
||||||
|
`music_file_list`.
|
||||||
|
"""
|
||||||
|
return redirect("http://www.example.com", code=302)
|
||||||
|
|
||||||
|
|
||||||
|
@app.route("/clear_pr_pls", methods=["GET"])
|
||||||
|
def clear_pr_pls():
|
||||||
|
"""
|
||||||
|
The function `get_music_list` returns a JSON object containing a list of music files.
|
||||||
|
:return: A JSON response containing a key "music_file_list" with the value of the variable
|
||||||
|
`music_file_list`.
|
||||||
|
"""
|
||||||
|
with open(
|
||||||
|
"/home/pi/Conjurer/priority_queue.playlist", "w", encoding="utf-8"
|
||||||
|
) as s_file:
|
||||||
|
s_file.truncate(0)
|
||||||
|
return_data = jsonify(isError=False, message="Success", statusCode=200, data=[])
|
||||||
|
return return_data, 200
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@app.route("/mp3", methods=["GET"])
|
@app.route("/mp3", methods=["GET"])
|
||||||
def get_music_list():
|
def get_music_list():
|
||||||
"""
|
"""
|
||||||
@@ -218,6 +320,7 @@ def update_music_list():
|
|||||||
)
|
)
|
||||||
return return_data
|
return return_data
|
||||||
|
|
||||||
|
|
||||||
@app.route("/get_music", methods=["POST"])
|
@app.route("/get_music", methods=["POST"])
|
||||||
def look_for_playlist():
|
def look_for_playlist():
|
||||||
"""
|
"""
|
||||||
@@ -234,7 +337,31 @@ def look_for_playlist():
|
|||||||
app.logger.info(record["lista_slow"])
|
app.logger.info(record["lista_slow"])
|
||||||
app.logger.info(record["UUID"])
|
app.logger.info(record["UUID"])
|
||||||
app.logger.info(record["dlugosc_plejlisty"])
|
app.logger.info(record["dlugosc_plejlisty"])
|
||||||
return_data = wyszukaj(record["lista_slow"],record["dlugosc_plejlisty"])
|
return_data = wyszukaj(record["lista_slow"], record["dlugosc_plejlisty"], app.logger, True)
|
||||||
|
|
||||||
|
return_data = (
|
||||||
|
jsonify(isError=False, message="Success", statusCode=200, data=return_data),
|
||||||
|
200,
|
||||||
|
)
|
||||||
|
return return_data
|
||||||
|
|
||||||
|
@app.route("/add_to_priority", methods=["POST"])
|
||||||
|
def add_to_priority():
|
||||||
|
"""
|
||||||
|
The function `update_music_list` receives a POST request with a JSON payload, logs the received
|
||||||
|
item, adds it to a music file list, and returns a success message along with the updated record.
|
||||||
|
:return: The function `update_music_list` is returning a tuple containing a JSON response and a
|
||||||
|
status code. The JSON response includes keys `isError`, `message`, `statusCode`, and `data`,
|
||||||
|
with values indicating the success of the operation and the data that was
|
||||||
|
received and added to the `music_file_list`.
|
||||||
|
The status code returned is 200, indicating a successful response.
|
||||||
|
"""
|
||||||
|
record = json.loads(request.data)
|
||||||
|
app.logger.info(record)
|
||||||
|
app.logger.info(record["lista_slow"])
|
||||||
|
app.logger.info(record["UUID"])
|
||||||
|
app.logger.info(record["dlugosc_plejlisty"])
|
||||||
|
return_data = wyszukaj(record["lista_slow"], record["dlugosc_plejlisty"], app.logger, False)
|
||||||
|
|
||||||
return_data = (
|
return_data = (
|
||||||
jsonify(isError=False, message="Success", statusCode=200, data=return_data),
|
jsonify(isError=False, message="Success", statusCode=200, data=return_data),
|
||||||
@@ -251,6 +378,7 @@ def flask_debug():
|
|||||||
# trunk-ignore(bandit/B201)
|
# trunk-ignore(bandit/B201)
|
||||||
app.run(debug=True, use_reloader=False, host=HOST_ADDRESS, port=PORT_ADDRESS)
|
app.run(debug=True, use_reloader=False, host=HOST_ADDRESS, port=PORT_ADDRESS)
|
||||||
|
|
||||||
|
|
||||||
def waitress_run():
|
def waitress_run():
|
||||||
"""
|
"""
|
||||||
The `waitress_run` function serves the `app` on host "0.0.0.0"
|
The `waitress_run` function serves the `app` on host "0.0.0.0"
|
||||||
@@ -258,12 +386,27 @@ def waitress_run():
|
|||||||
"""
|
"""
|
||||||
serve(app, host=HOST_ADDRESS, port=PORT_ADDRESS)
|
serve(app, host=HOST_ADDRESS, port=PORT_ADDRESS)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
logger = logging.getLogger('conjurer_musician')
|
logger = logging.getLogger("conjurer_musician")
|
||||||
logger.setLevel(logging.DEBUG)
|
logger.setLevel(logging.DEBUG)
|
||||||
|
formatter = logging.Formatter(
|
||||||
|
"%(asctime)s - %(name)s - %(levelname)s - %(message)s"
|
||||||
|
)
|
||||||
|
handler = handlers.RotatingFileHandler(
|
||||||
|
filename=LOGFILE,
|
||||||
|
encoding=ENCODING,
|
||||||
|
mode="a",
|
||||||
|
maxBytes=6 * 1024 * 1024,
|
||||||
|
backupCount=6,
|
||||||
|
)
|
||||||
|
handler.setFormatter(formatter)
|
||||||
|
logger.addHandler(handler)
|
||||||
|
|
||||||
rescan()
|
rescan()
|
||||||
logger.info("Started")
|
logger.info("Started")
|
||||||
threads = []
|
threads = []
|
||||||
|
#threads.append(threading.Thread(target=flask_debug))
|
||||||
threads.append(threading.Thread(target=waitress_run))
|
threads.append(threading.Thread(target=waitress_run))
|
||||||
threads.append(threading.Thread(target=thread_rescan))
|
threads.append(threading.Thread(target=thread_rescan))
|
||||||
|
|
||||||
@@ -272,5 +415,3 @@ if __name__ == "__main__":
|
|||||||
|
|
||||||
for worker in threads:
|
for worker in threads:
|
||||||
worker.join()
|
worker.join()
|
||||||
|
|
||||||
#TODO: Inne pliki - sadox, dokumenty naukowe itp.
|
|
||||||
@@ -0,0 +1,11 @@
|
|||||||
|
[Unit]
|
||||||
|
Description=Conjurer_musician
|
||||||
|
After=multi-user.target
|
||||||
|
|
||||||
|
[Service]
|
||||||
|
Type=simple
|
||||||
|
ExecStart=/home/pi/Conjurer/env/bin/python3 /home/pi/Conjurer/conjurer_musician.py
|
||||||
|
Restart=on-abort
|
||||||
|
|
||||||
|
[Install]
|
||||||
|
WantedBy=multi-user.target
|
||||||
Executable
+34804
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,48 @@
|
|||||||
|
let json.parse credentials = file.contents("/home/pi/Conjurer/icecast_credentials.json")
|
||||||
|
enable_replaygain_metadata()
|
||||||
|
|
||||||
|
s1 = replaygain(playlist(reload_mode="watch", "/home/pi/Conjurer/all_playlist.playlist"))
|
||||||
|
s2 = replaygain(playlist(reload_mode="watch", "/home/pi/Conjurer/priority_queue.playlist"))
|
||||||
|
|
||||||
|
s = random(id="randomizer", weights=[7, 10], [s2, s1])
|
||||||
|
|
||||||
|
interactive.harbor(port = 9999)
|
||||||
|
f = interactive.float("f", description="Frequency", min=0., max=1000.,unit="Hz", 200.)
|
||||||
|
g = interactive.float("g", description="Gain", min=0., max=20.,unit="dB", 8.)
|
||||||
|
b = bass_boost(frequency=f, gain=g, s)
|
||||||
|
s = add([s, b])
|
||||||
|
a = interactive.float("main_volume", min=0., max=20., 1.)
|
||||||
|
s = compress.multiband.interactive(bands=3, s)
|
||||||
|
|
||||||
|
s = nrj(normalize (s))
|
||||||
|
s = crossfade(fade_out=3., fade_in=3., duration=5., s)
|
||||||
|
s = amplify(a,s)
|
||||||
|
|
||||||
|
s = blank.skip(max_blank=2., s)
|
||||||
|
log_to_stdout = true
|
||||||
|
log_to_file = true
|
||||||
|
logpath = "/home/pi/Conjurer/radio_log.log"
|
||||||
|
loglevel = 4
|
||||||
|
set("log.stdout",log_to_stdout)
|
||||||
|
set("log.level",loglevel)
|
||||||
|
|
||||||
|
# Enable logging to file. See variable: logpath
|
||||||
|
set("log.file",log_to_file)
|
||||||
|
set("log.file.path",logpath)
|
||||||
|
|
||||||
|
emergency = single("/home/pi/RetroPie/mp3/Youtube/Dr. Peacock - Trip to Ireland [GvrvQTUbUcA].mp3")
|
||||||
|
radio = fallback(id="switcher",track_sensitive = false, [s, emergency])
|
||||||
|
|
||||||
|
handle_metadata = fun (m) -> begin
|
||||||
|
log.info(m["title"])
|
||||||
|
log.info(m["artist"])
|
||||||
|
log.info(m["filename"])
|
||||||
|
end
|
||||||
|
radio.on_metadata(handle_metadata)
|
||||||
|
interactive.persistent("script.params")
|
||||||
|
|
||||||
|
|
||||||
|
# Enable logging on Standard Output and set logging level.
|
||||||
|
|
||||||
|
output.file.hls("/tmp/hls",[("mp3-low", %mp3(bitrate=96)), ("mp3-hi", %mp3(bitrate=160))],radio)
|
||||||
|
output.icecast(%mp3, host="retropie", port=8000,password=credentials.password, icy_metadata="true", mount="mp3-stream", radio)
|
||||||
@@ -0,0 +1,14 @@
|
|||||||
|
[Unit]
|
||||||
|
Description=radio_conjurer Liquidsoap daemon
|
||||||
|
After=network.target
|
||||||
|
Documentation=http://liquidsoap.fm/
|
||||||
|
|
||||||
|
[Service]
|
||||||
|
User=pi
|
||||||
|
WorkingDirectory=/home/pi/Conjurer/
|
||||||
|
ExecStart=/home/pi/.opam/4.13.0/bin/liquidsoap /home/pi/Conjurer/radio_conjurer.liq
|
||||||
|
Restart=always
|
||||||
|
|
||||||
|
[Install]
|
||||||
|
WantedBy=multi-user.target
|
||||||
|
Alias=radio_conjurer-liquidsoap.service
|
||||||
File diff suppressed because one or more lines are too long
Executable
+6
@@ -0,0 +1,6 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
cd ./conjurer
|
||||||
|
git pull
|
||||||
|
cd ..
|
||||||
|
cp ./conjurer/conjurer_musician/conjurer_musician.py ./Conjurer
|
||||||
|
sudo systemctl restart conjurer_musician.service
|
||||||
@@ -0,0 +1,6 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
cd ./conjurer
|
||||||
|
git pull
|
||||||
|
cd ..
|
||||||
|
cp ./conjurer/conjurer_musician/radio_conjurer.liq ./Conjurer
|
||||||
|
sudo systemctl restart radio_service.service
|
||||||
@@ -1,11 +0,0 @@
|
|||||||
[Unit]
|
|
||||||
Description=Conjurer_musician
|
|
||||||
After=multi-user.target
|
|
||||||
|
|
||||||
[Service]
|
|
||||||
Type=simple
|
|
||||||
ExecStart=/home/pi/Conjurer-service/conjurer/file_webservice/bin/python3 /home/pi/Conjurer-service/conjurer/file_webservice/conjurer_musician.py
|
|
||||||
Restart=on-abort
|
|
||||||
|
|
||||||
[Install]
|
|
||||||
WantedBy=multi-user.target
|
|
||||||
@@ -1,11 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
python3 -m venv /home/pi/Conjurer-service/conjurer/file_webservice
|
|
||||||
source ./bin/activate
|
|
||||||
./bin/python3 -m pip install --upgrade pip
|
|
||||||
./bin/python3 -m pip install -r requirements_musician.txt
|
|
||||||
deactivate
|
|
||||||
|
|
||||||
sudo cp ./conjurer_musician.service /etc/systemd/system/
|
|
||||||
sudo systemctl daemon-reload
|
|
||||||
sudo systemctl start conjurer_musician.service
|
|
||||||
sudo systemctl enable conjurer_musician.service
|
|
||||||
+6
-2
@@ -1,5 +1,9 @@
|
|||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
|
cd /home/pi
|
||||||
|
mdkir Conjurer
|
||||||
|
cd Conjurer
|
||||||
python3 -m venv /home/pi/Conjurer/env
|
python3 -m venv /home/pi/Conjurer/env
|
||||||
|
cp /home/pi/conjurer/requirements_bot.txt /home/pi/Conjurer/
|
||||||
source ./env/bin/activate
|
source ./env/bin/activate
|
||||||
./env/bin/python3 -m pip install --upgrade pip
|
./env/bin/python3 -m pip install --upgrade pip
|
||||||
./env/bin/python3 -m pip install -r requirements_bot.txt
|
./env/bin/python3 -m pip install -r requirements_bot.txt
|
||||||
@@ -8,5 +12,5 @@ sed -i '1i\import shutil' ./env/lib/python3.11/site-packages/spotify_dl/youtube.
|
|||||||
deactivate
|
deactivate
|
||||||
sudo cp ./conjurer.service /etc/systemd/system/
|
sudo cp ./conjurer.service /etc/systemd/system/
|
||||||
sudo systemctl daemon-reload
|
sudo systemctl daemon-reload
|
||||||
sudo systemctl start conjurer_musician.service
|
sudo systemctl start conjurer.service
|
||||||
sudo systemctl enable conjurer_musician.service
|
sudo systemctl enable conjurer.service
|
||||||
|
|||||||
Executable
+26
@@ -0,0 +1,26 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
cd /home/pi
|
||||||
|
mkdir Conjurer
|
||||||
|
cd Conjurer
|
||||||
|
python3 -m venv /home/pi/Conjurer/env
|
||||||
|
source /home/pi/Conjurer/env/bin/activate
|
||||||
|
cp /home/pi/conjurer/conjurer_musician/requirements_musician.txt /home/pi/Conjurer/
|
||||||
|
cp /home/pi/conjurer/conjurer_musician/conjurer_musician.py /home/pi/Conjurer/
|
||||||
|
cp /home/pi/conjurer/conjurer_musician/radio_conjurer.liq /home/pi/Conjurer/
|
||||||
|
touch /home/pi/Conjurer/discord_mus_service.log
|
||||||
|
touch /home/pi/Conjurer/all_playlist.txt
|
||||||
|
touch /home/pi/Conjurer/playlist.json
|
||||||
|
touch /home/pi/Conjurer/priority_queue.playlist
|
||||||
|
touch /home/pi/Conjurer/radio_log.log
|
||||||
|
./env/bin/python3 -m pip install --upgrade pip
|
||||||
|
./env/bin/python3 -m pip install -r requirements_musician.txt
|
||||||
|
deactivate
|
||||||
|
sudo cp /home/pi/conjurer/conjurer_musician/conjurer_musician.service /etc/systemd/system/
|
||||||
|
sudo systemctl daemon-reload
|
||||||
|
sudo systemctl start conjurer_musician.service
|
||||||
|
sudo systemctl enable conjurer_musician.service
|
||||||
|
|
||||||
|
sudo cp /home/pi/conjurer/conjurer_musician/radio_service.service /etc/systemd/system/
|
||||||
|
sudo systemctl daemon-reload
|
||||||
|
sudo systemctl start radio_service.service
|
||||||
|
sudo systemctl enable radio_service.service
|
||||||
-161654
File diff suppressed because one or more lines are too long
@@ -1 +0,0 @@
|
|||||||
{"12345678-1234-5678-1234-567812345678": [{"DOI": "10.1057/9781137435026", "title": ["Queer BDSM Intimacies"], "type": "book", "container": "NO"}]}
|
|
||||||
Reference in New Issue
Block a user