Initial
This commit is contained in:
commit
63989dbf58
3 changed files with 248 additions and 0 deletions
169
combolinux.py
Normal file
169
combolinux.py
Normal file
|
@ -0,0 +1,169 @@
|
|||
#!/bin/env python3
|
||||
import subprocess
|
||||
import re
|
||||
import time
|
||||
import sqlite3
|
||||
import os
|
||||
import requests
|
||||
|
||||
def get_working_directory():
|
||||
if os.path.exists("/userdata/media/0/localwrapped"):
|
||||
return "/userdata/media/0/localwrapped"
|
||||
elif os.path.exists(os.environ["HOME"] + "/localwrapped"):
|
||||
return os.environ["HOME"] + "/localwrapped"
|
||||
else:
|
||||
print("Path not exist")
|
||||
try:
|
||||
if os.path.exists("/userdata/media/0"):
|
||||
os.makedirs("/userdata/media/0/localwrapped")
|
||||
return "/userdata/media/0/localwrapped"
|
||||
elif os.path.exists(os.environ["HOME"]):
|
||||
os.makedirs(os.environ["HOME"] + "/localwrapped")
|
||||
return os.environ["HOME"] + "/localwrapped"
|
||||
except:
|
||||
print("Can't create path")
|
||||
exit()
|
||||
|
||||
if not os.path.exists(f"{get_working_directory()}/key.txt"):
|
||||
print("Key not exist")
|
||||
exit()
|
||||
|
||||
def get_verification_code():
|
||||
try:
|
||||
with open(f"{get_working_directory()}/key.txt", "r") as f:
|
||||
key = f.read()
|
||||
try:
|
||||
verifyrequest = requests.post(os.environ["MSSERVER"] + "/verifykey", json={"key": key})
|
||||
if verifyrequest.text == "0":
|
||||
print("Verification failed")
|
||||
return None
|
||||
elif len(verifyrequest.text) == 64:
|
||||
print("Verification success")
|
||||
return verifyrequest.text
|
||||
else:
|
||||
print("Unexpected response")
|
||||
return None
|
||||
except:
|
||||
print("Can't reach the server.")
|
||||
return None
|
||||
except:
|
||||
print("Key not exist")
|
||||
return None
|
||||
|
||||
def get_music():
|
||||
try:
|
||||
metadata = subprocess.check_output(
|
||||
["playerctl", "metadata", "--format", "'{{artist}}|||{{title}}|||{{album}}'"],
|
||||
stderr=subprocess.DEVNULL,
|
||||
timeout=2
|
||||
).decode('utf-8').strip().replace("'", "")
|
||||
|
||||
if not metadata or "No players found" in metadata:
|
||||
return None
|
||||
|
||||
artist, title, album = [part.strip() or "Unknown" for part in metadata.split("|||")]
|
||||
return f"{title}, {artist}, {album}"
|
||||
|
||||
except subprocess.CalledProcessError:
|
||||
return None
|
||||
except subprocess.TimeoutExpired:
|
||||
print("Playerctl timeout, media player busy?")
|
||||
return None
|
||||
except Exception as e:
|
||||
print(f"Metadata parse error: {str(e)}")
|
||||
return None
|
||||
|
||||
def get_db_values(url: str = "https://halhadus.rocks/assets/localwrapped/music.db"):
|
||||
if not os.path.exists(get_working_directory() + "/music.db"):
|
||||
try:
|
||||
download = requests.get(url)
|
||||
with open(get_working_directory() + "/music.db", "wb") as f:
|
||||
f.write(download.content)
|
||||
except:
|
||||
print("Could not download the database.")
|
||||
exit()
|
||||
musicconn = sqlite3.connect(get_working_directory() + "/music.db")
|
||||
musicc = musicconn.cursor()
|
||||
musicc.execute("SELECT * FROM music")
|
||||
musicrows = musicc.fetchall()
|
||||
musicconn.close()
|
||||
musicvalues = []
|
||||
for row in musicrows:
|
||||
musicvalues.append({
|
||||
"listmusicname": row[0],
|
||||
"listvideoid": row[1],
|
||||
"fileartistname": row[4],
|
||||
"filealbumname": row[5],
|
||||
"filemusictitle": row[6]
|
||||
})
|
||||
return musicvalues
|
||||
|
||||
if not os.path.exists(get_working_directory() + "/count.db"):
|
||||
countconn = sqlite3.connect(get_working_directory() + "/count.db")
|
||||
countc = countconn.cursor()
|
||||
countc.execute("CREATE TABLE count (musicname TEXT, count INTEGER)")
|
||||
countconn.commit()
|
||||
countconn.close()
|
||||
|
||||
a = 0
|
||||
last_written = None
|
||||
last_sent = None
|
||||
resetstatus = "true"
|
||||
while True:
|
||||
if get_music() == None:
|
||||
continue
|
||||
last_music = get_music()
|
||||
time.sleep(5)
|
||||
if last_music == get_music():
|
||||
if last_written == last_music:
|
||||
continue
|
||||
a += 1
|
||||
if a >= 3:
|
||||
for i in get_db_values():
|
||||
if i["filealbumname"] == None:
|
||||
full_name = i["filemusictitle"] + ", " + i["fileartistname"] + ", " + "Music"
|
||||
else:
|
||||
full_name = i["filemusictitle"] + ", " + i["fileartistname"] + ", " + i["filealbumname"]
|
||||
if full_name == last_music:
|
||||
if a == 12:
|
||||
countconn = sqlite3.connect(get_working_directory() + "/count.db")
|
||||
countc = countconn.cursor()
|
||||
countc.execute("SELECT * FROM count")
|
||||
countdbrows = countc.fetchall()
|
||||
countdbvalues = []
|
||||
for row in countdbrows:
|
||||
countdbvalues.append({
|
||||
"musicname": row[0],
|
||||
"count": row[1]
|
||||
})
|
||||
if not [row for row in countdbvalues if row["musicname"] == i["listmusicname"]]:
|
||||
countc.execute("INSERT INTO count VALUES (?, ?)", (i["listmusicname"], 1))
|
||||
countconn.commit()
|
||||
countconn.close()
|
||||
else:
|
||||
countc.execute("UPDATE count SET count = count + 1 WHERE musicname = ?", (i["listmusicname"],))
|
||||
countconn.commit()
|
||||
countconn.close()
|
||||
print(f"Local Wrapped: {i['listmusicname']} has been played.")
|
||||
last_written = last_music
|
||||
a = 0
|
||||
else:
|
||||
if last_sent == i["listmusicname"]:
|
||||
continue
|
||||
try:
|
||||
verificationcode = get_verification_code()
|
||||
if verificationcode == None:
|
||||
continue
|
||||
requests.post(os.environ["MSSERVER"] + "/musicstatus.html", json={"music": i["listmusicname"], "time": str(int(time.time())),"videoid":i["listvideoid"], "verificationcode": verificationcode, "resetstatus": resetstatus})
|
||||
print("Data sent")
|
||||
resetstatus = "false"
|
||||
a += 1
|
||||
last_sent = i["listmusicname"]
|
||||
except:
|
||||
print("Can't reach the server.")
|
||||
a += 1
|
||||
continue
|
||||
else:
|
||||
a = 0
|
||||
continue
|
||||
|
17
combolinux.service
Normal file
17
combolinux.service
Normal file
|
@ -0,0 +1,17 @@
|
|||
[Unit]
|
||||
Description=Combo Music
|
||||
After=network.target sound.target
|
||||
Requires=dbus.socket
|
||||
|
||||
[Service]
|
||||
Type=simple
|
||||
ExecStart=/usr/local/bin/combomusic
|
||||
Environment="DISPLAY=:0"
|
||||
Environment="DBUS_SESSION_BUS_ADDRESS=unix:path=/run/user/%U/bus"
|
||||
Restart=always
|
||||
RestartSec=10
|
||||
StandardOutput=journal
|
||||
StandardError=journal
|
||||
|
||||
[Install]
|
||||
WantedBy=default.target
|
62
setup.sh
Normal file
62
setup.sh
Normal file
|
@ -0,0 +1,62 @@
|
|||
#!/bin/bash
|
||||
|
||||
if [ "$EUID" -ne 0 ]; then
|
||||
id
|
||||
exit
|
||||
fi
|
||||
|
||||
if [ ! -d /usr/local/bin ]; then
|
||||
mkdir -p /usr/local/bin
|
||||
fi
|
||||
|
||||
if [[ ":$PATH:" != *":/usr/local/bin:"* ]]; then
|
||||
export PATH="/usr/local/bin:$PATH"
|
||||
echo 'export PATH="/usr/local/bin:$PATH"' >> ~/.bashrc
|
||||
fi
|
||||
|
||||
if ! python3 -c "import requests" &> /dev/null; then
|
||||
if command -v apt-get &> /dev/null; then
|
||||
apt-get update
|
||||
apt-get install -y python3-requests
|
||||
elif command -v pacman &> /dev/null; then
|
||||
pacman -Syu python-requests
|
||||
elif command -v dnf &> /dev/null; then
|
||||
dnf install -y python3-requests
|
||||
elif commamd -v pip &> /dev/null; then
|
||||
pip install --user --break-system-packages requests
|
||||
elif command -v pip3 &> /dev/null; then
|
||||
pip3 install --user --break-system-packages requests
|
||||
else
|
||||
echo "requests"
|
||||
exit 1
|
||||
fi
|
||||
fi
|
||||
|
||||
if ! command -v playerctl &> /dev/null; then
|
||||
if command -v apt-get &> /dev/null; then
|
||||
apt-get update
|
||||
apt-get install -y playerctl
|
||||
elif command -v pacman &> /dev/null; then
|
||||
pacman -Syu playerctl
|
||||
elif command -v dnf &> /dev/null; then
|
||||
dnf install -y playerctl
|
||||
else
|
||||
echo "playerctl"
|
||||
exit 1
|
||||
fi
|
||||
fi
|
||||
|
||||
if [ ! -f /usr/local/bin/combolinux ]; then
|
||||
install -m 755 combolinux.py /usr/local/bin/combolinux
|
||||
fi
|
||||
|
||||
if [ -f /usr/local/bin/combolinux ]; then
|
||||
if command -v systemctl &> /dev/null; then
|
||||
install -m 644 combolinux.service /etc/systemd/user/combolinux.service
|
||||
systemctl --user enable --now combolinux
|
||||
else
|
||||
exit 1
|
||||
fi
|
||||
fi
|
||||
|
||||
exit 0
|
Loading…
Add table
Reference in a new issue