First self-hosted release

addon/hosted/shared.py

@@ -0,0 +1,73 @@
+import sqlite3
+import re
+import os
+import json
+from imdb import Cinemagoer
+
+SQLITE_PATH = "/sqlite/torrentio.sqlite"
+
+ia = Cinemagoer()
+CONFIG = "config.json"
+
+def read_config(provider, key):
+    if os.path.isfile(CONFIG):
+        f = open(CONFIG, "r")
+        cfg = json.load(f)
+        return cfg[provider][key]
+
+def filter_file(file):
+    allowed_ext = ['.mp4', '.mkv', '.avi', '.mpeg', '.mpg', '.mpv', '.mov']
+    if os.path.splitext(file)[1] in allowed_ext:
+        return True
+    return False
+
+def create_connection(db_file):
+    conn = None
+    try:
+        conn = sqlite3.connect(db_file, check_same_thread=False)
+    except Exception as e:
+        print(e)
+        exit(1)
+    return conn
+
+sqlite = create_connection(SQLITE_PATH)
+
+def build_and_write(torrent):
+    try:
+        print(f"Recording {torrent['title']} in the database")  
+        q = f"INSERT OR REPLACE INTO torrents (infoHash, provider, title, size, type, uploadDate, seeders, trackers) VALUES (?,?,?,?,?,?,?,?)"
+        p = (torrent['infoHash'],'1337x',torrent['title'],torrent['size'],'movie','1/1/2024',torrent['seeders'],','.join(torrent['trackers']))
+        cursor = sqlite.cursor()
+        cursor.execute(q,p)
+        for file in torrent['files']:
+            if filter_file(file):
+                q = f"INSERT OR REPLACE INTO files (infoHash, fileIndex, title, size, imdbId) VALUES (?,?,?,?,?)"
+                p = (torrent['infoHash'], torrent['files'].index(file), file, torrent['size'], torrent['imdbid'])
+                cursor.execute(q,p)
+        sqlite.commit()
+        cursor.close()
+    except sqlite3.Error as error:
+        print(error)
+
+def imdb_find(name):
+    movie = ia.search_movie(name)
+    if len(movie) >= 1:
+        return movie[0].movieID
+    return None
+
+def extract_title(filename):
+    try:
+        filename.strip()
+        filename = filename.replace('.', ' ')
+        res = re.search('([^\\\]+)\.(avi|mkv|mpeg|mpg|mov|mp4)$', filename)
+        if res:
+            filename = res.group(1)
+        res = re.search('(.*?)(dvdrip|xvid| cd[0-9]|dvdscr|brrip|divx|[\{\(\[]?[0-9]{4}).*', filename)
+        if res:
+            filename = res.group(1)
+        res = re.search('(.*?)\(.*\)(.*)', filename)
+        if res:
+            filename = res.group(1)
+        return filename
+    except:
+        return ""