aboutsummaryrefslogtreecommitdiff
path: root/scripts
diff options
context:
space:
mode:
Diffstat (limited to 'scripts')
-rw-r--r--scripts/schema1-to-2.py84
1 files changed, 84 insertions, 0 deletions
diff --git a/scripts/schema1-to-2.py b/scripts/schema1-to-2.py
new file mode 100644
index 0000000..ac74789
--- /dev/null
+++ b/scripts/schema1-to-2.py
@@ -0,0 +1,84 @@
+import sqlite3
+import vgstash as vgs
+import datetime as dt
+import argparse
+
+# Migrate VGStash schema v1 to schema v2:
+# + purchase_date : UNIX timestamp, date of a game's purchase
+# + beaten_date : UNIX timestamp, date a game was beaten
+# + complete_date : UNIX timestamp, date a game was completed 100%
+
+parser = argparse.ArgumentParser(
+ prog="schema1-to-2",
+ description="Migrate a VGStash database, in-place, from v1 to v2. This adds three columns and moves header data."
+ )
+
+parser.add_argument(
+ "dbfile",
+ help="path to a VGStash database file",
+ )
+
+opts = parser.parse_args()
+
+vgdb = vgs.DB(path=opts.dbfile)
+
+# Step 1: Add columns
+columns = [
+ ["Purchased: ", "p_date"],
+ ["Beaten: ", "b_date"],
+ ["Completed: ", "c_date"],
+ ]
+for c in columns:
+ try:
+ print("Adding '{}' column...".format(c[1]))
+ # This is RISKY and RIPE FOR INJECTION. SQLite does not support
+ # parameterized column names during creation; this is a workaround.
+ vgdb.conn.execute("ALTER TABLE games ADD COLUMN {} INTEGER DEFAULT NULL".format(c[1]))
+ vgdb.conn.commit()
+ print("Success!")
+ except sqlite3.OperationalError:
+ print("Column {} already exists in the database. Skipping.".format(c[1]))
+
+# Step 2: Fetch notes
+res = vgdb.conn.execute("SELECT rowid,notes FROM games WHERE notes NOT LIKE ''")
+targets = res.fetchall()
+converted_lines = 0
+removed_notes = 0
+for row in targets:
+ lines = row['notes'].splitlines()
+ newlines = []
+ # Step 3: Check for headers in notes
+ for l in lines:
+ has_header = False
+ for i in columns:
+ if l.startswith(i[0]):
+ # Step 4: Convert and add data to new column location
+ has_header = True
+ isodate = l.split(":")[1].strip()
+ unixdate = dt.date.fromisoformat(isodate).strftime("%s")
+ # Same warning applies to ''.format usage and SQL queries
+ set_ts = vgdb.conn.execute("UPDATE games SET {} = ? WHERE rowid = ?".format(i[1]), (unixdate, row['rowid']))
+ vgdb.conn.commit()
+ if (set_ts.rowcount > 0):
+ converted_lines+= 1
+ continue
+
+ if not has_header:
+ newlines.append(l)
+ if len(newlines) > 0:
+ for l in newlines:
+ if len(l) == 0:
+ newlines.remove(l)
+ else:
+ break
+
+ # Step 5: Update notes, if any lines were removed in Step 4.
+ if (len(newlines) < len(lines)):
+ set_notes = vgdb.conn.execute("UPDATE games SET notes = ? WHERE rowid = ?", ("\n".join(newlines), row['rowid']))
+ vgdb.conn.commit()
+ if (set_notes.rowcount > 0):
+ removed_notes+=1
+
+# Step 6: Summarize what was done.
+print("{} games' notes were updated.".format(removed_notes))
+print("{} details converted from Notes header to Column.".format(converted_lines))