Cloned and modified getmusic.py script for Magisk module(for pyinstaller)

localwrapped-pyi.py

@@ -0,0 +1,127 @@
+import subprocess
+import re
+import time
+import sqlite3
+import os
+
+def get_working_directory():
+    # If you see this function, that means you are using the pyinstaller. Use this command to create the executable:
+    # pyinstaller -s -F localwrapped-pyi.py --runtime-tmpdir /data/local/tmp
+    if os.path.exists("/sdcard/localwrapped"):
+        return "/sdcard/localwrapped"
+    elif os.path.exists("/storage/emulated/0/localwrapped"):
+        return "/storage/emulated/0/localwrapped"
+    elif os.path.exists("/data/media/0/localwrapped"):
+        return "/data/media/0/localwrapped"
+    elif os.path.exists("/mnt/user/0/emulated/0/localwrapped"):
+        return "/mnt/user/0/emulated/0/localwrapped"
+    else:
+        print("Path not found.")
+        try:
+            if os.path.exists("/sdcard"):
+                os.mkdir("/sdcard/localwrapped")
+                return "/sdcard/localwrapped"
+            else:
+                raise Exception
+        except:
+            try:
+                if os.path.exists("/storage/emulated/0"):
+                    os.mkdir("/storage/emulated/0/localwrapped")
+                    return "/storage/emulated/0/localwrapped"
+                else:
+                    raise Exception
+            except:
+                try:
+                    if os.path.exists("/data/media/0"):
+                        os.mkdir("/data/media/0/localwrapped")
+                        return "/data/media/0/localwrapped"
+                    else:
+                        raise Exception
+                except:
+                    try:
+                        if os.path.exists("/mnt/user/0/emulated/0"):
+                            os.mkdir("/mnt/user/0/emulated/0/localwrapped")
+                            return "/mnt/user/0/emulated/0/localwrapped"
+                        else:
+                            raise Exception
+                    except:
+                        print("Path not found and could not be created.")
+                        exit()
+
+def get_music():
+    dumpsys_output = subprocess.check_output("dumpsys media_session", shell=True).decode("utf-8").split("\n")
+    if not [line for line in dumpsys_output if "PLAYING" in line]:
+        return None
+    else:
+        description = [line for line in dumpsys_output if "description=" in line]
+        try:
+            return description[0].split("description=")[1]
+        except:
+            return None
+
+def get_db_values():
+    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],
+            "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
+while True:
+    if get_music() == None:
+        continue
+    last_music = get_music()
+    time.sleep(3)
+    if last_music == get_music():
+        if last_written == last_music:
+            continue
+        a += 1
+        if a == 20:
+            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:
+                    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()
+            last_written = last_music
+            a = 0
+        continue
+    else:
+        a = 0
+        continue
+