Compare commits
No commits in common. '2d955ab30aafe0bfe09d689aa91cf56adeb273ca' and 'be9c0cfbeb85273d12389578ccd73b3aedc8800a' have entirely different histories.
2d955ab30a
...
be9c0cfbeb
@ -1,5 +1,138 @@
|
|||||||
*.tmp
|
*.mp3
|
||||||
*.zip
|
*.ogg
|
||||||
*.tar
|
*.wav
|
||||||
|
*.flac
|
||||||
|
|
||||||
|
*.ipynb
|
||||||
|
*.idea
|
||||||
|
|
||||||
|
|
||||||
|
# Byte-compiled / optimized / DLL files
|
||||||
__pycache__/
|
__pycache__/
|
||||||
music/
|
*.py[cod]
|
||||||
|
*$py.class
|
||||||
|
|
||||||
|
# C extensions
|
||||||
|
*.so
|
||||||
|
|
||||||
|
# Distribution / packaging
|
||||||
|
.Python
|
||||||
|
build/
|
||||||
|
develop-eggs/
|
||||||
|
dist/
|
||||||
|
downloads/
|
||||||
|
eggs/
|
||||||
|
.eggs/
|
||||||
|
lib/
|
||||||
|
lib64/
|
||||||
|
parts/
|
||||||
|
sdist/
|
||||||
|
var/
|
||||||
|
wheels/
|
||||||
|
pip-wheel-metadata/
|
||||||
|
share/python-wheels/
|
||||||
|
*.egg-info/
|
||||||
|
.installed.cfg
|
||||||
|
*.egg
|
||||||
|
MANIFEST
|
||||||
|
|
||||||
|
# PyInstaller
|
||||||
|
# Usually these files are written by a python script from a template
|
||||||
|
# before PyInstaller builds the exe, so as to inject date/other infos into it.
|
||||||
|
*.manifest
|
||||||
|
*.spec
|
||||||
|
|
||||||
|
# Installer logs
|
||||||
|
pip-log.txt
|
||||||
|
pip-delete-this-directory.txt
|
||||||
|
|
||||||
|
# Unit test / coverage reports
|
||||||
|
htmlcov/
|
||||||
|
.tox/
|
||||||
|
.nox/
|
||||||
|
.coverage
|
||||||
|
.coverage.*
|
||||||
|
.cache
|
||||||
|
nosetests.xml
|
||||||
|
coverage.xml
|
||||||
|
*.cover
|
||||||
|
*.py,cover
|
||||||
|
.hypothesis/
|
||||||
|
.pytest_cache/
|
||||||
|
|
||||||
|
# Translations
|
||||||
|
*.mo
|
||||||
|
*.pot
|
||||||
|
|
||||||
|
# Django stuff:
|
||||||
|
*.log
|
||||||
|
local_settings.py
|
||||||
|
db.sqlite3
|
||||||
|
db.sqlite3-journal
|
||||||
|
|
||||||
|
# Flask stuff:
|
||||||
|
instance/
|
||||||
|
.webassets-cache
|
||||||
|
|
||||||
|
# Scrapy stuff:
|
||||||
|
.scrapy
|
||||||
|
|
||||||
|
# Sphinx documentation
|
||||||
|
docs/_build/
|
||||||
|
|
||||||
|
# PyBuilder
|
||||||
|
target/
|
||||||
|
|
||||||
|
# Jupyter Notebook
|
||||||
|
.ipynb_checkpoints
|
||||||
|
|
||||||
|
# IPython
|
||||||
|
profile_default/
|
||||||
|
ipython_config.py
|
||||||
|
|
||||||
|
# pyenv
|
||||||
|
.python-version
|
||||||
|
|
||||||
|
# pipenv
|
||||||
|
# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
|
||||||
|
# However, in case of collaboration, if having platform-specific dependencies or dependencies
|
||||||
|
# having no cross-platform support, pipenv may install dependencies that don't work, or not
|
||||||
|
# install all needed dependencies.
|
||||||
|
#Pipfile.lock
|
||||||
|
|
||||||
|
# PEP 582; used by e.g. github.com/David-OConnor/pyflow
|
||||||
|
__pypackages__/
|
||||||
|
|
||||||
|
# Celery stuff
|
||||||
|
celerybeat-schedule
|
||||||
|
celerybeat.pid
|
||||||
|
|
||||||
|
# SageMath parsed files
|
||||||
|
*.sage.py
|
||||||
|
|
||||||
|
# Environments
|
||||||
|
.env
|
||||||
|
.venv
|
||||||
|
env/
|
||||||
|
venv/
|
||||||
|
ENV/
|
||||||
|
env.bak/
|
||||||
|
venv.bak/
|
||||||
|
|
||||||
|
# Spyder project settings
|
||||||
|
.spyderproject
|
||||||
|
.spyproject
|
||||||
|
|
||||||
|
# Rope project settings
|
||||||
|
.ropeproject
|
||||||
|
|
||||||
|
# mkdocs documentation
|
||||||
|
/site
|
||||||
|
|
||||||
|
# mypy
|
||||||
|
.mypy_cache/
|
||||||
|
.dmypy.json
|
||||||
|
dmypy.json
|
||||||
|
|
||||||
|
# Pyre type checker
|
||||||
|
.pyre/
|
||||||
|
@ -1,12 +0,0 @@
|
|||||||
import importlib
|
|
||||||
import config
|
|
||||||
importlib.reload(config)
|
|
||||||
import utils
|
|
||||||
importlib.reload(utils)
|
|
||||||
import database
|
|
||||||
importlib.reload(database)
|
|
||||||
import ui
|
|
||||||
importlib.reload(ui)
|
|
||||||
import app
|
|
||||||
importlib.reload(app)
|
|
||||||
|
|
@ -1,9 +0,0 @@
|
|||||||
from config import music_path
|
|
||||||
from ui import UIEngine
|
|
||||||
from database import Database
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
# Add your main code here
|
|
||||||
db = Database(base_path=music_path)
|
|
||||||
ui = UIEngine(db)
|
|
||||||
ui.run()
|
|
@ -0,0 +1,300 @@
|
|||||||
|
import pygame
|
||||||
|
import threading
|
||||||
|
import time
|
||||||
|
import random
|
||||||
|
import os
|
||||||
|
import atexit
|
||||||
|
import ntpath
|
||||||
|
import json
|
||||||
|
import webbrowser
|
||||||
|
import subprocess
|
||||||
|
|
||||||
|
import dearpygui.dearpygui as dpg
|
||||||
|
|
||||||
|
from mutagen.mp3 import MP3
|
||||||
|
from tkinter import Tk,filedialog, simpledialog, Button, OptionMenu, N, S , E, W, Label, StringVar
|
||||||
|
import pytube
|
||||||
|
|
||||||
|
dpg.create_context()
|
||||||
|
dpg.create_viewport(title="celaigia, stai senza pensieri",large_icon="logo.ico",small_icon="logo.ico")
|
||||||
|
pygame.mixer.init()
|
||||||
|
|
||||||
|
global state
|
||||||
|
state=None
|
||||||
|
|
||||||
|
_SONG_FILE = "data/songs.json"
|
||||||
|
_NUM_YT_SEARCH_RESULTS = 5
|
||||||
|
_DEFAULT_DOWNLOAD_PATH = 'data/music'
|
||||||
|
_DEFAULT_MUSIC_VOLUME = 0.5
|
||||||
|
pygame.mixer.music.set_volume(_DEFAULT_MUSIC_VOLUME)
|
||||||
|
|
||||||
|
def bash(cmd):
|
||||||
|
subprocess.call(['/bin/bash', '-c', cmd])
|
||||||
|
|
||||||
|
def string_sanitizer(s: str)->str:
|
||||||
|
s = ''.join(filter(lambda x: x in '0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ ', s))
|
||||||
|
return s.replace(' ','_')
|
||||||
|
|
||||||
|
def ismusic(filename: str)->bool:
|
||||||
|
return filename.split('.')[-1] in ['mp3', 'ogg', 'flac', 'wav']
|
||||||
|
|
||||||
|
def update_volume(sender, app_data):
|
||||||
|
pygame.mixer.music.set_volume(app_data / 100.0)
|
||||||
|
|
||||||
|
def load_database():
|
||||||
|
songs = json.load(open(_SONG_FILE, "r+"))["songs"]
|
||||||
|
for filename in songs:
|
||||||
|
dpg.add_button(
|
||||||
|
label=f"{ntpath.basename(filename)}",
|
||||||
|
callback=play,
|
||||||
|
width=-1,
|
||||||
|
height=25,
|
||||||
|
user_data=filename.replace("\\", "/"),
|
||||||
|
parent="list"
|
||||||
|
)
|
||||||
|
dpg.add_spacer(height=2, parent="list")
|
||||||
|
|
||||||
|
|
||||||
|
def update_database(filename: str):
|
||||||
|
data = json.load(open(_SONG_FILE, "r+"))
|
||||||
|
if filename not in data["songs"]:
|
||||||
|
data["songs"] += [filename]
|
||||||
|
dpg.add_button(
|
||||||
|
label=f"{ntpath.basename(filename)}",
|
||||||
|
callback=play,
|
||||||
|
width=-1,
|
||||||
|
height=25,
|
||||||
|
user_data=filename.replace("\\", "/"),
|
||||||
|
parent="list"
|
||||||
|
)
|
||||||
|
dpg.add_spacer(height=2, parent="list")
|
||||||
|
json.dump(data, open(_SONG_FILE, "r+"), indent=4)
|
||||||
|
|
||||||
|
def update_slider():
|
||||||
|
global state
|
||||||
|
while pygame.mixer.music.get_busy():
|
||||||
|
dpg.configure_item(item="pos",default_value=pygame.mixer.music.get_pos()/1000)
|
||||||
|
time.sleep(0.7)
|
||||||
|
state=None
|
||||||
|
dpg.configure_item("cstate",default_value=f"State: None")
|
||||||
|
dpg.configure_item("csong",default_value="Now Playing : ")
|
||||||
|
dpg.configure_item("play",label="Play")
|
||||||
|
dpg.configure_item(item="pos",max_value=100)
|
||||||
|
dpg.configure_item(item="pos",default_value=0)
|
||||||
|
|
||||||
|
def play(sender, app_data, user_data):
|
||||||
|
global state
|
||||||
|
if user_data:
|
||||||
|
pygame.mixer.music.load(user_data)
|
||||||
|
audio = MP3(user_data)
|
||||||
|
dpg.configure_item(item="pos",max_value=audio.info.length)
|
||||||
|
pygame.mixer.music.play()
|
||||||
|
thread=threading.Thread(target=update_slider,daemon=False).start()
|
||||||
|
if pygame.mixer.music.get_busy():
|
||||||
|
dpg.configure_item("play",label="Pause")
|
||||||
|
state="playing"
|
||||||
|
dpg.configure_item("cstate",default_value=f"State: Playing")
|
||||||
|
dpg.configure_item("csong",default_value=f"Now Playing : {ntpath.basename(user_data)}")
|
||||||
|
|
||||||
|
def play_pause():
|
||||||
|
global state
|
||||||
|
if state=="playing":
|
||||||
|
state="paused"
|
||||||
|
pygame.mixer.music.pause()
|
||||||
|
dpg.configure_item("play",label="Play")
|
||||||
|
dpg.configure_item("cstate",default_value=f"State: Paused")
|
||||||
|
elif state=="paused":
|
||||||
|
state="playing"
|
||||||
|
pygame.mixer.music.unpause()
|
||||||
|
dpg.configure_item("play",label="Pause")
|
||||||
|
dpg.configure_item("cstate",default_value=f"State: Playing")
|
||||||
|
else:
|
||||||
|
song = json.load(open(_SONG_FILE, "r"))["songs"]
|
||||||
|
if song:
|
||||||
|
song=random.choice(song)
|
||||||
|
pygame.mixer.music.load(song)
|
||||||
|
pygame.mixer.music.play()
|
||||||
|
thread=threading.Thread(target=update_slider,daemon=False).start()
|
||||||
|
dpg.configure_item("play",label="Pause")
|
||||||
|
if pygame.mixer.music.get_busy():
|
||||||
|
audio = MP3(song)
|
||||||
|
dpg.configure_item(item="pos",max_value=audio.info.length)
|
||||||
|
state="playing"
|
||||||
|
dpg.configure_item("csong",default_value=f"Now Playing : {ntpath.basename(song)}")
|
||||||
|
dpg.configure_item("cstate",default_value=f"State: Playing")
|
||||||
|
|
||||||
|
def stop():
|
||||||
|
global state
|
||||||
|
pygame.mixer.music.stop()
|
||||||
|
state=None
|
||||||
|
|
||||||
|
def add_files():
|
||||||
|
data=json.load(open(_SONG_FILE,"r"))
|
||||||
|
root=Tk()
|
||||||
|
root.withdraw()
|
||||||
|
filename=filedialog.askopenfilename(filetypes=[("Music Files", ("*.mp3","*.wav","*.ogg"))])
|
||||||
|
root.quit()
|
||||||
|
if filename.endswith(".mp3" or ".wav" or ".ogg"):
|
||||||
|
if filename not in data["songs"]:
|
||||||
|
update_database(filename)
|
||||||
|
#dpg.add_button(label=f"{ntpath.basename(filename)}",callback=play,width=-1,height=25,,parent="list")
|
||||||
|
#dpg.add_spacer(height=2,parent="list")
|
||||||
|
|
||||||
|
def add_folder():
|
||||||
|
data=json.load(open(_SONG_FILE,"r"))
|
||||||
|
root=Tk()
|
||||||
|
root.withdraw()
|
||||||
|
folder=filedialog.askdirectory()
|
||||||
|
root.quit()
|
||||||
|
for filename in os.listdir(folder):
|
||||||
|
if filename.endswith(".mp3" or ".wav" or ".ogg"):
|
||||||
|
if filename not in data["songs"]:
|
||||||
|
update_database(os.path.join(folder,filename).replace("\\","/"))
|
||||||
|
#dpg.add_button(label=f"{ntpath.basename(filename)}",callback=play,width=-1,height=25,user_data=os.path.join(folder,filename).replace("\\","/"),parent="list")
|
||||||
|
#dpg.add_spacer(height=2,parent="list")
|
||||||
|
|
||||||
|
def search(sender, app_data, user_data):
|
||||||
|
songs = json.load(open(_SONG_FILE, "r"))["songs"]
|
||||||
|
dpg.delete_item("list", children_only=True)
|
||||||
|
for index, song in enumerate(songs):
|
||||||
|
if app_data in song.lower():
|
||||||
|
dpg.add_button(label=f"{ntpath.basename(song)}", callback=play,width=-1, height=25, user_data=song, parent="list")
|
||||||
|
dpg.add_spacer(height=2,parent="list")
|
||||||
|
|
||||||
|
def add_download_choices(sender, app_data, user_data)->None:
|
||||||
|
yt_urls = {string_sanitizer(r.title): r.watch_url for r in pytube.Search(app_data).results[:_NUM_YT_SEARCH_RESULTS]}
|
||||||
|
|
||||||
|
for url_key in yt_urls:
|
||||||
|
dpg.add_button(label=url_key, tag=url_key, callback= download ,width=-1, height=25, user_data = (url_key,yt_urls), parent="sidebar")
|
||||||
|
dpg.add_spacer(height=2,parent="sidebar")
|
||||||
|
|
||||||
|
def download(sender, app_data, user_data):
|
||||||
|
|
||||||
|
url_key, yt_urls = user_data
|
||||||
|
url = yt_urls[url_key]
|
||||||
|
for tmp_url_key in yt_urls:
|
||||||
|
dpg.delete_item(tmp_url_key)
|
||||||
|
video_file = f'{_DEFAULT_DOWNLOAD_PATH}/{url_key}.mp4'
|
||||||
|
audio_file = f"{video_file[:-4]}.mp3"
|
||||||
|
stream = pytube.YouTube(url).streams.filter(only_audio=True).first()
|
||||||
|
stream.download(filename=video_file, skip_existing=True)
|
||||||
|
bash(f"ffmpeg -i {video_file} -vn -acodec mp3 -y {audio_file}")
|
||||||
|
bash(f"rm {video_file}")
|
||||||
|
update_database(audio_file)
|
||||||
|
|
||||||
|
def removeall():
|
||||||
|
songs = json.load(open(_SONG_FILE, "r"))
|
||||||
|
songs["songs"].clear()
|
||||||
|
json.dump(songs,open(_SONG_FILE, "w"),indent=4)
|
||||||
|
dpg.delete_item("list", children_only=True)
|
||||||
|
load_database()
|
||||||
|
|
||||||
|
with dpg.theme(tag="base"):
|
||||||
|
with dpg.theme_component():
|
||||||
|
dpg.add_theme_color(dpg.mvThemeCol_Button, (130, 142, 250))
|
||||||
|
dpg.add_theme_color(dpg.mvThemeCol_ButtonActive, (137, 142, 255, 95))
|
||||||
|
dpg.add_theme_color(dpg.mvThemeCol_ButtonHovered, (137, 142, 255))
|
||||||
|
dpg.add_theme_style(dpg.mvStyleVar_FrameRounding, 3)
|
||||||
|
dpg.add_theme_style(dpg.mvStyleVar_ChildRounding, 4)
|
||||||
|
dpg.add_theme_style(dpg.mvStyleVar_FramePadding, 4, 4)
|
||||||
|
dpg.add_theme_style(dpg.mvStyleVar_WindowRounding, 4, 4)
|
||||||
|
dpg.add_theme_style(dpg.mvStyleVar_WindowTitleAlign, 0.50, 0.50)
|
||||||
|
dpg.add_theme_style(dpg.mvStyleVar_WindowBorderSize,0)
|
||||||
|
dpg.add_theme_style(dpg.mvStyleVar_WindowPadding,10,14)
|
||||||
|
dpg.add_theme_color(dpg.mvThemeCol_ChildBg, (25, 25, 25))
|
||||||
|
dpg.add_theme_color(dpg.mvThemeCol_Border, (0,0,0,0))
|
||||||
|
dpg.add_theme_color(dpg.mvThemeCol_ScrollbarBg, (0,0,0,0))
|
||||||
|
dpg.add_theme_color(dpg.mvThemeCol_TitleBgActive, (130, 142, 250))
|
||||||
|
dpg.add_theme_color(dpg.mvThemeCol_CheckMark, (221, 166, 185))
|
||||||
|
dpg.add_theme_color(dpg.mvThemeCol_FrameBgHovered, (172, 174, 197))
|
||||||
|
|
||||||
|
with dpg.theme(tag="slider_thin"):
|
||||||
|
with dpg.theme_component():
|
||||||
|
dpg.add_theme_color(dpg.mvThemeCol_FrameBgActive, (130, 142, 250,99))
|
||||||
|
dpg.add_theme_color(dpg.mvThemeCol_FrameBgHovered, (130, 142, 250,99))
|
||||||
|
dpg.add_theme_color(dpg.mvThemeCol_SliderGrabActive, (255, 255, 255))
|
||||||
|
dpg.add_theme_color(dpg.mvThemeCol_SliderGrab, (255, 255, 255))
|
||||||
|
dpg.add_theme_color(dpg.mvThemeCol_FrameBg, (130, 142, 250,99))
|
||||||
|
dpg.add_theme_style(dpg.mvStyleVar_GrabRounding, 3)
|
||||||
|
dpg.add_theme_style(dpg.mvStyleVar_GrabMinSize, 30)
|
||||||
|
|
||||||
|
with dpg.theme(tag="slider"):
|
||||||
|
with dpg.theme_component():
|
||||||
|
dpg.add_theme_color(dpg.mvThemeCol_FrameBgActive, (130, 142, 250,99))
|
||||||
|
dpg.add_theme_color(dpg.mvThemeCol_FrameBgHovered, (130, 142, 250,99))
|
||||||
|
dpg.add_theme_color(dpg.mvThemeCol_SliderGrabActive, (255, 255, 255))
|
||||||
|
dpg.add_theme_color(dpg.mvThemeCol_SliderGrab, (255, 255, 255))
|
||||||
|
dpg.add_theme_color(dpg.mvThemeCol_FrameBg, (130, 142, 250,99))
|
||||||
|
dpg.add_theme_style(dpg.mvStyleVar_GrabRounding, 3)
|
||||||
|
dpg.add_theme_style(dpg.mvStyleVar_GrabMinSize, 30)
|
||||||
|
|
||||||
|
with dpg.theme(tag="songs"):
|
||||||
|
with dpg.theme_component():
|
||||||
|
dpg.add_theme_style(dpg.mvStyleVar_FrameRounding, 2)
|
||||||
|
dpg.add_theme_color(dpg.mvThemeCol_Button, (89, 89, 144,40))
|
||||||
|
dpg.add_theme_color(dpg.mvThemeCol_ButtonHovered, (0,0,0,0))
|
||||||
|
|
||||||
|
with dpg.font_registry():
|
||||||
|
monobold = dpg.add_font("fonts/MonoLisa-Bold.ttf", 12)
|
||||||
|
head = dpg.add_font("fonts/MonoLisa-Bold.ttf", 15)
|
||||||
|
|
||||||
|
with dpg.window(tag="main",label="window title"):
|
||||||
|
with dpg.child_window(autosize_x=True,height=45,no_scrollbar=True):
|
||||||
|
dpg.add_text(f"Now Playing : ",tag="csong")
|
||||||
|
dpg.add_spacer(height=2)
|
||||||
|
|
||||||
|
with dpg.group(horizontal=True):
|
||||||
|
with dpg.child_window(width=250, tag="sidebar"):
|
||||||
|
dpg.add_text("Celaigia",color=(137, 142, 255))
|
||||||
|
dpg.add_text("phuturemachine")
|
||||||
|
dpg.add_spacer(height=2)
|
||||||
|
dpg.add_button(label="Support",width=-1,height=23,callback=lambda:webbrowser.open(url="gitgitigitigiti"))
|
||||||
|
dpg.add_spacer(height=5)
|
||||||
|
dpg.add_separator()
|
||||||
|
dpg.add_spacer(height=5)
|
||||||
|
dpg.add_button(label="Add File",width=-1,height=28,callback=add_files)
|
||||||
|
dpg.add_button(label="Add Folder",width=-1,height=28,callback=add_folder)
|
||||||
|
dpg.add_button(label="Remove All Songs",width=-1,height=28,callback=removeall)
|
||||||
|
dpg.add_spacer(height=5)
|
||||||
|
dpg.add_separator()
|
||||||
|
dpg.add_spacer(height=5)
|
||||||
|
dpg.add_text(f"State: {state}",tag="cstate")
|
||||||
|
dpg.add_spacer(height=5)
|
||||||
|
dpg.add_separator()
|
||||||
|
dpg.add_input_text(hint="search youtube",width=-1,callback=add_download_choices, on_enter=True)
|
||||||
|
dpg.add_spacer(height=5)
|
||||||
|
|
||||||
|
with dpg.child_window(autosize_x=True,border=False):
|
||||||
|
with dpg.child_window(autosize_x=True,height=50,no_scrollbar=True):
|
||||||
|
with dpg.group(horizontal=True):
|
||||||
|
dpg.add_button(label="Play",width=65,height=30,tag="play",callback=play_pause)
|
||||||
|
dpg.add_button(label="Stop",callback=stop,width=65,height=30)
|
||||||
|
dpg.add_slider_float(tag="volume", width=120,height=15,pos=(160,19),format="%.0f%.0%",default_value=_DEFAULT_MUSIC_VOLUME * 100,callback=update_volume)
|
||||||
|
dpg.add_slider_float(tag="pos",width=-1,pos=(295,19),format="")
|
||||||
|
|
||||||
|
with dpg.child_window(autosize_x=True,delay_search=True):
|
||||||
|
with dpg.group(horizontal=True,tag="query"):
|
||||||
|
dpg.add_input_text(hint="search for a song locally",width=-1,callback=search)
|
||||||
|
dpg.add_spacer(height=5)
|
||||||
|
with dpg.child_window(autosize_x=True,delay_search=True,tag="list"):
|
||||||
|
load_database()
|
||||||
|
|
||||||
|
dpg.bind_item_theme("volume","slider_thin")
|
||||||
|
dpg.bind_item_theme("pos","slider")
|
||||||
|
dpg.bind_item_theme("list","songs")
|
||||||
|
|
||||||
|
dpg.bind_theme("base")
|
||||||
|
dpg.bind_font(monobold)
|
||||||
|
|
||||||
|
def safe_exit():
|
||||||
|
pygame.mixer.music.stop()
|
||||||
|
pygame.quit()
|
||||||
|
|
||||||
|
atexit.register(safe_exit)
|
||||||
|
|
||||||
|
dpg.setup_dearpygui()
|
||||||
|
dpg.show_viewport()
|
||||||
|
dpg.set_primary_window("main",True)
|
||||||
|
dpg.maximize_viewport()
|
||||||
|
dpg.start_dearpygui()
|
||||||
|
dpg.destroy_context()
|
@ -1,11 +0,0 @@
|
|||||||
music_path = "./music"
|
|
||||||
|
|
||||||
yt_dlp_opts = {
|
|
||||||
'format': 'bestaudio/best',
|
|
||||||
'default_search': 'ytsearch5',
|
|
||||||
'postprocessors': [{
|
|
||||||
'key': 'FFmpegExtractAudio',
|
|
||||||
'preferredcodec': 'mp3',
|
|
||||||
'preferredquality': '192',
|
|
||||||
}],
|
|
||||||
}
|
|
@ -0,0 +1,4 @@
|
|||||||
|
{
|
||||||
|
"songs": [
|
||||||
|
]
|
||||||
|
}
|
@ -1,37 +0,0 @@
|
|||||||
# Updated database.py
|
|
||||||
import json
|
|
||||||
import datetime
|
|
||||||
|
|
||||||
class Database:
|
|
||||||
def __init__(self, base_path):
|
|
||||||
self.base_path = base_path
|
|
||||||
self.file_path = f'{base_path}/index.json'
|
|
||||||
try:
|
|
||||||
with open(self.file_path, 'r') as file:
|
|
||||||
self.data = json.load(file)
|
|
||||||
except FileNotFoundError:
|
|
||||||
self.data = []
|
|
||||||
|
|
||||||
def add_entry(self, entry):
|
|
||||||
entry['timestamp'] = datetime.datetime.now().strftime('%d/%m/%Y')
|
|
||||||
self.data.append(entry)
|
|
||||||
with open(self.file_path, 'w') as file:
|
|
||||||
json.dump(self.data, file, indent=4)
|
|
||||||
|
|
||||||
def downloadable(self, url, title):
|
|
||||||
idx = url.split('https://www.youtube.com/watch?v=')[-1]
|
|
||||||
for entry in self.data:
|
|
||||||
if entry['id'] == idx or entry['title'] == title:
|
|
||||||
return False
|
|
||||||
return True
|
|
||||||
|
|
||||||
def get_all_entries(self):
|
|
||||||
return self.data
|
|
||||||
|
|
||||||
def update_entry(self, entry_id, new_data):
|
|
||||||
for entry in self.data:
|
|
||||||
if entry['id'] == entry_id:
|
|
||||||
entry.update(new_data)
|
|
||||||
with open(self.file_path, 'w') as file:
|
|
||||||
json.dump(self.data, file, indent=4)
|
|
||||||
return
|
|
Binary file not shown.
@ -1 +1,6 @@
|
|||||||
yt-dlp
|
dearpygui
|
||||||
|
mutagen
|
||||||
|
pygame
|
||||||
|
pytube
|
||||||
|
tk
|
||||||
|
|
||||||
|
@ -1,137 +0,0 @@
|
|||||||
import tkinter as tk
|
|
||||||
from tkinter import ttk
|
|
||||||
import tkinter.simpledialog as sd
|
|
||||||
import time
|
|
||||||
import os
|
|
||||||
from utils import search_youtube, download_audio
|
|
||||||
|
|
||||||
class UIEngine:
|
|
||||||
def __init__(self, db):
|
|
||||||
self.db = db
|
|
||||||
self.root = tk.Tk()
|
|
||||||
self.root.title("celaigia")
|
|
||||||
self.root.geometry("400x300") # Set a fixed size
|
|
||||||
|
|
||||||
# Create a PanedWindow to split the UI vertically
|
|
||||||
self.paned_window = tk.PanedWindow(self.root, orient=tk.VERTICAL)
|
|
||||||
self.paned_window.pack(expand=True, fill='both')
|
|
||||||
|
|
||||||
# Create the top part of the UI
|
|
||||||
self.top_frame = ttk.Frame(self.paned_window)
|
|
||||||
self.paned_window.add(self.top_frame)
|
|
||||||
|
|
||||||
self.status_label = tk.Label(self.top_frame, text="Ready")
|
|
||||||
self.status_label.pack()
|
|
||||||
|
|
||||||
self.search_frame = ttk.Frame(self.top_frame)
|
|
||||||
self.search_frame.pack()
|
|
||||||
self.download_frame = ttk.Frame(self.top_frame)
|
|
||||||
self.download_frame.pack()
|
|
||||||
|
|
||||||
self.search_status_label = tk.Label(self.search_frame, text="Enter search query:")
|
|
||||||
self.search_status_label.grid(row=0, column=0)
|
|
||||||
self.search_entry = tk.Entry(self.search_frame)
|
|
||||||
self.search_entry.grid(row=0, column=1)
|
|
||||||
self.search_button = tk.Button(self.search_frame, text="Search", command=lambda: self.search_and_display())
|
|
||||||
self.search_button.grid(row=0, column=2)
|
|
||||||
|
|
||||||
self.download_status_label = tk.Label(self.download_frame, text="Enter URL to download:")
|
|
||||||
self.download_status_label.grid(row=0, column=0)
|
|
||||||
self.download_entry = tk.Entry(self.download_frame)
|
|
||||||
self.download_entry.grid(row=0,column=1)
|
|
||||||
self.download_button = tk.Button(self.download_frame, text="Download", command=lambda: self.handle_download(self.download_entry.get()))
|
|
||||||
self.download_button.grid(row=0, column=2)
|
|
||||||
|
|
||||||
# Create the bottom part of the UI with tabs
|
|
||||||
self.bottom_frame = ttk.Frame(self.paned_window)
|
|
||||||
self.paned_window.add(self.bottom_frame)
|
|
||||||
|
|
||||||
self.notebook = ttk.Notebook(self.bottom_frame)
|
|
||||||
self.notebook.pack(fill='both', expand=True)
|
|
||||||
|
|
||||||
self.create_search_tab()
|
|
||||||
self.create_database_tab(db)
|
|
||||||
|
|
||||||
def create_search_tab(self):
|
|
||||||
search_tab = ttk.Frame(self.notebook)
|
|
||||||
self.notebook.add(search_tab, text='Search&Download')
|
|
||||||
self.status_label = tk.Label(search_tab, text="Ready")
|
|
||||||
self.status_label.pack()
|
|
||||||
|
|
||||||
# Create a frame for displaying search results
|
|
||||||
self.results_frame = ttk.Frame(search_tab)
|
|
||||||
self.results_frame.pack()
|
|
||||||
|
|
||||||
def create_database_tab(self, db):
|
|
||||||
database_tab = ttk.Frame(self.notebook)
|
|
||||||
self.notebook.add(database_tab, text='Database')
|
|
||||||
|
|
||||||
tree = ttk.Treeview(database_tab, columns=("title", "artist", "filename", "tags", "timestamp"))
|
|
||||||
tree.heading("#0", text="ID")
|
|
||||||
tree.heading("title", text="Title")
|
|
||||||
tree.heading("artist", text="Artist")
|
|
||||||
tree.heading("filename", text="Filename")
|
|
||||||
tree.heading("tags", text="Tags")
|
|
||||||
tree.heading("timestamp", text="Timestamp")
|
|
||||||
|
|
||||||
# Populate the treeview with data from the database
|
|
||||||
for entry in db.get_all_entries():
|
|
||||||
tree.insert("", "end", text=entry["id"], values=(entry["title"], entry["artist"], entry["filename"], ", ".join(entry["tags"]), entry["timestamp"]))
|
|
||||||
|
|
||||||
tree.pack(fill='both', expand=True)
|
|
||||||
|
|
||||||
|
|
||||||
def search_and_display(self):
|
|
||||||
self.set_status("searching")
|
|
||||||
query = self.search_entry.get()
|
|
||||||
# Clear the existing results
|
|
||||||
for widget in self.results_frame.winfo_children():
|
|
||||||
widget.destroy()
|
|
||||||
results = search_youtube(query)
|
|
||||||
self.display_results(results)
|
|
||||||
self.set_status("Enter search query or select item to download")
|
|
||||||
|
|
||||||
def set_status(self, status):
|
|
||||||
self.status_label.config(text=status)
|
|
||||||
self.root.update() # Force GUI update
|
|
||||||
|
|
||||||
# Modify the display_results method to use the results_frame in the search_tab
|
|
||||||
def display_results(self, results):
|
|
||||||
for i, result in enumerate(results):
|
|
||||||
if self.db.downloadable(result['webpage_url'], result['title']):
|
|
||||||
result_button = ttk.Button(self.results_frame, text=result['title'], command=lambda url=result['webpage_url'], filename=result['title']: self.handle_download(url, filename))
|
|
||||||
result_button.pack()
|
|
||||||
else:
|
|
||||||
result_button = tk.Label(self.results_frame, text=f"celaigia: {result['title']}")
|
|
||||||
result_button.pack()
|
|
||||||
|
|
||||||
def handle_download(self, url, filename):
|
|
||||||
|
|
||||||
if self.db.downloadable(url, filename):
|
|
||||||
self.set_status("Downloading...")
|
|
||||||
success = download_audio(url, self.db.base_path)
|
|
||||||
if success:
|
|
||||||
self.set_status("Download completed, adding to db")
|
|
||||||
self.prompt_user_for_details(url, filename)
|
|
||||||
self.set_status("Enter search query or select item to download")
|
|
||||||
|
|
||||||
# Modify the prompt_user_for_details method
|
|
||||||
def prompt_user_for_details(self,url, filename):
|
|
||||||
title = sd.askstring("Enter Title", "Enter Title:", initialvalue=filename)
|
|
||||||
artist = sd.askstring("Enter Artist", "Enter Artist:")
|
|
||||||
tags = sd.askstring("Enter Tags", "Enter Tags (comma separated):")
|
|
||||||
entry = {
|
|
||||||
"id": url.split('https://www.youtube.com/watch?v=')[-1].strip(),
|
|
||||||
"title": title.strip(),
|
|
||||||
"artist": artist.strip(),
|
|
||||||
"filename": filename.strip(),
|
|
||||||
"tags": [tag.strip() for tag in tags.split(",")] if tags else [],
|
|
||||||
"timestamp": time.time()
|
|
||||||
}
|
|
||||||
self.add_to_db(entry)
|
|
||||||
|
|
||||||
def add_to_db(self, entry):
|
|
||||||
self.db.add_entry(entry)
|
|
||||||
|
|
||||||
def run(self):
|
|
||||||
self.root.mainloop()
|
|
@ -1,28 +0,0 @@
|
|||||||
import yt_dlp
|
|
||||||
from config import yt_dlp_opts
|
|
||||||
|
|
||||||
|
|
||||||
# Function to search YouTube
|
|
||||||
def search_youtube(query, opts = yt_dlp_opts):
|
|
||||||
with yt_dlp.YoutubeDL(opts) as ydl:
|
|
||||||
info = ydl.extract_info(query, download=False)
|
|
||||||
return info['entries']
|
|
||||||
|
|
||||||
|
|
||||||
def download_audio(url, base_path, opts = yt_dlp_opts):
|
|
||||||
|
|
||||||
opts['outtmpl'] = f'{base_path}/audios/%(title)s.%(ext)s'
|
|
||||||
if not isinstance(url, list):
|
|
||||||
url = [url]
|
|
||||||
|
|
||||||
success = False
|
|
||||||
|
|
||||||
try:
|
|
||||||
with yt_dlp.YoutubeDL(opts) as ydl:
|
|
||||||
ydl.download(url)
|
|
||||||
success = True
|
|
||||||
except:
|
|
||||||
success = False
|
|
||||||
return success
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue