mirror of
https://github.com/migatu/conjurer.git
synced 2026-07-14 21:38:38 +00:00
Compare commits
42 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| adb0f9b468 | |||
| ce806c873f | |||
| a54a6686c2 | |||
| d226d99236 | |||
| a827c5e429 | |||
| 71d8e03933 | |||
| 803ad01a83 | |||
| 81d5fcfb8f | |||
| 4d79c51e20 | |||
| f320b20850 | |||
| c39900e15f | |||
| c728396205 | |||
| 0f7e5d4a25 | |||
| ce91aec6ff | |||
| 759c06aa50 | |||
| 1ebd58b8ac | |||
| 82bb6cbbad | |||
| e54150fd5d | |||
| 167fe59456 | |||
| 3a02846a17 | |||
| 9c3e51631c | |||
| c3dd4ad8cf | |||
| 5d51e99c70 | |||
| 2c9c86990c | |||
| b87cf7f4d8 | |||
| a68745713b | |||
| bbe1b89007 | |||
| 27fefdccb5 | |||
| 64849f728f | |||
| 9b09c2c13f | |||
| f2fb2232ee | |||
| 7e4dfbf8cd | |||
| 81a387eb4f | |||
| 408f6e5c4e | |||
| a266e5d240 | |||
| 917b57c586 | |||
| 0e15218abd | |||
| f56773b2e1 | |||
| 944dfe2c18 | |||
| 2fd362a1d5 | |||
| 47945ab3dd | |||
| c0b362df6d |
@@ -16,10 +16,10 @@ import os
|
|||||||
import random
|
import random
|
||||||
import re
|
import re
|
||||||
import shutil
|
import shutil
|
||||||
|
import subprocess
|
||||||
import sys
|
import sys
|
||||||
import threading
|
import threading
|
||||||
import uuid
|
import uuid
|
||||||
import subprocess
|
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from logging import handlers
|
from logging import handlers
|
||||||
from pathlib import Path, PurePath
|
from pathlib import Path, PurePath
|
||||||
@@ -42,11 +42,16 @@ from discord.ext import commands, tasks
|
|||||||
from spotipy.oauth2 import SpotifyClientCredentials
|
from spotipy.oauth2 import SpotifyClientCredentials
|
||||||
|
|
||||||
import yt_dlp
|
import yt_dlp
|
||||||
|
from communication_subroutine import (
|
||||||
|
IN_COMM_Q,
|
||||||
|
OUT_COMM_Q,
|
||||||
|
QueryControl,
|
||||||
|
comm_subroutine,
|
||||||
|
PREPPED_TRACKS,
|
||||||
|
)
|
||||||
from spotify_dl import spotify
|
from spotify_dl import spotify
|
||||||
from spotify_dl import youtube as youtube_download
|
from spotify_dl import youtube as youtube_download
|
||||||
|
|
||||||
from communication_subroutine import comm_subroutine, IN_COMM_Q, OUT_COMM_Q, QueryControl
|
|
||||||
|
|
||||||
# *=========================== Classes definition
|
# *=========================== Classes definition
|
||||||
|
|
||||||
# The above code is defining a TypedDict called `Music_Config` in Python. This TypedDict has three
|
# The above code is defining a TypedDict called `Music_Config` in Python. This TypedDict has three
|
||||||
@@ -63,7 +68,6 @@ Music_Config = TypedDict(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class MusicFileList(object):
|
class MusicFileList(object):
|
||||||
"""
|
"""
|
||||||
The `MusicFileList` class manages a list of music files, with the ability to refresh the list from a
|
The `MusicFileList` class manages a list of music files, with the ability to refresh the list from a
|
||||||
@@ -136,10 +140,15 @@ MUZYKA_MOJEGO_LUDU_SLOWA_KLUCZOWE = 15
|
|||||||
MUZYKA_MOJEGO_LUDU_PLAJLISTA = 30
|
MUZYKA_MOJEGO_LUDU_PLAJLISTA = 30
|
||||||
|
|
||||||
FILE_SERVICE_ADDRESS = "http://192.168.1.15:5000"
|
FILE_SERVICE_ADDRESS = "http://192.168.1.15:5000"
|
||||||
|
RADIO_HARBOR_ADDRESS = "http://192.168.1.15:54321"
|
||||||
|
SKIP_TRACK = "/skip"
|
||||||
|
|
||||||
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"
|
ADD_TO_PRIO_PLAYLIST = "/add_to_priority"
|
||||||
|
CREATE_PRIO_PLAYLIST = "/create_priority_playlist"
|
||||||
|
|
||||||
REQUEST_MUSIC = "/request_radio_file"
|
REQUEST_MUSIC = "/request_radio_file"
|
||||||
CLEAR_PRIO = "/clear_pr_pls"
|
CLEAR_PRIO = "/clear_pr_pls"
|
||||||
LIBRARIAN_SERVICE_ADDRESS = "http://192.168.1.192:5001"
|
LIBRARIAN_SERVICE_ADDRESS = "http://192.168.1.192:5001"
|
||||||
@@ -279,7 +288,6 @@ for key in word_reactions:
|
|||||||
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="$")
|
||||||
@@ -399,7 +407,13 @@ async def on_message(message):
|
|||||||
result, MESSAGE_TABLE = await handle_response(
|
result, MESSAGE_TABLE = await handle_response(
|
||||||
prompt, vykidailo, bartender, MESSAGE_TABLE, username, "CONVERSATION"
|
prompt, vykidailo, bartender, MESSAGE_TABLE, username, "CONVERSATION"
|
||||||
)
|
)
|
||||||
await message.reply(result)
|
|
||||||
|
if len(result) < 1500:
|
||||||
|
await message.reply(result)
|
||||||
|
else:
|
||||||
|
while len(result) > 1500:
|
||||||
|
await message.reply(result[:1500])
|
||||||
|
result = result[1500:]
|
||||||
if "imaginuje sobie:" in message.content:
|
if "imaginuje sobie:" in message.content:
|
||||||
async with channel.typing():
|
async with channel.typing():
|
||||||
logger.info("Poczatek procedury obrazkowej")
|
logger.info("Poczatek procedury obrazkowej")
|
||||||
@@ -632,7 +646,12 @@ async def play(ctx, zamawial=None, arg=None):
|
|||||||
query, vykidailo, bartender, message_table_muzyka, username, "MUSIC"
|
query, vykidailo, bartender, message_table_muzyka, username, "MUSIC"
|
||||||
)
|
)
|
||||||
logger.debug("Obecna historia czatu: %s", message_table_muzyka)
|
logger.debug("Obecna historia czatu: %s", message_table_muzyka)
|
||||||
await ctx.send(result)
|
if len(result) < 1500:
|
||||||
|
await ctx.send(result)
|
||||||
|
else:
|
||||||
|
while len(result) > 1500:
|
||||||
|
await ctx.send(result[:1500])
|
||||||
|
result = result[1500:]
|
||||||
|
|
||||||
|
|
||||||
async def archive_channel(channel_no):
|
async def archive_channel(channel_no):
|
||||||
@@ -671,6 +690,7 @@ async def check_music():
|
|||||||
await play(MUZYKA["ctx"], MUZYKA["requester"])
|
await play(MUZYKA["ctx"], MUZYKA["requester"])
|
||||||
await asyncio.sleep(2)
|
await asyncio.sleep(2)
|
||||||
|
|
||||||
|
|
||||||
@tasks.loop(seconds=3)
|
@tasks.loop(seconds=3)
|
||||||
async def check_data_q():
|
async def check_data_q():
|
||||||
"""
|
"""
|
||||||
@@ -686,40 +706,42 @@ async def check_data_q():
|
|||||||
for doi in fresh_data.entries:
|
for doi in fresh_data.entries:
|
||||||
logger.info(doi)
|
logger.info(doi)
|
||||||
desc = fresh_data.entries[doi]
|
desc = fresh_data.entries[doi]
|
||||||
title = desc['Title'][0]
|
title = desc["Title"][0]
|
||||||
entries.append(f"{l_p}. {title} pod linkiem https://www.sci-hub.se/{doi} i jest to {desc['type']}\n")
|
entries.append(
|
||||||
l_p+=1
|
f"{l_p}. {title} pod linkiem https://www.sci-hub.se/{doi} i jest to {desc['type']}\n"
|
||||||
message = '*Z podłogi wysuwa się winda na książki*'
|
)
|
||||||
|
l_p += 1
|
||||||
|
message = "*Z podłogi wysuwa się winda na książki*"
|
||||||
if fresh_data.ctx is not None:
|
if fresh_data.ctx is not None:
|
||||||
ctx = fresh_data.ctx
|
ctx = fresh_data.ctx
|
||||||
message += f' Dzielny poszukiwaczu @{searcher} w odpowiedzi na twoje zapytanie {query} nr {fresh_data.uuid}'
|
message += f" Dzielny poszukiwaczu @{searcher} w odpowiedzi na twoje zapytanie {query} nr {fresh_data.uuid}"
|
||||||
else:
|
else:
|
||||||
ctx = client.get_channel(1062047571557744721)
|
ctx = client.get_channel(1062047571557744721)
|
||||||
message += f' Mam wynik dla zapytania {fresh_data.uuid} ale że Hammer coś grzebał nie pamiętam kto to chciał i co chciał'
|
message += f" Mam wynik dla zapytania {fresh_data.uuid} ale że Hammer coś grzebał nie pamiętam kto to chciał i co chciał"
|
||||||
|
|
||||||
|
message += "nasza biblioteka służy Ci odpowiedzią. Przy dźwięku fanfar winda się otwiera a w środku "
|
||||||
message += 'nasza biblioteka służy Ci odpowiedzią. Przy dźwięku fanfar winda się otwiera a w środku '
|
if len(entries) < 1:
|
||||||
if len(entries)<1:
|
|
||||||
message += "niestety nie ma nic"
|
message += "niestety nie ma nic"
|
||||||
await ctx.send(message)
|
await ctx.send(message)
|
||||||
elif len(entries)>=1 and len(entries)<5:
|
elif len(entries) >= 1 and len(entries) < 5:
|
||||||
message += " znajduje się coś:\n"
|
message += " znajduje się coś:\n"
|
||||||
for item in entries:
|
for item in entries:
|
||||||
message+=item
|
message += item
|
||||||
await ctx.send(message)
|
await ctx.send(message)
|
||||||
else:
|
else:
|
||||||
message+= " znajduje się cholernie dużo:\n"
|
message += " znajduje się cholernie dużo:\n"
|
||||||
for item in entries:
|
for item in entries:
|
||||||
message+=item
|
message += item
|
||||||
if len(message) > 1500:
|
if len(message) > 1500:
|
||||||
await ctx.send(message)
|
await ctx.send(message)
|
||||||
message = ""
|
message = ""
|
||||||
|
|
||||||
#Kept for sentimental reasons
|
# Kept for sentimental reasons
|
||||||
#await ctx.send(f"O. A tak będzie wyglądało coś ciekawego w przyszłości: {data}")
|
# 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
|
||||||
|
|
||||||
|
|
||||||
@tasks.loop(seconds=120)
|
@tasks.loop(seconds=120)
|
||||||
async def check_self():
|
async def check_self():
|
||||||
"""
|
"""
|
||||||
@@ -1077,9 +1099,13 @@ async def get_file(ctx, source, link):
|
|||||||
logger.info(item_type)
|
logger.info(item_type)
|
||||||
logger.info(item_id)
|
logger.info(item_id)
|
||||||
logger.info("Spotify link provided")
|
logger.info("Spotify link provided")
|
||||||
file_list = await asyncio.to_thread(spotify.fetch_tracks, spotify_ctrl, item_type, link)
|
file_list = await asyncio.to_thread(
|
||||||
|
spotify.fetch_tracks, spotify_ctrl, item_type, link
|
||||||
|
)
|
||||||
logger.info(file_list)
|
logger.info(file_list)
|
||||||
directory_name = await asyncio.to_thread(spotify.get_item_name, spotify_ctrl, item_type, item_id)
|
directory_name = await asyncio.to_thread(
|
||||||
|
spotify.get_item_name, spotify_ctrl, item_type, item_id
|
||||||
|
)
|
||||||
logger.info(directory_name)
|
logger.info(directory_name)
|
||||||
logger.info("Spotify scrape done")
|
logger.info("Spotify scrape done")
|
||||||
url_data = {"urls": []}
|
url_data = {"urls": []}
|
||||||
@@ -1346,6 +1372,7 @@ async def max_weight(lista):
|
|||||||
|
|
||||||
# *=========================================== Define Commands
|
# *=========================================== Define Commands
|
||||||
|
|
||||||
|
|
||||||
@client.hybrid_command(
|
@client.hybrid_command(
|
||||||
name="skip_track",
|
name="skip_track",
|
||||||
description="Przeskocz kawałek w radiu",
|
description="Przeskocz kawałek w radiu",
|
||||||
@@ -1355,10 +1382,19 @@ async def skip_track(ctx):
|
|||||||
async with ctx.typing():
|
async with ctx.typing():
|
||||||
allowed = False
|
allowed = False
|
||||||
for role in ctx.author.roles:
|
for role in ctx.author.roles:
|
||||||
if role.name == "Thane":
|
if role.name in ("Thane", "Jarl"):
|
||||||
allowed = True
|
allowed = True
|
||||||
if not allowed:
|
if not allowed:
|
||||||
await ctx.send("Łapy precz od radia")
|
await ctx.send("Łapy precz od radia")
|
||||||
|
else:
|
||||||
|
coroutine = asyncio.to_thread(
|
||||||
|
requests.get,
|
||||||
|
f"{RADIO_HARBOR_ADDRESS}{SKIP_TRACK}",
|
||||||
|
timeout=360,
|
||||||
|
)
|
||||||
|
result = await coroutine
|
||||||
|
logger.info("Done %s", result)
|
||||||
|
await ctx.send("Zrobione szefie!")
|
||||||
|
|
||||||
|
|
||||||
@client.hybrid_command(
|
@client.hybrid_command(
|
||||||
@@ -1370,12 +1406,18 @@ async def zrestartuj_radio(ctx):
|
|||||||
async with ctx.typing():
|
async with ctx.typing():
|
||||||
allowed = False
|
allowed = False
|
||||||
for role in ctx.author.roles:
|
for role in ctx.author.roles:
|
||||||
if role.name == "Thane":
|
if role.name in ("Thane", "Jarl"):
|
||||||
allowed = True
|
allowed = True
|
||||||
if not allowed:
|
if not allowed:
|
||||||
await ctx.send("Łapy precz od radia")
|
await ctx.send("Łapy precz od radia")
|
||||||
else:
|
else:
|
||||||
subprocess.call(['sh', './restart_radio.sh'])
|
retcode = subprocess.run(
|
||||||
|
"/home/pi/Conjurer/restart_radio.sh",
|
||||||
|
shell=False,
|
||||||
|
check=False,
|
||||||
|
capture_output=True,
|
||||||
|
)
|
||||||
|
logger.info("Wynik: %s", retcode)
|
||||||
|
|
||||||
|
|
||||||
@client.hybrid_command(
|
@client.hybrid_command(
|
||||||
@@ -1619,6 +1661,7 @@ async def graj_muzyko(ctx):
|
|||||||
logger.info("Press play on tape completed")
|
logger.info("Press play on tape completed")
|
||||||
check_music.start()
|
check_music.start()
|
||||||
|
|
||||||
|
|
||||||
@client.hybrid_command(
|
@client.hybrid_command(
|
||||||
name="radio_hardkor",
|
name="radio_hardkor",
|
||||||
description="Włącza radio Conjurer na kanale #nocna-zmiana.",
|
description="Włącza radio Conjurer na kanale #nocna-zmiana.",
|
||||||
@@ -1643,9 +1686,10 @@ async def radio_hardkor(ctx):
|
|||||||
voice_client = client.voice_clients[0]
|
voice_client = client.voice_clients[0]
|
||||||
voice_client.play(
|
voice_client.play(
|
||||||
discord.PCMVolumeTransformer(
|
discord.PCMVolumeTransformer(
|
||||||
discord.FFmpegPCMAudio("http://95.175.16.246:666/mp3-stream"), volume=0.3
|
discord.FFmpegPCMAudio("http://95.175.16.246:666/mp3-stream"),
|
||||||
)
|
volume=0.3,
|
||||||
)
|
)
|
||||||
|
)
|
||||||
logger.info("Press play on radio completed")
|
logger.info("Press play on radio completed")
|
||||||
else:
|
else:
|
||||||
logger.error("Already playing")
|
logger.error("Already playing")
|
||||||
@@ -1653,7 +1697,6 @@ async def radio_hardkor(ctx):
|
|||||||
check_music.start()
|
check_music.start()
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@client.hybrid_command(
|
@client.hybrid_command(
|
||||||
name="cisza",
|
name="cisza",
|
||||||
description="Wyłącza muzykę i czyści plejliste.",
|
description="Wyłącza muzykę i czyści plejliste.",
|
||||||
@@ -1796,6 +1839,7 @@ 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(
|
@client.hybrid_command(
|
||||||
name="dodaj_do_ulubionych",
|
name="dodaj_do_ulubionych",
|
||||||
description="Dodaje do listy ulubionych w radiu",
|
description="Dodaje do listy ulubionych w radiu",
|
||||||
@@ -1827,7 +1871,11 @@ async def dodaj_do_ulubionych(ctx):
|
|||||||
json=jrequest,
|
json=jrequest,
|
||||||
timeout=360,
|
timeout=360,
|
||||||
)
|
)
|
||||||
_ = await coroutine
|
result = await coroutine
|
||||||
|
logger.info("Done %s", result)
|
||||||
|
await ctx.send("Zrobione szefie!")
|
||||||
|
|
||||||
|
|
||||||
@client.hybrid_command(
|
@client.hybrid_command(
|
||||||
name="ja_chciol",
|
name="ja_chciol",
|
||||||
description="Dodaje do listy ulubionych w radiu",
|
description="Dodaje do listy ulubionych w radiu",
|
||||||
@@ -1857,7 +1905,47 @@ async def request_radio(ctx):
|
|||||||
json=jrequest,
|
json=jrequest,
|
||||||
timeout=360,
|
timeout=360,
|
||||||
)
|
)
|
||||||
_ = await coroutine
|
result = await coroutine
|
||||||
|
logger.info("Done %s", result)
|
||||||
|
await ctx.send("Zrobione szefie!")
|
||||||
|
|
||||||
|
|
||||||
|
@client.hybrid_command(
|
||||||
|
name="stworz_audycje",
|
||||||
|
description="Dodaje do listy ulubionych w radiu",
|
||||||
|
guild=discord.Object(id=664789470779932693),
|
||||||
|
)
|
||||||
|
async def stworz_audycje(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}{CREATE_PRIO_PLAYLIST}",
|
||||||
|
json=jrequest,
|
||||||
|
timeout=360,
|
||||||
|
)
|
||||||
|
result = await coroutine
|
||||||
|
logger.info("Done %s", result)
|
||||||
|
await ctx.send("Zrobione szefie!")
|
||||||
|
|
||||||
|
|
||||||
@client.hybrid_command(
|
@client.hybrid_command(
|
||||||
name="wyczysc_ulubione",
|
name="wyczysc_ulubione",
|
||||||
description="Czysci liste ulubionych w radiu",
|
description="Czysci liste ulubionych w radiu",
|
||||||
@@ -1879,7 +1967,9 @@ async def wyczysc_ulubione(ctx):
|
|||||||
f"{FILE_SERVICE_ADDRESS}{CLEAR_PRIO}",
|
f"{FILE_SERVICE_ADDRESS}{CLEAR_PRIO}",
|
||||||
timeout=360,
|
timeout=360,
|
||||||
)
|
)
|
||||||
_ = await coroutine
|
result = await coroutine
|
||||||
|
logger.info("Done %s", result)
|
||||||
|
await ctx.send("Zrobione szefie!")
|
||||||
|
|
||||||
|
|
||||||
@client.hybrid_command(
|
@client.hybrid_command(
|
||||||
@@ -2259,11 +2349,18 @@ async def wyszukaj_linki_do_dokumentow(ctx):
|
|||||||
"""
|
"""
|
||||||
query = ctx.message.content
|
query = ctx.message.content
|
||||||
query_uuid = uuid.uuid4()
|
query_uuid = uuid.uuid4()
|
||||||
#TODO: TESTING ONLY!!
|
# TODO: TESTING ONLY!!
|
||||||
#query_uuid = uuid.UUID('{12345678-1234-5678-1234-567812345678}')
|
# query_uuid = uuid.UUID('{12345678-1234-5678-1234-567812345678}')
|
||||||
ctx.message.content = ctx.message.content.replace("$wyszukaj_linki_do_dokumentow", "")
|
ctx.message.content = ctx.message.content.replace(
|
||||||
|
"$wyszukaj_linki_do_dokumentow", ""
|
||||||
|
)
|
||||||
|
|
||||||
json_query = {"UUID": str(query_uuid), "query": str(query), "page": 1, "deep_search": False}
|
json_query = {
|
||||||
|
"UUID": str(query_uuid),
|
||||||
|
"query": str(query),
|
||||||
|
"page": 1,
|
||||||
|
"deep_search": False,
|
||||||
|
}
|
||||||
coroutine = asyncio.to_thread(
|
coroutine = asyncio.to_thread(
|
||||||
requests.post,
|
requests.post,
|
||||||
f"{LIBRARIAN_SERVICE_ADDRESS}{SEND_QUERY}",
|
f"{LIBRARIAN_SERVICE_ADDRESS}{SEND_QUERY}",
|
||||||
@@ -2271,14 +2368,16 @@ async def wyszukaj_linki_do_dokumentow(ctx):
|
|||||||
timeout=360,
|
timeout=360,
|
||||||
)
|
)
|
||||||
await ctx.send(
|
await ctx.send(
|
||||||
"*Conjurer powoli notuje podane przez Ciebie. Podchodzi do sprytnie ukrytej przy barze rury od poczty pneumatycznej i" +
|
"*Conjurer powoli notuje podane przez Ciebie. Podchodzi do sprytnie ukrytej przy barze rury od poczty pneumatycznej i"
|
||||||
" wrzuca twój liścik do środka* To teraz trza poczekać kilka godzin. Biblioteka to 3/4 stacji."+
|
+ " wrzuca twój liścik do środka* To teraz trza poczekać kilka godzin. Biblioteka to 3/4 stacji."
|
||||||
" A przecież pół stacji to browar, motelik dla zabaw cielesnych i ten Bar. Więc 12/8 stacji teraz będzie ciężko pracować"
|
+ " A przecież pół stacji to browar, motelik dla zabaw cielesnych i ten Bar. Więc 12/8 stacji teraz będzie ciężko pracować"
|
||||||
)
|
)
|
||||||
query_response = await coroutine
|
query_response = await coroutine
|
||||||
if not query_response.status_code == 200:
|
if not query_response.status_code == 200:
|
||||||
await ctx.send("*Z rury wydobywa się dym. Conjurer łapie pierwszą ciecz pod ręką i pryska na rurę. Następuje drobna eksplozja i wszystko zaczyna się palić."+
|
await ctx.send(
|
||||||
" Conjurer jest skonfundowany..* Wołaj szefa - mam nieodparte wrażenie że się coś wyjebało")
|
"*Z rury wydobywa się dym. Conjurer łapie pierwszą ciecz pod ręką i pryska na rurę. Następuje drobna eksplozja i wszystko zaczyna się palić."
|
||||||
|
+ " Conjurer jest skonfundowany..* Wołaj szefa - mam nieodparte wrażenie że się coś wyjebało"
|
||||||
|
)
|
||||||
return
|
return
|
||||||
|
|
||||||
query, query_uuid, queue_size = (
|
query, query_uuid, queue_size = (
|
||||||
@@ -2292,8 +2391,11 @@ async def wyszukaj_linki_do_dokumentow(ctx):
|
|||||||
username = ctx.message.author.name
|
username = ctx.message.author.name
|
||||||
query_object = QueryControl(username, query_uuid, query, logger, ctx)
|
query_object = QueryControl(username, query_uuid, query, logger, ctx)
|
||||||
OUT_COMM_Q.put(query_object)
|
OUT_COMM_Q.put(query_object)
|
||||||
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(
|
||||||
" 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.")
|
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."
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@client.hybrid_command(
|
@client.hybrid_command(
|
||||||
name="glebokie_gardlo",
|
name="glebokie_gardlo",
|
||||||
@@ -2337,13 +2439,23 @@ async def wyszukaj_linki_do_dokumentow_deep(ctx):
|
|||||||
result, MESSAGE_TABLE = await handle_response(
|
result, MESSAGE_TABLE = await handle_response(
|
||||||
prompt, vykidailo, bartender, MESSAGE_TABLE, username, "CONVERSATION"
|
prompt, vykidailo, bartender, MESSAGE_TABLE, username, "CONVERSATION"
|
||||||
)
|
)
|
||||||
await ctx.send(result)
|
if len(result) < 1500:
|
||||||
|
await ctx.send(result)
|
||||||
|
else:
|
||||||
|
while len(result) > 1500:
|
||||||
|
await ctx.send(result[:1500])
|
||||||
|
result = result[1500:]
|
||||||
return
|
return
|
||||||
|
|
||||||
query_uuid = uuid.uuid4()
|
query_uuid = uuid.uuid4()
|
||||||
#TODO: TESTING ONLY!!
|
# TODO: TESTING ONLY!!
|
||||||
#query_uuid = uuid.UUID('{12345678-1234-5678-1234-567812345678}')
|
# query_uuid = uuid.UUID('{12345678-1234-5678-1234-567812345678}')
|
||||||
json_query = {"UUID": str(query_uuid), "query": str(query), "page": 1, "deep_search": True}
|
json_query = {
|
||||||
|
"UUID": str(query_uuid),
|
||||||
|
"query": str(query),
|
||||||
|
"page": 1,
|
||||||
|
"deep_search": True,
|
||||||
|
}
|
||||||
coroutine = asyncio.to_thread(
|
coroutine = asyncio.to_thread(
|
||||||
requests.post,
|
requests.post,
|
||||||
f"{LIBRARIAN_SERVICE_ADDRESS}{SEND_QUERY}",
|
f"{LIBRARIAN_SERVICE_ADDRESS}{SEND_QUERY}",
|
||||||
@@ -2351,14 +2463,16 @@ async def wyszukaj_linki_do_dokumentow_deep(ctx):
|
|||||||
timeout=360,
|
timeout=360,
|
||||||
)
|
)
|
||||||
await ctx.send(
|
await ctx.send(
|
||||||
"*Conjurer mruga okiem i zamiast drinka wyjmuje dysk usb z terminala. Podchodzi do sprytnie ukrytej przy barze rury od poczty pneumatycznej i" +
|
"*Conjurer mruga okiem i zamiast drinka wyjmuje dysk usb z terminala. Podchodzi do sprytnie ukrytej przy barze rury od poczty pneumatycznej i"
|
||||||
" wrzuca ten dysk do środk* To teraz trza poczekać kilka godzin. TO będzie głębokie wyszukanie"+
|
+ " wrzuca ten dysk do środk* To teraz trza poczekać kilka godzin. TO będzie głębokie wyszukanie"
|
||||||
" Rozgląda się dookoła i stawia przed tobą mętnego drinka o zapachu mięty, gwiezdnego pyłu i oleju silnikowego"
|
+ " Rozgląda się dookoła i stawia przed tobą mętnego drinka o zapachu mięty, gwiezdnego pyłu i oleju silnikowego"
|
||||||
)
|
)
|
||||||
query_response = await coroutine
|
query_response = await coroutine
|
||||||
if not query_response.status_code == 200:
|
if not query_response.status_code == 200:
|
||||||
await ctx.send("*Z rury wydobywa się dym. Conjurer łapie pierwszą ciecz pod ręką i pryska na rurę. Następuje drobna eksplozja i wszystko zaczyna się palić."+
|
await ctx.send(
|
||||||
" Conjurer jest skonfundowany..* Wołaj szefa - mam nieodparte wrażenie że się coś wyjebało")
|
"*Z rury wydobywa się dym. Conjurer łapie pierwszą ciecz pod ręką i pryska na rurę. Następuje drobna eksplozja i wszystko zaczyna się palić."
|
||||||
|
+ " Conjurer jest skonfundowany..* Wołaj szefa - mam nieodparte wrażenie że się coś wyjebało"
|
||||||
|
)
|
||||||
return
|
return
|
||||||
|
|
||||||
query, query_uuid, queue_size = (
|
query, query_uuid, queue_size = (
|
||||||
@@ -2372,9 +2486,10 @@ async def wyszukaj_linki_do_dokumentow_deep(ctx):
|
|||||||
username = ctx.message.author.name
|
username = ctx.message.author.name
|
||||||
query_object = QueryControl(username, query_uuid, query, logger, ctx)
|
query_object = QueryControl(username, query_uuid, query, logger, ctx)
|
||||||
OUT_COMM_Q.put(query_object)
|
OUT_COMM_Q.put(query_object)
|
||||||
await ctx.send(f"Wypij wypi {query_uuid} - to identyfikator twojego zapytania. Jesteś {queue_size} w kolejce niestety kolejka obowiazuje zawsze."+
|
await ctx.send(
|
||||||
" Zapytania obsługuje algorytm zasilany czterema chomikami zapierdalającymi w kołowrotku - więc wyniki najwcześniej za kilka godzin - ale mogą być też dni. Głębokie zabawy trwają dłużej. *Znowu mruga*")
|
f"Wypij wypi {query_uuid} - to identyfikator twojego zapytania. Jesteś {queue_size} w kolejce niestety kolejka obowiazuje zawsze."
|
||||||
|
+ " Zapytania obsługuje algorytm zasilany czterema chomikami zapierdalającymi w kołowrotku - więc wyniki najwcześniej za kilka godzin - ale mogą być też dni. Głębokie zabawy trwają dłużej. *Znowu mruga*"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@client.hybrid_command(
|
@client.hybrid_command(
|
||||||
@@ -2406,11 +2521,29 @@ async def krecimy_pornola(ctx):
|
|||||||
sciezka, files = await get_file(ctx, "Pornol", item_yt)
|
sciezka, files = await get_file(ctx, "Pornol", item_yt)
|
||||||
if files:
|
if files:
|
||||||
logger.info("Pornol udany")
|
logger.info("Pornol udany")
|
||||||
ctx.reply(f"Jest w tajnym archiwum pod adresem{sciezka})")
|
await ctx.send(f"Jest w tajnym archiwum pod adresem{sciezka})")
|
||||||
else:
|
else:
|
||||||
ctx.reply("Jak ładnie szefa poprosisz.")
|
await ctx.reply("Jak ładnie szefa poprosisz.")
|
||||||
|
|
||||||
|
@client.hybrid_command(
|
||||||
|
name="co_na_plejliscie_wariacie",
|
||||||
|
description="Wyswietl kawalki ktore sa na pocztku list radia",
|
||||||
|
guild=discord.Object(id=664789470779932693),
|
||||||
|
)
|
||||||
|
async def co_na_plejliscie_wariacie(ctx):
|
||||||
|
"""
|
||||||
|
Download a music number from youtube.
|
||||||
|
|
||||||
|
: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. It allows the command to interact
|
||||||
|
with the Discord API and
|
||||||
|
"""
|
||||||
|
await ctx.send("Dej mnie chwilkę")
|
||||||
|
async with ctx.typing():
|
||||||
|
logger.info("plejlista")
|
||||||
|
logger.info(PREPPED_TRACKS)
|
||||||
|
await ctx.send(f"Didżej Hammer i jego sztuczna inteligencja (||he he||) prezentują: \n- obecnie grany jest {(PREPPED_TRACKS['now_playing']).rstrip()} \n- na liście priorytetowej {(PREPPED_TRACKS['priority']).rstrip()} \n- na liście hitów obecnie czeka kawałek {(PREPPED_TRACKS['hit']).rstrip()} \n- na liście wszystkich kawałków {(PREPPED_TRACKS['all']).rstrip()} \n- najświeższym zamówieniem od słuchaczy jest {(PREPPED_TRACKS['requests']).rstrip()} \nZ kolei jingiel to: {(PREPPED_TRACKS['jingles']).rstrip()}")
|
||||||
|
|
||||||
# *================================== Run
|
# *================================== Run
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
@@ -2418,23 +2551,16 @@ if __name__ == "__main__":
|
|||||||
threads = []
|
threads = []
|
||||||
logger.info("Starting discord bot: Creating threads")
|
logger.info("Starting discord bot: Creating threads")
|
||||||
threads.append(threading.Thread(target=client.run, args=(TOKEN,)))
|
threads.append(threading.Thread(target=client.run, args=(TOKEN,)))
|
||||||
threads.append(threading.Thread(target=comm_subroutine, args=(logger,)))
|
threads.append(threading.Thread(target=comm_subroutine))
|
||||||
logger.info("Starting discord bot: Starting threads")
|
logger.info("Starting discord bot: Starting threads")
|
||||||
WRK_CNT = 0
|
WRK_CNT = 0
|
||||||
for worker in threads:
|
for worker in threads:
|
||||||
WRK_CNT+=1
|
WRK_CNT += 1
|
||||||
logger.info("Starting discord bot: Starting thread %s", WRK_CNT)
|
logger.info("Starting discord bot: Starting thread %s", WRK_CNT)
|
||||||
worker.start()
|
worker.start()
|
||||||
logger.info("Starting discord bot: Joining threads")
|
logger.info("Starting discord bot: Joining threads")
|
||||||
WRK_CNT=0
|
WRK_CNT = 0
|
||||||
for worker in threads:
|
for worker in threads:
|
||||||
WRK_CNT+=1
|
WRK_CNT += 1
|
||||||
logger.info("Starting discord bot: Joining thread %s", WRK_CNT)
|
logger.info("Starting discord bot: Joining thread %s", WRK_CNT)
|
||||||
worker.join()
|
worker.join()
|
||||||
|
|
||||||
# widok - generuje się 5 stron po 4 linki
|
|
||||||
# odśwież - zamienić na "dalej" - zaciąga kolejne 20 rezultatów
|
|
||||||
# po wyslaniu czekamy na odbior i dodajemy UUID oraz nick poszukiwacza do queue ktora jest sprawdzana
|
|
||||||
# flask sobie chodzi i jak dostaje posta z wynikami to wyciaga z queue UUID i odpowiada że znalazła
|
|
||||||
# po samym wyszukaniu dajemy info że "Zaczynam szukać ale pare godzin to potrwa."
|
|
||||||
# OPCJONALNIE: Dosyłamy znalezione wyniki na bieżąco - wtedy wkładamy do queue dla UUID wyniki
|
|
||||||
|
|||||||
+51
-22
@@ -13,16 +13,26 @@ IN_COMM_Q = Queue()
|
|||||||
awaiting_q = []
|
awaiting_q = []
|
||||||
incoming_q = Queue()
|
incoming_q = Queue()
|
||||||
app = Flask(__name__)
|
app = Flask(__name__)
|
||||||
|
PREPPED_TRACKS = {
|
||||||
|
"requests" : "",
|
||||||
|
"hit" : "",
|
||||||
|
"all": "",
|
||||||
|
"priority":"",
|
||||||
|
"jingles":"",
|
||||||
|
"now_playing":""
|
||||||
|
}
|
||||||
|
logger = logging.getLogger("discord")
|
||||||
|
|
||||||
class QueryControl:
|
class QueryControl:
|
||||||
"""
|
"""
|
||||||
This class `QueryControl` is used to manage queries with information about the author, UUID,
|
This class `QueryControl` is used to manage queries with information about the author, UUID,
|
||||||
content, logger, context, and replies.
|
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, ctx) -> None:
|
||||||
self.author = query_author
|
self.author = query_author
|
||||||
self.uuid = query_uuid
|
self.uuid = query_uuid
|
||||||
self.content = query_content
|
self.content = query_content
|
||||||
self.logger = uplogger
|
self.logger = logging.getLogger("discord")
|
||||||
self.stop = False
|
self.stop = False
|
||||||
self.ctx = ctx
|
self.ctx = ctx
|
||||||
self.logger.info(
|
self.logger.info(
|
||||||
@@ -30,6 +40,18 @@ class QueryControl:
|
|||||||
)
|
)
|
||||||
self.replies = []
|
self.replies = []
|
||||||
|
|
||||||
|
@app.route("/prepped_tracks", methods=["POST"])
|
||||||
|
def log_radio_tracks():
|
||||||
|
logger = logging.getLogger("discord")
|
||||||
|
|
||||||
|
logger.info(request)
|
||||||
|
record = json.loads(request.data)
|
||||||
|
logger.info(record)
|
||||||
|
PREPPED_TRACKS[record[0]] = record[1]
|
||||||
|
logger.info("DATA RECEIVED")
|
||||||
|
return jsonify("SUCCESS")
|
||||||
|
|
||||||
|
|
||||||
@app.route("/conjurer", methods=["POST"])
|
@app.route("/conjurer", methods=["POST"])
|
||||||
def answer_external_command():
|
def answer_external_command():
|
||||||
"""
|
"""
|
||||||
@@ -38,10 +60,11 @@ def answer_external_command():
|
|||||||
:return: The function `answer_external_command()` is returning a JSON response with the message
|
:return: The function `answer_external_command()` is returning a JSON response with the message
|
||||||
"SUCCESS".
|
"SUCCESS".
|
||||||
"""
|
"""
|
||||||
app.logger.info(request)
|
logger = logging.getLogger("discord")
|
||||||
|
logger.info(request)
|
||||||
record = json.loads(request.data)
|
record = json.loads(request.data)
|
||||||
app.logger.info(record)
|
logger.info(record)
|
||||||
app.logger.info("DATA RECEIVED")
|
logger.info("DATA RECEIVED")
|
||||||
incoming_q.put(record)
|
incoming_q.put(record)
|
||||||
return jsonify("SUCCESS")
|
return jsonify("SUCCESS")
|
||||||
|
|
||||||
@@ -56,25 +79,29 @@ def check_alive():
|
|||||||
"""
|
"""
|
||||||
return jsonify("ALIVE")
|
return jsonify("ALIVE")
|
||||||
|
|
||||||
def flask_debug(_logger):
|
def flask_debug():
|
||||||
"""
|
"""
|
||||||
The `flask_debug` function starts a Flask application in debug mode without using the reloader.
|
The `flask_debug` function starts a Flask application in debug mode without using the reloader.
|
||||||
Do not use for production for fucks sake.
|
Do not use for production for fucks sake.
|
||||||
"""
|
"""
|
||||||
_logger.info("Attempt debug")
|
logger = logging.getLogger("discord")
|
||||||
|
|
||||||
|
logger.info("Attempt 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(_logger):
|
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"
|
||||||
and port 5000 using the Waitress WSGI server.
|
and port 5000 using the Waitress WSGI server.
|
||||||
"""
|
"""
|
||||||
_logger.info("Attempt waitress")
|
logger = logging.getLogger("discord")
|
||||||
|
|
||||||
|
logger.info("Attempt waitress")
|
||||||
serve(app, host=HOST_ADDRESS, port=PORT_ADDRESS)
|
serve(app, host=HOST_ADDRESS, port=PORT_ADDRESS)
|
||||||
|
|
||||||
def scan_queue(_logger):
|
def scan_queue():
|
||||||
"""
|
"""
|
||||||
The function `scan_queue` reads data from a queue, logs it, and appends it to another queue.
|
The function `scan_queue` reads data from a queue, logs it, and appends it to another queue.
|
||||||
|
|
||||||
@@ -83,12 +110,13 @@ def scan_queue(_logger):
|
|||||||
and provide information for debugging purposes. In this code snippet, the `_logger` object is used
|
and provide information for debugging purposes. In this code snippet, the `_logger` object is used
|
||||||
to log the
|
to log the
|
||||||
"""
|
"""
|
||||||
|
logger = logging.getLogger("discord")
|
||||||
while True:
|
while True:
|
||||||
data = OUT_COMM_Q.get()
|
data = OUT_COMM_Q.get()
|
||||||
_logger.info(data)
|
logger.info(data)
|
||||||
awaiting_q.append(data)
|
awaiting_q.append(data)
|
||||||
|
|
||||||
def scan_incoming(_logger):
|
def scan_incoming():
|
||||||
"""
|
"""
|
||||||
The `scan_incoming` function continuously checks for incoming data, processes it, and logs when data
|
The `scan_incoming` function continuously checks for incoming data, processes it, and logs when data
|
||||||
is found.
|
is found.
|
||||||
@@ -98,10 +126,11 @@ def scan_incoming(_logger):
|
|||||||
debugging, monitoring, or tracking the flow of the program. In this case, the `_logger` is being
|
debugging, monitoring, or tracking the flow of the program. In this case, the `_logger` is being
|
||||||
used
|
used
|
||||||
"""
|
"""
|
||||||
|
logger = logging.getLogger("discord")
|
||||||
while True:
|
while True:
|
||||||
try:
|
try:
|
||||||
answer = incoming_q.get(block=False)
|
answer = incoming_q.get(block=False)
|
||||||
_logger.info("DATA FOUND")
|
logger.info("DATA FOUND")
|
||||||
record_stored = False
|
record_stored = False
|
||||||
for record in awaiting_q:
|
for record in awaiting_q:
|
||||||
if record.uuid in answer.keys():
|
if record.uuid in answer.keys():
|
||||||
@@ -111,7 +140,7 @@ def scan_incoming(_logger):
|
|||||||
IN_COMM_Q.put(record)
|
IN_COMM_Q.put(record)
|
||||||
if not record_stored:
|
if not record_stored:
|
||||||
for key in answer.keys():
|
for key in answer.keys():
|
||||||
record = QueryControl("Orphaned", key, "Orphan", _logger, None)
|
record = QueryControl("Orphaned", key, "Orphan", None)
|
||||||
record.stop = True
|
record.stop = True
|
||||||
record.entries = answer[record.uuid]
|
record.entries = answer[record.uuid]
|
||||||
IN_COMM_Q.put(record)
|
IN_COMM_Q.put(record)
|
||||||
@@ -119,7 +148,7 @@ def scan_incoming(_logger):
|
|||||||
time.sleep(1)
|
time.sleep(1)
|
||||||
|
|
||||||
|
|
||||||
def comm_subroutine(logger):
|
def comm_subroutine():
|
||||||
"""
|
"""
|
||||||
The `comm_subroutine` function starts multiple threads to run different tasks concurrently.
|
The `comm_subroutine` function starts multiple threads to run different tasks concurrently.
|
||||||
|
|
||||||
@@ -128,12 +157,13 @@ def comm_subroutine(logger):
|
|||||||
the provided code snippet, the logger is used to log messages at the "info" level
|
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 = logging.getLogger("discord")
|
||||||
|
logger.info("Started comms")
|
||||||
threads = []
|
threads = []
|
||||||
#threads.append(threading.Thread(target=flask_debug, args=(logger,)))
|
#threads.append(threading.Thread(target=flask_debug))
|
||||||
threads.append(threading.Thread(target=waitress_run, args=(logger,)))
|
threads.append(threading.Thread(target=waitress_run))
|
||||||
threads.append(threading.Thread(target=scan_queue,args=(logger,)))
|
threads.append(threading.Thread(target=scan_queue))
|
||||||
threads.append(threading.Thread(target=scan_incoming,args=(logger,)))
|
threads.append(threading.Thread(target=scan_incoming))
|
||||||
|
|
||||||
|
|
||||||
for worker in threads:
|
for worker in threads:
|
||||||
@@ -142,5 +172,4 @@ def comm_subroutine(logger):
|
|||||||
worker.join()
|
worker.join()
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
logging_client = logging.getLogger(__name__)
|
comm_subroutine()
|
||||||
comm_subroutine(logging_client)
|
|
||||||
|
|||||||
@@ -0,0 +1,8 @@
|
|||||||
|
|
||||||
|
FILEPTH = r"C:\Database\chunks\4_chunk.txt"
|
||||||
|
|
||||||
|
with open(FILEPTH, "r", encoding="utf-8") as file:
|
||||||
|
tab = file.readlines()
|
||||||
|
print(len(tab))
|
||||||
|
|
||||||
|
#1842295
|
||||||
@@ -23,6 +23,7 @@ from queue import Empty, Queue
|
|||||||
from threading import Thread
|
from threading import Thread
|
||||||
import time
|
import time
|
||||||
q = Queue()
|
q = Queue()
|
||||||
|
#TODO: Count number of lines in files and print to approximate on which part of the file search is
|
||||||
|
|
||||||
# DATA FOR TEST ONLY
|
# DATA FOR TEST ONLY
|
||||||
# MAXTHREADS = 5
|
# MAXTHREADS = 5
|
||||||
|
|||||||
@@ -6,7 +6,9 @@ API endpoints.
|
|||||||
"""
|
"""
|
||||||
import json
|
import json
|
||||||
import logging
|
import logging
|
||||||
|
import os
|
||||||
import re
|
import re
|
||||||
|
import requests
|
||||||
import threading
|
import threading
|
||||||
import time
|
import time
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
@@ -17,6 +19,8 @@ from sys import platform
|
|||||||
from flask import Flask, jsonify, redirect, request, send_from_directory, render_template
|
from flask import Flask, jsonify, redirect, request, send_from_directory, render_template
|
||||||
from waitress import serve
|
from waitress import serve
|
||||||
|
|
||||||
|
MAIN_BOT_ADDRESS = "http://192.168.1.191:5000"
|
||||||
|
MUSIC_TRACKER = "/prepped_tracks"
|
||||||
HOST_ADDRESS = "192.168.1.15"
|
HOST_ADDRESS = "192.168.1.15"
|
||||||
PORT_ADDRESS = 5000
|
PORT_ADDRESS = 5000
|
||||||
if platform in ("linux", "linux2"):
|
if platform in ("linux", "linux2"):
|
||||||
@@ -27,6 +31,8 @@ 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"
|
||||||
|
RADIOLOG_PATH = '/home/pi/Conjurer/radio_log.log'
|
||||||
|
PERSISTENCE_PATH = '/home/pi/Conjurer/persistence.log'
|
||||||
|
|
||||||
else:
|
else:
|
||||||
LOGFILE = "/home/pi/Conjurer/discord_mus_service.log"
|
LOGFILE = "/home/pi/Conjurer/discord_mus_service.log"
|
||||||
@@ -35,6 +41,8 @@ if platform in ("linux", "linux2"):
|
|||||||
ENCODING = "utf-8"
|
ENCODING = "utf-8"
|
||||||
MUSIC_FOLDER = "/home/pi/RetroPie/mp3/"
|
MUSIC_FOLDER = "/home/pi/RetroPie/mp3/"
|
||||||
PRIORITY_FOLDER = "/home/pi/RetroPie/mp3/Magiczne i chuj/"
|
PRIORITY_FOLDER = "/home/pi/RetroPie/mp3/Magiczne i chuj/"
|
||||||
|
RADIOLOG_PATH = '/home/pi/Conjurer/radio_log.log'
|
||||||
|
PERSISTENCE_PATH = '/home/pi/Conjurer/persistence.log'
|
||||||
|
|
||||||
|
|
||||||
music_file_list = []
|
music_file_list = []
|
||||||
@@ -45,7 +53,8 @@ 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,
|
||||||
and adds them to a list of music files.
|
and adds them to a list of music files.
|
||||||
"""
|
"""
|
||||||
logging.info("Rescan triggered")
|
logger = logging.getLogger("conjurer_musician")
|
||||||
|
logger.info("Rescan triggered")
|
||||||
|
|
||||||
for mp3_item in Path.glob(Path(MUSIC_FOLDER), "**/*.mp3"):
|
for mp3_item in Path.glob(Path(MUSIC_FOLDER), "**/*.mp3"):
|
||||||
temp_music_file = mp3_item.as_posix()
|
temp_music_file = mp3_item.as_posix()
|
||||||
@@ -82,13 +91,87 @@ def thread_rescan():
|
|||||||
The `thread_rescan` function periodically triggers a rescan operation after a specified time
|
The `thread_rescan` function periodically triggers a rescan operation after a specified time
|
||||||
interval.
|
interval.
|
||||||
"""
|
"""
|
||||||
logging.info("Starting filesystemupdater")
|
logger = logging.getLogger("conjurer_musician")
|
||||||
|
logger.info("Starting filesystemupdater")
|
||||||
while True:
|
while True:
|
||||||
time.sleep(60 * 60 * 24)
|
time.sleep(60 * 60 * 24)
|
||||||
logging.info("Rescan triggered")
|
logger.info("Rescan triggered")
|
||||||
rescan()
|
rescan()
|
||||||
|
|
||||||
|
|
||||||
|
def scan_tracks():
|
||||||
|
#Set the filename and open the file
|
||||||
|
logger = logging.getLogger("conjurer_musician")
|
||||||
|
|
||||||
|
file = open(RADIOLOG_PATH,'r')
|
||||||
|
#Find the size of the file and move to the end
|
||||||
|
st_results = os.stat(RADIOLOG_PATH)
|
||||||
|
st_size = st_results[6]
|
||||||
|
file.seek(st_size)
|
||||||
|
st_results1 = os.stat(PERSISTENCE_PATH)
|
||||||
|
prev_st_size1 = st_results[6]
|
||||||
|
|
||||||
|
while 1:
|
||||||
|
|
||||||
|
st_results1 = os.stat(PERSISTENCE_PATH)
|
||||||
|
st_size1 = st_results1[6]
|
||||||
|
if prev_st_size1 != st_size1:
|
||||||
|
while prev_st_size1 != st_size1:
|
||||||
|
prev_st_size1 = st_size1
|
||||||
|
st_results1 = os.stat(PERSISTENCE_PATH)
|
||||||
|
st_size1 = st_results1[6]
|
||||||
|
time.sleep(0.1)
|
||||||
|
file1 = open(PERSISTENCE_PATH, 'r')
|
||||||
|
lines = file1.readlines()
|
||||||
|
result = ["now_playing", lines[2]]
|
||||||
|
file1.close()
|
||||||
|
returned = requests.post(
|
||||||
|
f"{MAIN_BOT_ADDRESS}{MUSIC_TRACKER}",
|
||||||
|
json=result,
|
||||||
|
timeout=360)
|
||||||
|
logger.info("SENT")
|
||||||
|
logger.info(returned.status_code)
|
||||||
|
logger.info("SEND CONFIRMED")
|
||||||
|
|
||||||
|
where = file.tell()
|
||||||
|
line = file.readline()
|
||||||
|
if not line:
|
||||||
|
time.sleep(1)
|
||||||
|
file.seek(where)
|
||||||
|
else:
|
||||||
|
if re.match(".*Prepared.*",line):
|
||||||
|
result = None
|
||||||
|
if re.match(".*jingles.*", line):
|
||||||
|
logger.info("jingles")
|
||||||
|
logger.info(line) # already has newline
|
||||||
|
result = ["jingles", line]
|
||||||
|
elif re.match(".*priority.*", line):
|
||||||
|
logger.info("priority")
|
||||||
|
logger.info(line) # already has newline
|
||||||
|
result = ["priority", line]
|
||||||
|
elif re.match(".*hit.*", line):
|
||||||
|
logger.info("hit")
|
||||||
|
logger.info(line) # already has newline
|
||||||
|
result = ["hit", line]
|
||||||
|
elif re.match(".*all_playlist.*", line):
|
||||||
|
logger.info("all")
|
||||||
|
logger.info(line) # already has newline
|
||||||
|
result = ["all", line]
|
||||||
|
elif re.match(".*request.*", line):
|
||||||
|
logger.info("requests")
|
||||||
|
logger.info(line) # already has newline
|
||||||
|
result = ["requests", line]
|
||||||
|
if result:
|
||||||
|
returned = requests.post(
|
||||||
|
f"{MAIN_BOT_ADDRESS}{MUSIC_TRACKER}",
|
||||||
|
json=result,
|
||||||
|
timeout=360)
|
||||||
|
logger.info("SENT")
|
||||||
|
logger.info(returned.status_code)
|
||||||
|
logger.info("SEND CONFIRMED")
|
||||||
|
time.sleep(0.1)
|
||||||
|
|
||||||
|
|
||||||
app = Flask(__name__)
|
app = Flask(__name__)
|
||||||
|
|
||||||
|
|
||||||
@@ -206,6 +289,7 @@ def wyszukaj(word_list, how_many, _logger=None, return_to_bot=True):
|
|||||||
fun_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)])
|
||||||
|
fun_logger.info("Done: %s", return_list)
|
||||||
return return_list
|
return return_list
|
||||||
|
|
||||||
|
|
||||||
@@ -369,6 +453,37 @@ def add_request():
|
|||||||
)
|
)
|
||||||
return return_data
|
return return_data
|
||||||
|
|
||||||
|
@app.route("/create_priority_playlist", methods=["POST"])
|
||||||
|
def create_priority_playlist():
|
||||||
|
"""
|
||||||
|
The function `create_priority_playlist` 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: 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
|
||||||
|
)
|
||||||
|
with open(
|
||||||
|
"/home/pi/Conjurer/request.playlist", "a", encoding="utf-8"
|
||||||
|
) as s_file:
|
||||||
|
for item in return_data:
|
||||||
|
s_file.write(item[1] + "\n")
|
||||||
|
return_data = (
|
||||||
|
jsonify(isError=False, message="Success", statusCode=200, data={"status": "OK"}),
|
||||||
|
200,
|
||||||
|
)
|
||||||
|
return return_data
|
||||||
|
|
||||||
|
|
||||||
@app.route("/add_to_priority", methods=["POST"])
|
@app.route("/add_to_priority", methods=["POST"])
|
||||||
@@ -453,6 +568,7 @@ if __name__ == "__main__":
|
|||||||
# threads.append(threading.Thread(target=flask_debug))
|
# 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))
|
||||||
|
threads.append(threading.Thread(target=scan_tracks))
|
||||||
|
|
||||||
for worker in threads:
|
for worker in threads:
|
||||||
worker.start()
|
worker.start()
|
||||||
|
|||||||
@@ -0,0 +1,14 @@
|
|||||||
|
$dodaj_do_ulubionych 100 nightwish
|
||||||
|
$dodaj_do_ulubionych 100 amon amarth
|
||||||
|
$dodaj_do_ulubionych 100 tactical sekt
|
||||||
|
$dodaj_do_ulubionych 100 lacuna coil
|
||||||
|
$dodaj_do_ulubionych 100 sefa
|
||||||
|
$dodaj_do_ulubionych 100 frenchcore
|
||||||
|
$dodaj_do_ulubionych 100 peacock
|
||||||
|
$dodaj_do_ulubionych 100 vavamuffin
|
||||||
|
$dodaj_do_ulubionych 100 abradab
|
||||||
|
$dodaj_do_ulubionych 100 eluveitie
|
||||||
|
$dodaj_do_ulubionych 100 youtube
|
||||||
|
|
||||||
|
$dodaj_do_ulubionych 100 elyose
|
||||||
|
$dodaj_do_ulubionych 1 artisans du chaos
|
||||||
@@ -34,10 +34,9 @@ requests_queue = request.queue()
|
|||||||
|
|
||||||
# Function to process the request queue and add new requests
|
# Function to process the request queue and add new requests
|
||||||
def queue_processing()
|
def queue_processing()
|
||||||
f1=file.read("/home/pi/Conjurer/request.playlist")
|
text=file.lines("/home/pi/Conjurer/request.playlist")
|
||||||
text = f1()
|
if text != [] then
|
||||||
if text != "" then
|
list.iter(fun(item) -> requests_queue.push.uri(item), text)
|
||||||
requests_queue.push.uri(text)
|
|
||||||
file.remove("/home/pi/Conjurer/request.playlist")
|
file.remove("/home/pi/Conjurer/request.playlist")
|
||||||
f = file.open("/home/pi/Conjurer/request.playlist", create=true)
|
f = file.open("/home/pi/Conjurer/request.playlist", create=true)
|
||||||
f.close()
|
f.close()
|
||||||
@@ -45,7 +44,7 @@ def queue_processing()
|
|||||||
end
|
end
|
||||||
|
|
||||||
# Randomly select a playlist with weights
|
# Randomly select a playlist with weights
|
||||||
s4 = random(id="randomizer", weights=[1, 3, 5], [s1, s2, s3])
|
s4 = random(id="randomizer", weights=[2, 3, 5], [s1, s2, s3])
|
||||||
|
|
||||||
# Load jingles playlist
|
# Load jingles playlist
|
||||||
jingles = (playlist(reload_mode="watch", "/home/pi/Conjurer/jingles.playlist"))
|
jingles = (playlist(reload_mode="watch", "/home/pi/Conjurer/jingles.playlist"))
|
||||||
@@ -74,7 +73,7 @@ s = amplify(a, s)
|
|||||||
|
|
||||||
mic = buffer(input.pulseaudio()) # Microphone
|
mic = buffer(input.pulseaudio()) # Microphone
|
||||||
mic = gate(threshold=-30., range=-80., mic)
|
mic = gate(threshold=-30., range=-80., mic)
|
||||||
mic = blank.strip(max_blank=2., min_noise=.1, threshold=-20., mic)
|
mic = blank.strip(max_blank=10., min_noise=.1, threshold=-20., mic)
|
||||||
|
|
||||||
s = add([s, mic])
|
s = add([s, mic])
|
||||||
#w = interactive.float("wetness", min=0., max=1., 1.)
|
#w = interactive.float("wetness", min=0., max=1., 1.)
|
||||||
@@ -82,7 +81,7 @@ s = add([s, mic])
|
|||||||
|
|
||||||
|
|
||||||
# Skip blank sections in the stream
|
# Skip blank sections in the stream
|
||||||
s = blank.skip(max_blank=2., s)
|
s = blank.skip(max_blank=10., s)
|
||||||
|
|
||||||
# Configure logging settings
|
# Configure logging settings
|
||||||
log_to_stdout = true
|
log_to_stdout = true
|
||||||
@@ -102,11 +101,11 @@ radio = fallback(id="switcher2", track_sensitive=false, [s, emergency])
|
|||||||
# Set up an interactive control for skipping tracks
|
# Set up an interactive control for skipping tracks
|
||||||
p = interactive.bool("Skip track", false)
|
p = interactive.bool("Skip track", false)
|
||||||
|
|
||||||
def http_skip(_):
|
def http_skip(~protocol, ~data, ~headers, uri)=
|
||||||
radio.skip()
|
radio.skip()
|
||||||
http.response(data="Skipped")
|
http.response(code=200,data="Skipped")
|
||||||
end
|
end
|
||||||
harbor.http.register.simple(port=54321, "/skip", http_skip)
|
harbor.http.register(port=54321,method="GET","/skip", http_skip)
|
||||||
|
|
||||||
|
|
||||||
# Function to process the request queue and skip the current track if requested
|
# Function to process the request queue and skip the current track if requested
|
||||||
|
|||||||
Reference in New Issue
Block a user