Old data for migrate + scripts

nxy/migrate.py

@@ -1,11 +1,12 @@
 # -*- coding: utf-8 -*-
 import sqlite3
+import sys
+
 import os
 
-DATA = 'data'
-MIGRATE = os.path.join(DATA, 'migrate')
+MIGRATE = 'migrate'
 
-db = sqlite3.connect(os.path.join(DATA, 'nxy.db'))
+db = None
 
 
 def read_file(file):
@@ -19,5 +20,11 @@ def batch_insert(table, column, data):
     db.commit()
 
 
-batch_insert('mcmaniacs', 'item', read_file('mcmaniac.txt'))
-batch_insert('yiffs', 'item', read_file('yiff.txt'))
+def main(file):
+    db = sqlite3.connect(file)
+    batch_insert('mcmaniacs', 'item', read_file('mcmaniac.txt'))
+    batch_insert('yiffs', 'item', read_file('yiff.txt'))
+
+
+if __name__ == '__main__':
+    main('db.sqlite' if len(sys.argv) is 0 else sys.argv[0])