iartamonov/mautrix-telegram
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Rename db_migrate script to dbms_migrate

Browse Source
This commit is contained in:
Tulir Asokan
2018-07-16 23:30:02 +03:00
parent d4b725a508
commit e9348c9550
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mautrix_telegram/scripts/db_migrate/__main__.py → mautrix_telegram/scripts/dbms_migrate/__main__.py
+6 -6
View File
@@ -5,11 +5,11 @@ from sqlalchemy.ext.declarative import declarative_base
from alchemysession import AlchemySessionContainer from alchemysession import AlchemySessionContainer
parser = argparse.ArgumentParser(description="mautrix-telegram database migration script", parser = argparse.ArgumentParser(description="mautrix-telegram dbms migration script",
prog="python -m mautrix_telegram.scripts.db_migrate") prog="python -m mautrix_telegram.scripts.dbms_migrate")
parser.add_argument("-f", "--from-uri", type=str, required=True, metavar="<uri>", parser.add_argument("-f", "--from-url", type=str, required=True, metavar="<url>",
help="the old database path") help="the old database path")
parser.add_argument("-t", "--to-uri", type=str, required=True, metavar="<uri>", parser.add_argument("-t", "--to-url", type=str, required=True, metavar="<url>",
help="the new database path") help="the new database path")
args = parser.parse_args() args = parser.parse_args()
@@ -46,13 +46,13 @@ def connect(to):
} }
session, tables = connect(args.from_uri) session, tables = connect(args.from_url)
data = {} data = {}
for name, table in tables.items(): for name, table in tables.items():
data[name] = session.query(table).all() data[name] = session.query(table).all()
session, tables = connect(args.to_uri) session, tables = connect(args.to_url)
for name, table in tables.items(): for name, table in tables.items():
for row in data[name]: for row in data[name]:
session.merge(row) session.merge(row)