mirror of
https://github.com/Alexander-D-Karpov/akarpov
synced 2024-12-01 05:13:45 +03:00
Compare commits
1 Commits
bbe08d0f43
...
9f18605069
Author | SHA1 | Date | |
---|---|---|---|
|
9f18605069 |
|
@ -13,11 +13,6 @@
|
|||
from ..documents import FileDocument
|
||||
from .lema import lemmatize_and_remove_stopwords
|
||||
|
||||
"""
|
||||
Calculus on types of searches:
|
||||
https://new.akarpov.ru/files/FZUTFBIyfbdlDHVzxUNU
|
||||
"""
|
||||
|
||||
|
||||
class BaseSearch:
|
||||
def __init__(self, queryset: QuerySet | None = None):
|
||||
|
|
|
@ -92,7 +92,9 @@ def get_queryset(self):
|
|||
):
|
||||
return self.filter(BaseFileItem.objects.none())
|
||||
return self.filter(
|
||||
BaseFileItem.objects.filter(user=self.request.user, parent__isnull=True)
|
||||
BaseFileItem.objects.cache().filter(
|
||||
user=self.request.user, parent__isnull=True
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
|
|
|
@ -7,8 +7,10 @@ class MusicConfig(AppConfig):
|
|||
name = "akarpov.music"
|
||||
|
||||
def ready(self):
|
||||
import akarpov.music.signals # noqa F401
|
||||
|
||||
try:
|
||||
import akarpov.music.signals # noqa F401
|
||||
except ImportError:
|
||||
pass
|
||||
try:
|
||||
from akarpov.music.tasks import start_next_song
|
||||
|
||||
|
|
|
@ -1,27 +0,0 @@
|
|||
# Generated by Django 4.2.8 on 2023-12-17 22:22
|
||||
|
||||
from django.db import migrations, models
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
dependencies = [
|
||||
("music", "0011_alter_playlist_private_userlistenhistory"),
|
||||
]
|
||||
|
||||
operations = [
|
||||
migrations.AddField(
|
||||
model_name="album",
|
||||
name="meta",
|
||||
field=models.JSONField(blank=True, null=True),
|
||||
),
|
||||
migrations.AddField(
|
||||
model_name="author",
|
||||
name="albums",
|
||||
field=models.ManyToManyField(related_name="authors", to="music.album"),
|
||||
),
|
||||
migrations.AddField(
|
||||
model_name="author",
|
||||
name="meta",
|
||||
field=models.JSONField(blank=True, null=True),
|
||||
),
|
||||
]
|
|
@ -1,31 +0,0 @@
|
|||
# Generated by Django 4.2.8 on 2023-12-18 11:44
|
||||
|
||||
from django.db import migrations, models
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
dependencies = [
|
||||
("music", "0012_album_meta_author_albums_author_meta"),
|
||||
]
|
||||
|
||||
operations = [
|
||||
migrations.RemoveField(
|
||||
model_name="author",
|
||||
name="albums",
|
||||
),
|
||||
migrations.AddField(
|
||||
model_name="album",
|
||||
name="authors",
|
||||
field=models.ManyToManyField(related_name="albums", to="music.album"),
|
||||
),
|
||||
migrations.AlterField(
|
||||
model_name="album",
|
||||
name="name",
|
||||
field=models.CharField(max_length=200, unique=True),
|
||||
),
|
||||
migrations.AlterField(
|
||||
model_name="author",
|
||||
name="name",
|
||||
field=models.CharField(max_length=200, unique=True),
|
||||
),
|
||||
]
|
|
@ -1,17 +0,0 @@
|
|||
# Generated by Django 4.2.8 on 2023-12-18 14:30
|
||||
|
||||
from django.db import migrations, models
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
dependencies = [
|
||||
("music", "0013_remove_author_albums_album_authors_alter_album_name_and_more"),
|
||||
]
|
||||
|
||||
operations = [
|
||||
migrations.AlterField(
|
||||
model_name="album",
|
||||
name="authors",
|
||||
field=models.ManyToManyField(related_name="albums", to="music.author"),
|
||||
),
|
||||
]
|
|
@ -8,9 +8,8 @@
|
|||
|
||||
|
||||
class Author(BaseImageModel, ShortLinkModel):
|
||||
name = models.CharField(max_length=200, unique=True)
|
||||
name = models.CharField(max_length=200)
|
||||
link = models.URLField(blank=True)
|
||||
meta = models.JSONField(blank=True, null=True)
|
||||
|
||||
def get_absolute_url(self):
|
||||
return reverse("music:author", kwargs={"slug": self.slug})
|
||||
|
@ -20,10 +19,8 @@ def __str__(self):
|
|||
|
||||
|
||||
class Album(BaseImageModel, ShortLinkModel):
|
||||
name = models.CharField(max_length=200, unique=True)
|
||||
name = models.CharField(max_length=200)
|
||||
link = models.URLField(blank=True)
|
||||
meta = models.JSONField(blank=True, null=True)
|
||||
authors = models.ManyToManyField("Author", related_name="albums")
|
||||
|
||||
def get_absolute_url(self):
|
||||
return reverse("music:album", kwargs={"slug": self.slug})
|
||||
|
|
|
@ -27,28 +27,16 @@ def load_track(
|
|||
if album and type(album) is str and album.startswith("['"):
|
||||
album = album.replace("['", "").replace("']", "")
|
||||
|
||||
re_authors = []
|
||||
if authors:
|
||||
for x in authors:
|
||||
try:
|
||||
re_authors.append(Author.objects.get(name=x))
|
||||
except Author.DoesNotExist:
|
||||
re_authors.append(Author.objects.create(name=x))
|
||||
authors = re_authors
|
||||
album_name = None
|
||||
authors = [Author.objects.get_or_create(name=x)[0] for x in authors if authors]
|
||||
else:
|
||||
authors = []
|
||||
if album:
|
||||
if type(album) is str:
|
||||
album_name = album
|
||||
album = Album.objects.get_or_create(name=album)[0]
|
||||
elif type(album) is list:
|
||||
album_name = album[0]
|
||||
else:
|
||||
album_name = None
|
||||
if album_name:
|
||||
try:
|
||||
album = Album.objects.get(name=album_name)
|
||||
except Album.DoesNotExist:
|
||||
album = Album.objects.create(name=album_name)
|
||||
if not album_name:
|
||||
album = Album.objects.get_or_create(name=album[0])[0]
|
||||
else:
|
||||
album = None
|
||||
|
||||
if sng := Song.objects.filter(
|
||||
|
@ -90,7 +78,7 @@ def load_track(
|
|||
str(
|
||||
slugify(
|
||||
GoogleTranslator(source="auto", target="en").translate(
|
||||
f"{song.name} {' '.join([x.name for x in authors])}",
|
||||
f"{song.name} {' '.join([x.name for x in song.authors])}",
|
||||
target_language="en",
|
||||
)
|
||||
)
|
||||
|
@ -113,7 +101,7 @@ def load_track(
|
|||
album.save()
|
||||
|
||||
if authors:
|
||||
song.authors.set([x.id for x in authors])
|
||||
song.authors.set(authors)
|
||||
|
||||
# set music meta
|
||||
tag = MutagenFile(song.file.path)
|
||||
|
@ -136,9 +124,9 @@ def load_track(
|
|||
data=f.read(),
|
||||
)
|
||||
)
|
||||
if "release" in kwargs and kwargs["release"]:
|
||||
if "release" in kwargs:
|
||||
tag.tags.add(TORY(text=kwargs["release"]))
|
||||
if "genre" in kwargs and kwargs["genre"]:
|
||||
if "genre" in kwargs:
|
||||
tag.tags.add(TCON(text=kwargs["genre"]))
|
||||
tag.save()
|
||||
|
||||
|
|
|
@ -2,6 +2,8 @@
|
|||
from django.conf import settings
|
||||
from spotipy.oauth2 import SpotifyClientCredentials
|
||||
|
||||
from akarpov.music.services.yandex import search_ym
|
||||
|
||||
|
||||
def login() -> spotipy.Spotify:
|
||||
if not settings.MUSIC_SPOTIFY_ID or not settings.MUSIC_SPOTIFY_SECRET:
|
||||
|
@ -45,7 +47,15 @@ def get_track_info(name: str) -> dict:
|
|||
# try to get genre
|
||||
sp = login()
|
||||
genres = sp.album(res["album"]["external_urls"]["spotify"])["genres"]
|
||||
if genres:
|
||||
if not genres:
|
||||
ym_info = search_ym(info["artist"] + " " + info["title"])
|
||||
if ym_info and "genre" in ym_info:
|
||||
info["genre"] = ym_info["genre"]
|
||||
else:
|
||||
genres = sp.artist(res["artists"][0]["external_urls"]["spotify"])["genres"]
|
||||
if genres:
|
||||
info["genre"] = genres[0]
|
||||
else:
|
||||
info["genre"] = genres[0]
|
||||
|
||||
return info
|
||||
|
|
|
@ -2,13 +2,11 @@
|
|||
from random import randint
|
||||
|
||||
from django.conf import settings
|
||||
from django.core.files import File
|
||||
from django.utils.text import slugify
|
||||
from yandex_music import Client, Playlist, Search, Track
|
||||
from yandex_music.exceptions import NotFoundError
|
||||
|
||||
from akarpov.music import tasks
|
||||
from akarpov.music.models import Album as AlbumModel
|
||||
from akarpov.music.models import Author, Song, SongInQue
|
||||
from akarpov.music.models import Song, SongInQue
|
||||
from akarpov.music.services.db import load_track
|
||||
|
||||
|
||||
|
@ -45,7 +43,7 @@ def search_ym(name: str):
|
|||
return info
|
||||
|
||||
|
||||
def load_file_meta(track: int, user_id: int) -> str:
|
||||
def load_file_meta(track: int, user_id: int):
|
||||
que = SongInQue.objects.create()
|
||||
client = login()
|
||||
track = client.tracks(track)[0] # type: Track
|
||||
|
@ -57,27 +55,19 @@ def load_file_meta(track: int, user_id: int) -> str:
|
|||
name=track.title, album__name=track.albums[0].title
|
||||
):
|
||||
que.delete()
|
||||
return str(sng.first())
|
||||
return sng.first()
|
||||
except IndexError:
|
||||
que.delete()
|
||||
return ""
|
||||
return
|
||||
|
||||
filename = f"_{str(randint(10000, 9999999))}"
|
||||
filename = slugify(f"{track.artists[0].name} - {track.title}")
|
||||
orig_path = f"{settings.MEDIA_ROOT}/{filename}.mp3"
|
||||
album = track.albums[0]
|
||||
|
||||
track.download(filename=orig_path, codec="mp3")
|
||||
img_pth = str(settings.MEDIA_ROOT + f"/_{str(randint(10000, 99999))}.png")
|
||||
|
||||
try:
|
||||
track.download_cover(filename=img_pth)
|
||||
except NotFoundError:
|
||||
img_pth = None
|
||||
|
||||
try:
|
||||
lyrics = track.get_lyrics("LRC").fetch_lyrics()
|
||||
except NotFoundError:
|
||||
lyrics = ""
|
||||
track.download_cover(filename=img_pth)
|
||||
song = load_track(
|
||||
orig_path,
|
||||
img_pth,
|
||||
|
@ -87,9 +77,6 @@ def load_file_meta(track: int, user_id: int) -> str:
|
|||
track.title,
|
||||
release=album.release_date,
|
||||
genre=album.genre,
|
||||
lyrics=lyrics,
|
||||
explicit=track.explicit,
|
||||
track_source=track.track_source,
|
||||
)
|
||||
if os.path.exists(orig_path):
|
||||
os.remove(orig_path)
|
||||
|
@ -109,64 +96,3 @@ def load_playlist(link: str, user_id: int):
|
|||
tasks.load_ym_file_meta.apply_async(
|
||||
kwargs={"track": track.track.id, "user_id": user_id}
|
||||
)
|
||||
|
||||
|
||||
def update_album_info(album: AlbumModel) -> None:
|
||||
client = login()
|
||||
search = client.search(album.name, type_="album") # type: Search
|
||||
|
||||
if search.albums:
|
||||
search_album = search.albums.results[0]
|
||||
data = {
|
||||
"name": search_album.title,
|
||||
"tracks": search_album.track_count,
|
||||
"explicit": search_album.explicit,
|
||||
"year": search_album.year,
|
||||
"genre": search_album.genre,
|
||||
"description": search_album.description,
|
||||
"type": search_album.type,
|
||||
}
|
||||
|
||||
album.meta = data
|
||||
image_path = str(settings.MEDIA_ROOT + f"/_{str(randint(10000, 99999))}.png")
|
||||
if search_album.cover_uri:
|
||||
search_album.download_cover(filename=image_path)
|
||||
with open(image_path, "rb") as f:
|
||||
album.image = File(f, name=image_path.split("/")[-1])
|
||||
album.save()
|
||||
os.remove(image_path)
|
||||
|
||||
authors = []
|
||||
if search_album.artists:
|
||||
for x in search_album.artists:
|
||||
try:
|
||||
authors.append(Author.objects.get(name=x.name))
|
||||
except Author.DoesNotExist:
|
||||
authors.append(Author.objects.create(name=x.name))
|
||||
album.authors.set([x.id for x in authors])
|
||||
album.save()
|
||||
|
||||
|
||||
def update_author_info(author: Author) -> None:
|
||||
client = login()
|
||||
search = client.search(author.name, type_="artist") # type: Search
|
||||
|
||||
if search.artists:
|
||||
search_artist = search.artists.results[0]
|
||||
data = {
|
||||
"name": search_artist.name,
|
||||
"description": search_artist.description,
|
||||
"genres": search_artist.genres,
|
||||
}
|
||||
|
||||
author.meta = data
|
||||
|
||||
image_path = str(settings.MEDIA_ROOT + f"/_{str(randint(10000, 99999))}.png")
|
||||
if not search_artist.cover:
|
||||
author.save()
|
||||
return
|
||||
search_artist.cover.download(filename=image_path)
|
||||
with open(image_path, "rb") as f:
|
||||
author.image = File(f, name=image_path.split("/")[-1])
|
||||
author.save()
|
||||
os.remove(image_path)
|
||||
|
|
|
@ -4,7 +4,6 @@
|
|||
from django.dispatch import receiver
|
||||
|
||||
from akarpov.music.models import Album, Author, PlaylistSong, Song, SongUserRating
|
||||
from akarpov.music.services.yandex import update_album_info, update_author_info
|
||||
|
||||
|
||||
@receiver(post_delete, sender=Song)
|
||||
|
@ -17,13 +16,15 @@ def auto_delete_file_on_delete(sender, instance, **kwargs):
|
|||
@receiver(post_save, sender=Author)
|
||||
def author_create(sender, instance, created, **kwargs):
|
||||
if created:
|
||||
update_author_info(instance)
|
||||
# TODO: add logic to retrieve author info here
|
||||
return
|
||||
|
||||
|
||||
@receiver(post_save, sender=Album)
|
||||
def album_create(sender, instance, created, **kwargs):
|
||||
if created:
|
||||
update_album_info(instance)
|
||||
# TODO: add logic to retrieve author info here
|
||||
return
|
||||
|
||||
|
||||
@receiver(post_save)
|
||||
|
|
|
@ -3,6 +3,7 @@
|
|||
set -o errexit
|
||||
set -o nounset
|
||||
|
||||
python manage.py makemigrations
|
||||
python manage.py migrate
|
||||
rm -f './celerybeat.pid'
|
||||
celery -A config.celery_app beat -l INFO
|
||||
|
|
1903
poetry.lock
generated
1903
poetry.lock
generated
File diff suppressed because it is too large
Load Diff
|
@ -28,7 +28,7 @@ django-crispy-forms = "^2.1"
|
|||
crispy-bootstrap5 = "^0.7"
|
||||
django-redis = "^5.2.0"
|
||||
django-ckeditor = "^6.5.1"
|
||||
django-colorfield = "^0.11.0"
|
||||
django-colorfield = "^0.8.0"
|
||||
djangorestframework = "^3.14.0"
|
||||
django-rest-auth = "^0.9.5"
|
||||
django-cors-headers = "^4.0.0"
|
||||
|
@ -77,7 +77,7 @@ pydub = "^0.25.1"
|
|||
python-mpd2 = "^3.0.5"
|
||||
yandex-music = "^2.1.1"
|
||||
pyjwt = "^2.8.0"
|
||||
rawpy = "^0.19.0"
|
||||
rawpy = "^0.18.1"
|
||||
xvfbwrapper = "^0.2.9"
|
||||
vtk = "^9.2.6"
|
||||
ffmpeg-python = "^0.2.0"
|
||||
|
|
Loading…
Reference in New Issue
Block a user