mirror of
https://github.com/Alexander-D-Karpov/scripts.git
synced 2024-11-22 03:56:32 +03:00
added podcasts docker, minor fixes
This commit is contained in:
parent
0396771e15
commit
0a5e849ce7
20
podcasts/yandex/Dockerfile
Normal file
20
podcasts/yandex/Dockerfile
Normal file
|
|
@ -0,0 +1,20 @@
|
||||||
|
ARG PYTHON_VERSION=3.11-slim
|
||||||
|
FROM python:${PYTHON_VERSION} as python
|
||||||
|
|
||||||
|
|
||||||
|
FROM python as python-build-stage
|
||||||
|
|
||||||
|
ARG BUILD_ENVIRONMENT=local
|
||||||
|
RUN mkdir "/app"
|
||||||
|
WORKDIR /app
|
||||||
|
ARG BUILD_ENVIRONMENT=local
|
||||||
|
RUN apt-get update && apt-get install -y ffmpeg && \
|
||||||
|
apt-get purge -y --auto-remove -o APT:AutoRemove:RecommendsImportant=false && \
|
||||||
|
rm -rf /var/lib/apt/lists/*
|
||||||
|
|
||||||
|
ADD requirements.txt requirements.txt
|
||||||
|
ADD main.py main.py
|
||||||
|
|
||||||
|
RUN pip install -r requirements.txt
|
||||||
|
|
||||||
|
CMD ["python", "main.py"]
|
||||||
7
podcasts/yandex/compose.yaml
Normal file
7
podcasts/yandex/compose.yaml
Normal file
|
|
@ -0,0 +1,7 @@
|
||||||
|
version: "3"
|
||||||
|
|
||||||
|
services:
|
||||||
|
podcasts:
|
||||||
|
build: .
|
||||||
|
env_file:
|
||||||
|
- .env
|
||||||
105
podcasts/yandex/main.py
Normal file
105
podcasts/yandex/main.py
Normal file
|
|
@ -0,0 +1,105 @@
|
||||||
|
import asyncio
|
||||||
|
import os
|
||||||
|
import tempfile
|
||||||
|
from io import BytesIO
|
||||||
|
|
||||||
|
from time import sleep
|
||||||
|
|
||||||
|
from aiogram import Bot
|
||||||
|
from aiogram.bot.api import TelegramAPIServer
|
||||||
|
from mutagen.easyid3 import EasyID3
|
||||||
|
from mutagen.mp3 import MP3
|
||||||
|
from mutagen.id3 import APIC, ID3, TORY
|
||||||
|
from pydub import AudioSegment
|
||||||
|
from yandex_music import Client, Track
|
||||||
|
from dotenv import load_dotenv
|
||||||
|
|
||||||
|
load_dotenv(dotenv_path=".env")
|
||||||
|
|
||||||
|
YANDEX_TOKEN = os.getenv("YANDEX_TOKEN")
|
||||||
|
CHAT_ID = os.getenv("CHAT_ID")
|
||||||
|
TOKEN = os.getenv("BOT_TOKEN")
|
||||||
|
TELEGRAM_SERVER = os.getenv("TELEGRAM_SERVER", default=None)
|
||||||
|
|
||||||
|
if TELEGRAM_SERVER:
|
||||||
|
local_server = TelegramAPIServer.from_base(TELEGRAM_SERVER)
|
||||||
|
bot = Bot(TOKEN, server=local_server)
|
||||||
|
else:
|
||||||
|
bot = Bot(TOKEN)
|
||||||
|
|
||||||
|
|
||||||
|
client = Client(YANDEX_TOKEN).init()
|
||||||
|
latest_podcast = None
|
||||||
|
latest_sent = True
|
||||||
|
podcasts_listened = []
|
||||||
|
|
||||||
|
while True:
|
||||||
|
try:
|
||||||
|
queues = client.queues_list()
|
||||||
|
last_queue = client.queue(queues[0].id)
|
||||||
|
|
||||||
|
last_track_id = last_queue.get_current_track()
|
||||||
|
last_track: Track = last_track_id.fetch_track()
|
||||||
|
|
||||||
|
if "podcast" in last_track.type:
|
||||||
|
if last_track_id not in podcasts_listened:
|
||||||
|
if last_track_id == latest_podcast and not latest_sent:
|
||||||
|
latest_sent = True
|
||||||
|
podcasts_listened.append(last_track_id)
|
||||||
|
|
||||||
|
title = last_track.title
|
||||||
|
album = last_track.albums[0]
|
||||||
|
url = f"https://music.yandex.ru/track/{last_track.id}"
|
||||||
|
desc = last_track.short_description.split("\n")[0]
|
||||||
|
with tempfile.TemporaryDirectory() as temp:
|
||||||
|
last_track.download_cover(filename=temp + "cover.png")
|
||||||
|
img_path = os.path.abspath(temp + "cover.png")
|
||||||
|
|
||||||
|
last_track.download(filename=temp + "file", codec="mp3")
|
||||||
|
orig_path = os.path.abspath(temp + "file")
|
||||||
|
path = os.path.abspath(temp + "file.mp3")
|
||||||
|
|
||||||
|
AudioSegment.from_file(orig_path).export(path)
|
||||||
|
os.remove(orig_path)
|
||||||
|
|
||||||
|
# set music meta
|
||||||
|
tag = MP3(path, ID3=ID3)
|
||||||
|
tag.tags.add(
|
||||||
|
APIC(
|
||||||
|
encoding=3, # 3 is for utf-8
|
||||||
|
mime="image/png", # image/jpeg or image/png
|
||||||
|
type=3, # 3 is for the cover image
|
||||||
|
desc="Cover",
|
||||||
|
data=open(img_path, "rb").read(),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
tag.tags.add(TORY(text=str(album.year)))
|
||||||
|
tag.save()
|
||||||
|
tag = EasyID3(path)
|
||||||
|
|
||||||
|
tag["title"] = title
|
||||||
|
tag["album"] = album.title
|
||||||
|
|
||||||
|
tag.save()
|
||||||
|
|
||||||
|
with open(path, "rb") as tmp:
|
||||||
|
obj = BytesIO(tmp.read())
|
||||||
|
obj.name = f"{title}.mp3"
|
||||||
|
loop = asyncio.get_event_loop()
|
||||||
|
coroutine = bot.send_audio(
|
||||||
|
chat_id=CHAT_ID,
|
||||||
|
audio=obj,
|
||||||
|
caption=f"{title} - {album.title}\n{desc}\n\n{url}",
|
||||||
|
title=title,
|
||||||
|
performer=album.title,
|
||||||
|
)
|
||||||
|
loop.run_until_complete(coroutine)
|
||||||
|
|
||||||
|
else:
|
||||||
|
latest_podcast = last_track_id
|
||||||
|
latest_sent = False
|
||||||
|
except BaseException as e:
|
||||||
|
loop = asyncio.get_event_loop()
|
||||||
|
coroutine = bot.send_message(868474142, text=str(e))
|
||||||
|
loop.run_until_complete(coroutine)
|
||||||
|
sleep(60)
|
||||||
|
|
@ -1,104 +0,0 @@
|
||||||
import asyncio
|
|
||||||
import os
|
|
||||||
from io import BytesIO
|
|
||||||
|
|
||||||
from time import sleep
|
|
||||||
|
|
||||||
from aiogram import Bot
|
|
||||||
from aiogram.bot.api import TelegramAPIServer
|
|
||||||
from mutagen.easyid3 import EasyID3
|
|
||||||
from mutagen.mp3 import MP3
|
|
||||||
from mutagen.id3 import APIC, ID3, TORY
|
|
||||||
from pydub import AudioSegment
|
|
||||||
from yandex_music import Client, Track
|
|
||||||
from dotenv import load_dotenv
|
|
||||||
|
|
||||||
load_dotenv(dotenv_path=".env")
|
|
||||||
|
|
||||||
YANDEX_TOKEN = os.getenv("YANDEX_TOKEN")
|
|
||||||
CHAT_ID = os.getenv("CHAT_ID")
|
|
||||||
TOKEN = os.getenv("BOT_TOKEN")
|
|
||||||
TELEGRAM_SERVER = os.getenv("TELEGRAM_SERVER", default=None)
|
|
||||||
|
|
||||||
if TELEGRAM_SERVER:
|
|
||||||
local_server = TelegramAPIServer.from_base(TELEGRAM_SERVER)
|
|
||||||
bot = Bot(TOKEN, server=local_server)
|
|
||||||
else:
|
|
||||||
bot = Bot(TOKEN)
|
|
||||||
|
|
||||||
|
|
||||||
client = Client(YANDEX_TOKEN).init()
|
|
||||||
latest_podcast = None
|
|
||||||
latest_sent = True
|
|
||||||
podcasts_listened = []
|
|
||||||
|
|
||||||
while True:
|
|
||||||
try:
|
|
||||||
queues = client.queues_list()
|
|
||||||
last_queue = client.queue(queues[0].id)
|
|
||||||
|
|
||||||
last_track_id = last_queue.get_current_track()
|
|
||||||
last_track: Track = last_track_id.fetch_track()
|
|
||||||
|
|
||||||
if "podcast" in last_track.type:
|
|
||||||
if last_track_id not in podcasts_listened:
|
|
||||||
if last_track_id == latest_podcast and not latest_sent:
|
|
||||||
latest_sent = True
|
|
||||||
podcasts_listened.append(last_track_id)
|
|
||||||
|
|
||||||
title = last_track.title
|
|
||||||
album = last_track.albums[0]
|
|
||||||
url = f"https://music.yandex.ru/track/{last_track.id}"
|
|
||||||
desc = last_track.short_description.split("\n")[0]
|
|
||||||
|
|
||||||
last_track.download_cover(filename="cover.png")
|
|
||||||
img_path = os.path.abspath("cover.png")
|
|
||||||
|
|
||||||
last_track.download(filename="file", codec="mp3")
|
|
||||||
orig_path = os.path.abspath("file")
|
|
||||||
path = os.path.abspath("file.mp3")
|
|
||||||
|
|
||||||
AudioSegment.from_file(orig_path).export(path)
|
|
||||||
os.remove(orig_path)
|
|
||||||
|
|
||||||
# set music meta
|
|
||||||
tag = MP3(path, ID3=ID3)
|
|
||||||
tag.tags.add(
|
|
||||||
APIC(
|
|
||||||
encoding=3, # 3 is for utf-8
|
|
||||||
mime="image/png", # image/jpeg or image/png
|
|
||||||
type=3, # 3 is for the cover image
|
|
||||||
desc="Cover",
|
|
||||||
data=open(img_path, "rb").read(),
|
|
||||||
)
|
|
||||||
)
|
|
||||||
tag.tags.add(TORY(text=str(album.year)))
|
|
||||||
tag.save()
|
|
||||||
tag = EasyID3(path)
|
|
||||||
|
|
||||||
tag["title"] = title
|
|
||||||
tag["album"] = album.title
|
|
||||||
|
|
||||||
tag.save()
|
|
||||||
|
|
||||||
with open(path, "rb") as tmp:
|
|
||||||
obj = BytesIO(tmp.read())
|
|
||||||
obj.name = f"{title}.mp3"
|
|
||||||
loop = asyncio.get_event_loop()
|
|
||||||
coroutine = bot.send_audio(
|
|
||||||
chat_id=CHAT_ID,
|
|
||||||
audio=obj,
|
|
||||||
caption=f"{title} - {album.title}\n{desc}\n\n{url}",
|
|
||||||
title=title,
|
|
||||||
performer=album.title,
|
|
||||||
)
|
|
||||||
loop.run_until_complete(coroutine)
|
|
||||||
|
|
||||||
else:
|
|
||||||
latest_podcast = last_track_id
|
|
||||||
latest_sent = False
|
|
||||||
except BaseException as e:
|
|
||||||
loop = asyncio.get_event_loop()
|
|
||||||
coroutine = bot.send_message(CHAT_ID, text=str(e))
|
|
||||||
loop.run_until_complete(coroutine)
|
|
||||||
sleep(5 * 60)
|
|
||||||
Loading…
Reference in New Issue
Block a user