updates and backlog

2026-08-08 11:58:01 +02:00
parent e4c0590a81
commit 7da0c922ad
9 changed files with 444 additions and 18 deletions
@@ -0,0 +1,63 @@
+#!/usr/bin/python
+
+import argparse
+import shutil
+import sqlite3
+import tempfile
+import os
+
+def main():
+    parser = argparse.ArgumentParser(description='Command line arguments')
+    parser.add_argument('-d', '--db', required=True, help='database location')
+    parser.add_argument('-t', '--tag', type=int, required=False, help='tag id')
+    parser.add_argument('-n', '--number', type=int, default=4, help='number')
+    args = parser.parse_args()
+
+    # Copy db to temp location
+    temp_dir = tempfile.gettempdir()
+    temp_db = os.path.join(temp_dir, 'pmvg.db')
+    if not os.path.exists(temp_db):
+        shutil.copy2(args.db, temp_db)
+
+    # Open temp db with sqlite
+    conn = sqlite3.connect(temp_db)
+    cursor = conn.cursor()
+
+    # Print names of all tables
+    cursor.execute("SELECT name FROM sqlite_master WHERE type='table'")
+    tables = cursor.fetchall()
+    print("Tables:")
+    for table in tables:
+        print(table[0])
+    print()
+
+    # Print schema of scenes_tags table
+    cursor.execute("SELECT sql FROM sqlite_master WHERE type='table' AND name='scenes_tags'")
+    schema = cursor.fetchone()
+    print("Schema for scenes_tags:")
+    print(schema[0])
+    print()
+
+    # Print data of tags table if tag id not provided
+    if not args.tag:
+        cursor.execute("SELECT * FROM tags")
+        tags = cursor.fetchall()
+        print("Table: tags")
+        for tag in tags:
+            print(tag)
+        print("Select a tag with -t")
+        exit()
+
+    # Get random scenes based on tag id
+    cursor.execute(f"SELECT * FROM scenes_tags WHERE tag_id = {args.tag} ORDER BY RANDOM() LIMIT {args.number}")
+    scenes = cursor.fetchall()
+    print(f"\nRandom {args.number} scenes with tag_id {args.tag}:")
+    for scene in scenes:
+        scene_id = scene[0]
+        url = f"http://localhost:9999/scene/{scene_id}/stream"
+        os.system(f"nohup mpv --volume=0 {url} </dev/null >/dev/null 2>&1 &")
+        print(scene)
+    os._exit(0)
+
+if __name__ == '__main__':
+    main()