fix imports in dbsearch (no need to import the whole thing), rename it so we have something uniform

1 files changed, 131 insertions, 0 deletions

diff --git a/src/backend/searchbinary.py b/src/backend/searchbinary.py
new file mode 100644
index 0000000..0305a14
--- /dev/null
+++ b/src/backend/searchbinary.py
@@ -0,0 +1,131 @@
+#!/usr/bin/python3
+
+import os
+import sisyphus.filesystem
+import sisyphus.update
+import sqlite3
+
+def searchDB(filter, cat = '', pn = '', desc = ''):
+    NOVIRT = "AND cat NOT LIKE 'virtual'"
+    SELECTS = {
+        'any': f'''SELECT
+                    i.category AS cat,
+                    i.name as pn,
+                    i.version as iv,
+                    IFNULL(a.version, 'alien') AS av,
+                    d.description AS desc
+                    FROM local_packages AS i LEFT OUTER JOIN remote_packages as a
+                    ON i.category = a.category
+                    AND i.name = a.name
+                    AND i.slot = a.slot
+    				LEFT JOIN remote_descriptions AS d ON i.name = d.name AND i.category = d.category
+                    WHERE cat LIKE '%{cat}%' AND pn LIKE '%{pn}%' AND desc LIKE '%{desc}%' {NOVIRT}
+                    UNION
+                    SELECT
+                    a.category AS cat,
+                    a.name as pn,
+                    IFNULL(i.version, 'None') AS iv,
+                    a.version as av,
+                    d.description AS desc
+                    FROM remote_packages AS a LEFT OUTER JOIN local_packages AS i
+                    ON a.category = i.category
+                    AND a.name = i.name
+                    AND a.slot = i.slot
+    				LEFT JOIN remote_descriptions AS d ON a.name = d.name AND a.category = d.category
+                    WHERE cat LIKE '%{cat}%' AND pn LIKE '%{pn}%' AND desc LIKE '%{desc}%' {NOVIRT}''',
+        'installed': f'''SELECT
+                    i.category AS cat,
+                    i.name AS pn,
+                    i.version AS iv,
+                    a.version as av,
+                    d.description AS desc
+                    FROM local_packages AS i
+                    LEFT JOIN remote_packages AS a
+                    ON i.category = a.category
+                    AND i.name = a.name
+                    AND i.slot = a.slot
+    				LEFT JOIN remote_descriptions AS d ON i.name = d.name AND i.category = d.category
+                    WHERE cat LIKE '%{cat}%' AND pn LIKE '%{pn}%' AND desc LIKE '%{desc}%' {NOVIRT}''',
+        'alien': f'''SELECT
+                        i.category AS cat,
+                        i.name AS pn,
+                        i.version as iv,
+                        IFNULL(a.version, 'alien') AS av,
+                        d.description AS desc
+                        FROM local_packages AS i
+                        LEFT JOIN remote_packages AS a
+                        ON a.category = i.category
+                        AND a.name = i.name
+                        AND a.slot = i.slot
+        				LEFT JOIN remote_descriptions AS d ON i.name = d.name AND i.category = d.category
+                        WHERE cat LIKE '%{cat}%' AND pn LIKE '%{pn}%' AND desc LIKE '%{desc}%' {NOVIRT}
+                        AND av IS 'alien' ''',
+        'remote': f'''SELECT
+                    a.category AS cat,
+                    a.name AS pn,
+                    i.version as iv,
+                    a.version AS av,
+                    d.description AS desc
+                    FROM remote_packages AS a
+                    LEFT JOIN local_packages AS i
+                    ON a.category = i.category
+                    AND a.name = i.name
+                    AND a.slot = i.slot
+    				LEFT JOIN remote_descriptions AS d ON a.name = d.name AND a.category = d.category
+                    WHERE cat LIKE '%{cat}%' AND pn LIKE '%{pn}%' AND desc LIKE '%{desc}%' {NOVIRT}
+                    AND iv IS NULL''',
+        'upgrade': f'''SELECT
+                    i.category AS cat,
+                    i.name AS pn,
+                    i.version as iv,
+                    a.version AS av,
+                    d.description AS desc
+                    FROM local_packages AS i
+                    INNER JOIN remote_packages AS a
+                    ON i.category = a.category
+                    AND i.name = a.name
+                    AND i.slot = a.slot
+    				LEFT JOIN remote_descriptions AS d ON i.name = d.name AND i.category = d.category
+                    WHERE cat LIKE '%{cat}%' AND pn LIKE '%{pn}%' AND desc LIKE '%{desc}%' {NOVIRT}
+                    AND iv <> av'''
+    }
+
+    with sqlite3.connect(sisyphus.filesystem.localDatabase) as db:
+        db.row_factory = sqlite3.Row
+        cursor = db.cursor()
+        cursor.execute(SELECTS[filter])
+        rows = cursor.fetchall()
+
+    return rows
+
+def tosql(string):
+    return '%%' if string == '' else string.replace('*', '%').replace('?', '_')
+
+def start(filter, cat, pn, desc, single = False):
+    if os.getuid() == 0:
+        sisyphus.update.start()
+    else:
+        print('You are not root, cannot fetch updates.\nSearch result may be inaccurate!')
+    print(f"Looking for {filter} matching packages ...\n")
+    pkglist = searchDB(filter, tosql(cat), tosql(pn), tosql(desc))
+
+    if len(pkglist) == 0:
+        print("No binary package found!")
+    else:
+        if single:
+            print(f"{'Package':45} {'Installed':20} Available")
+        for pkg in pkglist:
+            if not single:
+                print(f"*  {pkg['cat']}/{pkg['pn']}")
+                print(f"\tInstalled version: {pkg['iv']}")
+                if pkg['av'] != 'alien':
+                    print(f"\tLatest available version: {pkg['av']}")
+                else:
+                    print(f"\tAlien package, use `sisyphus search --ebuild {pkg['pn']}` for available version")
+                print(f"\tDescription: {pkg['desc']}\n")
+            else:
+                cpn = f"{pkg['cat']}/{pkg['pn']}"
+                print(f"{cpn:45} {str(pkg['iv']):20} {str(pkg['av'])}")
+        print(f"\nFound {len(pkglist)} binary package(s)")
+
+    print("To search for source packages, use the '--ebuild' option.")