edg-l/ddnet
1
0
Fork 0
mirror of https://github.com/ddnet/ddnet.git synced 2024-11-10 01:58:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

SQLite transfer script: Warn if database does not exist yet and handle partially created sqlite tables

Browse source
This commit is contained in:
Zwelf 2020-08-12 23:23:47 +02:00
parent 4928d52bc9
commit 9848b3c759
1 changed files with 16 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
scripts/move_sqlite.py
View file

@ -17,8 +17,14 @@ import argparse
from time import strftime
import os
def sqlite_table_exists(cursor, table):
cursor.execute("SELECT COUNT(*) FROM sqlite_master WHERE type='table' AND name='{}'".format(table))
return cursor.fetchone()[0] != 0
def sqlite_num_transfer(conn, table):
c = conn.cursor()
if not sqlite_table_exists(c, table):
return 0
c.execute('SELECT COUNT(*) FROM {}'.format(table))
num = c.fetchone()[0]
return num
@ -36,9 +42,12 @@ def transfer(file_from, file_to):
conn_to.close()
cursor_from = conn_from.cursor()
cursor_from.execute('DELETE FROM record_race')
cursor_from.execute('DELETE FROM record_teamrace')
cursor_from.execute('DELETE FROM record_saves')
if sqlite_table_exists(cursor_from, 'record_race'):
cursor_from.execute('DELETE FROM record_race')
if sqlite_table_exists(cursor_from, 'record_teamrace'):
cursor_from.execute('DELETE FROM record_teamrace')
if sqlite_table_exists(cursor_from, 'record_saves'):
cursor_from.execute('DELETE FROM record_saves')
cursor_from.close()
conn_from.commit()
conn_from.close()
@ -56,6 +65,10 @@ def main():
help='Output file where ranks are saved adds current date by default')
args = parser.parse_args()
if not os.path.exists(args.f):
print("Warning: '{}' does not exist (yet). Is the path specified correctly?".format(args.f))
return
conn = sqlite3.connect(args.f)
num_ranks = sqlite_num_transfer(conn, 'record_race')
num_teamranks = sqlite_num_transfer(conn, 'record_teamrace')