move syncDatabase -> syncdb

1 files changed, 57 insertions, 0 deletions

diff --git a/src/backend/syncdb.py b/src/backend/syncdb.py
new file mode 100644
index 0000000..b5d86ae
--- /dev/null
+++ b/src/backend/syncdb.py
@@ -0,0 +1,57 @@
+#!/usr/bin/python3
+
+import csv
+import shutil
+import urllib3
+import sqlite3
+import subprocess
+import sisyphus.getenv
+import sisyphus.getfs
+
+def remoteCSV():
+    packagesCsvURL,descriptionsCsvURL = sisyphus.getenv.csvURL()
+    http = urllib3.PoolManager()
+
+    with http.request('GET', packagesCsvURL, preload_content=False) as tmp_buffer, open(sisyphus.getfs.remotePackagesCsv, 'wb') as output_file:
+        shutil.copyfileobj(tmp_buffer, output_file)
+
+    with http.request('GET', descriptionsCsvURL, preload_content=False) as tmp_buffer, open(sisyphus.getfs.remoteDescriptionsCsv, 'wb') as output_file:
+        shutil.copyfileobj(tmp_buffer, output_file)
+
+def localCSV():
+    subprocess.call(['/usr/share/sisyphus/helpers/make_local_csv'])
+
+def remoteTable():
+    remoteCSV()
+
+    sisyphusdb = sqlite3.connect(sisyphus.getfs.localDatabase)
+    sisyphusdb.cursor().execute('''drop table if exists remote_packages''')
+    sisyphusdb.cursor().execute('''drop table if exists remote_descriptions''')
+    sisyphusdb.cursor().execute('''create table remote_packages (category TEXT,name TEXT,version TEXT,slot TEXT)''')
+    sisyphusdb.cursor().execute('''create table remote_descriptions (category TEXT,name TEXT,description TEXT)''')
+
+    with open(sisyphus.getfs.remotePackagesCsv) as input_file:
+        for row in csv.reader(input_file):
+            sisyphusdb.cursor().execute("insert into remote_packages (category, name, version, slot) values (?, ?, ?, ?);", row)
+
+    with open(sisyphus.getfs.remoteDescriptionsCsv) as input_file:
+        for row in csv.reader(input_file):
+            sisyphusdb.cursor().execute("insert into remote_descriptions (category, name, description) values (?, ?, ?);", row)
+
+    sisyphusdb.commit()
+    sisyphusdb.close()
+
+def localTable():
+    localCSV()
+
+    sisyphusdb = sqlite3.connect(sisyphus.getfs.localDatabase)
+    sisyphusdb.cursor().execute('''drop table if exists local_packages''')
+    sisyphusdb.cursor().execute('''create table local_packages (category TEXT,name TEXT,version TEXT,slot TEXT)''')
+
+    with open(sisyphus.getfs.localPackagesCsv) as input_file:
+        for row in csv.reader(input_file):
+            sisyphusdb.cursor().execute("insert into local_packages (category, name, version, slot) values (?, ?, ?, ?);", row)
+
+    sisyphusdb.commit()
+    sisyphusdb.close()
+