db migrated

This commit is contained in:
Eriks K 2020-04-19 13:49:40 +03:00
parent 7c0d5d936c
commit 2a69b6195f

10
db.py
View File

@ -19,16 +19,6 @@ class DiscordDB:
migrate_db = False
if "member" not in self._db.table_names():
self._db.create_table("member", {"id": int, "name": str}, pk="id", not_null={"id", "name"})
else:
if "mention_number" in self._db.table('member').columns_dict.keys():
memb = namedtuple('Member', ['id', 'name'])
members: Set['memb'] = {*[]}
for row in self._db.table('member').rows:
members.add(memb(row['mention_number'], row['name']))
self._db.table('member').drop()
self._db.create_table("member", {"id": int, "name": str}, pk="id", not_null={"id", "name"})
self._db.table('member').insert_all([m._asdict() for m in members])
migrate_db = True
if "player" not in self._db.table_names():
self._db.create_table("player", {"id": int, "name": str}, pk="id", not_null={"id", "name"})