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

Adapt move_sqlite.py script to _backup tables

Browse source
This commit is contained in:
Dennis Felsing 2022-12-07 16:36:48 +01:00
parent 3438e272af
commit f4517c1c1d
1 changed files with 18 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
scripts/move_sqlite.py
View file

@ -16,6 +16,10 @@ import sqlite3
import argparse import argparse
from time import strftime from time import strftime
import os import os
from datetime import datetime, timedelta
tables = ['record_race', 'record_teamrace', 'record_saves']
date = (datetime.now() - timedelta(hours=1)).strftime('%Y-%m-%d %H:%M:%S')
def sqlite_table_exists(cursor, table): def sqlite_table_exists(cursor, table):
cursor.execute(f"SELECT COUNT(*) FROM sqlite_master WHERE type='table' AND name='{table}'") cursor.execute(f"SELECT COUNT(*) FROM sqlite_master WHERE type='table' AND name='{table}'")
@ -38,17 +42,19 @@ def transfer(file_from, file_to):
for line in conn_from.iterdump(): for line in conn_from.iterdump():
cursor_to.execute(line) cursor_to.execute(line)
print(line.encode('utf-8')) print(line.encode('utf-8'))
for table in tables:
cursor_to.execute(f'INSERT INTO {table} SELECT * FROM {table}_backup WHERE Timestamp < "{date}"')
cursor_to.close() cursor_to.close()
conn_to.commit() conn_to.commit()
conn_to.close() conn_to.close()
cursor_from = conn_from.cursor() cursor_from = conn_from.cursor()
if sqlite_table_exists(cursor_from, 'record_race'): for table in tables:
cursor_from.execute('DELETE FROM record_race') if sqlite_table_exists(cursor_from, table):
if sqlite_table_exists(cursor_from, 'record_teamrace'): cursor_from.execute(f'DELETE FROM {table}')
cursor_from.execute('DELETE FROM record_teamrace') backup_table = f'{table}_backup'
if sqlite_table_exists(cursor_from, 'record_saves'): if sqlite_table_exists(cursor_from, backup_table):
cursor_from.execute('DELETE FROM record_saves') cursor_from.execute(f'DELETE FROM {backup_table} WHERE Timestamp < "{date}"')
cursor_from.close() cursor_from.close()
conn_from.commit() conn_from.commit()
conn_from.close() conn_from.close()
@ -71,15 +77,15 @@ def main():
return return
conn = sqlite3.connect(args.f) conn = sqlite3.connect(args.f)
num_ranks = sqlite_num_transfer(conn, 'record_race') num = {}
num_teamranks = sqlite_num_transfer(conn, 'record_teamrace') for table in tables:
num_saves = sqlite_num_transfer(conn, 'record_saves') num[table] = sqlite_num_transfer(conn, table)
num = num_ranks + num_teamranks + num_saves num[table] += sqlite_num_transfer(conn, f'{table}_backup')
conn.close() conn.close()
if num == 0: if sum(num.values()) == 0:
return return
print(f'{num} new entries in backup database found ({num_ranks} ranks, {num_teamranks} teamranks, {num_saves} saves)') print(f'{num} new entries in backup database found ({num["record_race"]} ranks, {num["record_teamrace"]} teamranks, {num["record_saves"]} saves)')
print(f'Moving entries from {os.path.abspath(args.f)} to {os.path.abspath(args.to)}') print(f'Moving entries from {os.path.abspath(args.f)} to {os.path.abspath(args.to)}')
print("You can use the following commands to import the entries to MySQL (using https://github.com/techouse/sqlite3-to-mysql/):") print("You can use the following commands to import the entries to MySQL (using https://github.com/techouse/sqlite3-to-mysql/):")
print() print()