From e6345cbd4198c7821106821b0a4567f4d4563969 Mon Sep 17 00:00:00 2001
From: norohind <60548839+norohind@users.noreply.github.com>
Date: Sun, 28 Nov 2021 19:21:12 +0300
Subject: [PATCH] sqlite2postgres: cli interface

---
 sqlite2postgres.py | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

diff --git a/sqlite2postgres.py b/sqlite2postgres.py
index 0f179f6..c78d159 100644
--- a/sqlite2postgres.py
+++ b/sqlite2postgres.py
@@ -8,6 +8,7 @@ import sqlite3
 from model import postgres_sql_requests
 import time
 import config
+import sys
 
 insert_pg = """insert into squads_stats_states (action_id, leaderboard_type, platform, squadron_id, score, 
 percentile, rank, name, tag, timestamp) 
@@ -74,7 +75,26 @@ def continues_pull_sqlite_to_postgres() -> None:
             cursor.executemany(insert_pg, new_records)
 
 
-continues_pull_sqlite_to_postgres()
+cli_error = 'argument must be "initial" or "continues"'
+
+if len(sys.argv) != 2:
+    print(cli_error)
+    pg_conn.close()
+    sqlite_conn.close()
+    exit(1)
+
+if sys.argv[1] == 'initial':
+    initial_pull_sqlite_to_postgres()
+    pg_conn.close()
+    sqlite_conn.close()
+
+elif sys.argv[1] == 'continues':
+    continues_pull_sqlite_to_postgres()
+    pg_conn.close()
+    sqlite_conn.close()
+
+else:
+    print(cli_error)
 
 
 pg_conn.close()