mirror of
https://github.com/ddnet/ddnet.git
synced 2024-09-20 09:34:19 +00:00
101 lines
3.8 KiB
Python
Executable file
101 lines
3.8 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
# This script is intended to be called automatically every day (e.g. via cron).
|
|
# It only output stuff if new ranks have to be inserted. Therefore the output
|
|
# may be redirected to email notifying about manual action to transfer the
|
|
# ranks to MySQL.
|
|
#
|
|
# Configure cron as the user running the DDNet-Server processes
|
|
#
|
|
# $ crontab -e
|
|
# 30 5 * * * /path/to/this/script/move_sqlite.py --from /path/to/ddnet-server.sqlite
|
|
#
|
|
# Afterwards configure a MTA (e.g. postfix) and the users email address.
|
|
|
|
import sqlite3
|
|
import argparse
|
|
from time import strftime
|
|
import os
|
|
|
|
def sqlite_table_exists(cursor, table):
|
|
cursor.execute(f"SELECT COUNT(*) FROM sqlite_master WHERE type='table' AND name='{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(f'SELECT COUNT(*) FROM {table}')
|
|
num = c.fetchone()[0]
|
|
return num
|
|
|
|
def transfer(file_from, file_to):
|
|
conn_to = sqlite3.connect(file_to, isolation_level='EXCLUSIVE')
|
|
cursor_to = conn_to.cursor()
|
|
|
|
conn_from = sqlite3.connect(file_from, isolation_level='EXCLUSIVE')
|
|
for line in conn_from.iterdump():
|
|
cursor_to.execute(line)
|
|
print(line.encode('utf-8'))
|
|
cursor_to.close()
|
|
conn_to.commit()
|
|
conn_to.close()
|
|
|
|
cursor_from = conn_from.cursor()
|
|
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()
|
|
|
|
def main():
|
|
default_output = 'ddnet-server-' + strftime('%Y-%m-%d') + '.sqlite'
|
|
parser = argparse.ArgumentParser(
|
|
description='Move DDNet ranks, teamranks and saves from a possible active SQLite3 to a new one',
|
|
formatter_class=argparse.ArgumentDefaultsHelpFormatter)
|
|
parser.add_argument('--from', '-f', dest='f',
|
|
default='ddnet-server.sqlite',
|
|
help='Input file where ranks are deleted from when moved successfully (default: ddnet-server.sqlite)')
|
|
parser.add_argument('--to', '-t',
|
|
default=default_output,
|
|
help='Output file where ranks are saved adds current date by default')
|
|
args = parser.parse_args()
|
|
|
|
if not os.path.exists(args.f):
|
|
print(f"Warning: '{args.f}' does not exist (yet). Is the path specified correctly?")
|
|
return
|
|
|
|
conn = sqlite3.connect(args.f)
|
|
num_ranks = sqlite_num_transfer(conn, 'record_race')
|
|
num_teamranks = sqlite_num_transfer(conn, 'record_teamrace')
|
|
num_saves = sqlite_num_transfer(conn, 'record_saves')
|
|
num = num_ranks + num_teamranks + num_saves
|
|
conn.close()
|
|
if num == 0:
|
|
return
|
|
|
|
print(f'{num} new entries in backup database found ({num_ranks} ranks, {num_teamranks} teamranks, {num_saves} saves')
|
|
print('Moving entries from {os.path.abspath(args.f)} to {os.path.abspath(args.to)}')
|
|
sql_file = 'ddnet-server-' + strftime('%Y-%m-%d') + '.sql'
|
|
print("You can use the following commands to import the entries to MySQL (use sed 's/record_/prefix_/' for other database prefixes):")
|
|
print()
|
|
print((f" echo '.dump --preserve-rowids' | sqlite3 {os.path.abspath(args.to)} | " + # including rowids, this forces sqlite to name all columns in each INSERT statement
|
|
"grep -E '^INSERT INTO record_(race|teamrace|saves)' | " + # filter out inserts
|
|
"sed -e 's/INSERT INTO/INSERT IGNORE INTO/' | " + # ignore duplicate rows
|
|
f"sed -e 's/rowid,//' -e 's/VALUES([0-9]*,/VALUES(/' > {sql_file}")) # filter out rowids again
|
|
print(f" mysql -u teeworlds -p'PW2' teeworlds < {sql_file}")
|
|
print()
|
|
print(f"When the ranks are transfered successfully to mysql, {os.path.abspath(args.to)} can be removed")
|
|
print()
|
|
print("Log of the transfer:")
|
|
print()
|
|
|
|
transfer(args.f, args.to)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|