From f05ae2b006afd7c55378d2b9df567c7ecdddb8f4 Mon Sep 17 00:00:00 2001 From: Migatu Date: Wed, 19 Jul 2023 14:18:17 +0200 Subject: [PATCH] Improvements --- bot.py | 35 ++++++++++++++++++++++++++--------- 1 file changed, 26 insertions(+), 9 deletions(-) diff --git a/bot.py b/bot.py index d639896..0ba433c 100644 --- a/bot.py +++ b/bot.py @@ -1535,6 +1535,7 @@ async def get_image_stable_diffusion(ctx): guild=discord.Object(id=664789470779932693), ) # trunk-ignore(pylint/R0912) +# trunk-ignore(pylint/R0915) async def zagraj_muzyke_mego_ludu(ctx): """ Play completeley random playlist based on messaging history. @@ -1546,17 +1547,22 @@ async def zagraj_muzyke_mego_ludu(ctx): """ stats = {} if ctx: + logger.info("Zagraj muzyke mego ludu: wywolane") stat_iter = get_stats(ctx, MUZYKA_MOJEGO_LUDU_HISTORIA) async for name, how_many in stat_iter: if len(name) > 3: stats[name] = how_many sorted_stats = sorted(stats.items(), key=lambda x: x[1]) + logger.info("Zagraj muzyke mego ludu:zebrano statystyki") key_words = [] for stat in sorted_stats: key_words.append(stat[0]) - logger.info(sorted_stats[:30]) - logger.info(key_words[:30]) + logger.info(sorted_stats[:MUZYKA_MOJEGO_LUDU_SLOWA_KLUCZOWE]) + logger.info(key_words[:MUZYKA_MOJEGO_LUDU_SLOWA_KLUCZOWE]) + logger.info(sorted_stats[-MUZYKA_MOJEGO_LUDU_SLOWA_KLUCZOWE:]) + logger.info(key_words[-MUZYKA_MOJEGO_LUDU_SLOWA_KLUCZOWE:]) + final_key_words = key_words[:MUZYKA_MOJEGO_LUDU_SLOWA_KLUCZOWE] final_key_words.extend(key_words[-MUZYKA_MOJEGO_LUDU_SLOWA_KLUCZOWE:]) reply = "Wygenerowana plejlista:\n" @@ -1564,12 +1570,13 @@ async def zagraj_muzyke_mego_ludu(ctx): logger.info("Zaczynam szukać timestamp %s", datetime.now()) search_time_glob = datetime.now() file = await wyszukaj(ctx=ctx, how_many=MUZYKA_MOJEGO_LUDU_PLAJLISTA, slowa_kluczowe=final_key_words) + logger.info("Zagraj muzyke mego ludu:wygenerowano plejliste") + logger.info(file) logger.info( "Koniec szukania(timestamp %s zajęło %s", datetime.now(), datetime.now() - search_time_glob, ) - logger.info(file) index = 1 if file: for plik in file: @@ -1629,12 +1636,22 @@ async def parametry_muzyki_mego_ludu(ctx, ile_historii=1500, ile_slow_kluczowych if ctx: async with ctx.typing(): - global MUZYKA_MOJEGO_LUDU_HISTORIA # pylint: disable=global-statement - MUZYKA_MOJEGO_LUDU_HISTORIA = ile_historii - global MUZYKA_MOJEGO_LUDU_SLOWA_KLUCZOWE # pylint: disable=global-statement - MUZYKA_MOJEGO_LUDU_SLOWA_KLUCZOWE = ile_slow_kluczowych - global MUZYKA_MOJEGO_LUDU_PLAJLISTA # pylint: disable=global-statement - MUZYKA_MOJEGO_LUDU_PLAJLISTA = jak_dluga_plejlista + if ':' in ile_historii: + _, _, ile_historii = ile_historii.partition(':') + if ':' in ile_slow_kluczowych: + _, _, ile_slow_kluczowych = ile_slow_kluczowych.partition(':') + if ':' in jak_dluga_plejlista: + _, _, jak_dluga_plejlista = jak_dluga_plejlista.partition(':') + try: + global MUZYKA_MOJEGO_LUDU_HISTORIA # pylint: disable=global-statement + MUZYKA_MOJEGO_LUDU_HISTORIA = ile_historii + global MUZYKA_MOJEGO_LUDU_SLOWA_KLUCZOWE # pylint: disable=global-statement + MUZYKA_MOJEGO_LUDU_SLOWA_KLUCZOWE = ile_slow_kluczowych + global MUZYKA_MOJEGO_LUDU_PLAJLISTA # pylint: disable=global-statement + MUZYKA_MOJEGO_LUDU_PLAJLISTA = jak_dluga_plejlista + except discord.ext.commands.errors.BadArgument as exce: + ctx.send("Spierdalaj") + logger.info(exce) # *================================== Run