Skip to content
Snippets Groups Projects
Commit 6a23c120 authored by Bo-Chun Chen's avatar Bo-Chun Chen
Browse files

Include fromisoformat for python 3.6

parent 2f7d0668
No related branches found
No related tags found
No related merge requests found
import re
import csv import csv
import dataset import dataset
from datetime import datetime from datetime import datetime
import rabbit_config as rcfg import rabbit_config as rcfg
def fromisoformat(date_string: str):
regex = re.compile(
r"(\d{4})-(\d{2})-(\d{2}) (\d{2}):(\d{2}):(\d{2})\.(\d{1,6})"
)
mo = re.search(regex, date_string)
return datetime(*[int(m) for m in mo.groups()])
def upgrade(): def upgrade():
db_path = rcfg.db_path db_path = rcfg.db_path
db_name = "user_reg.db" db_name = "user_reg.db"
...@@ -25,13 +34,13 @@ def upgrade(): ...@@ -25,13 +34,13 @@ def upgrade():
next(reader) next(reader)
for row in reader: for row in reader:
if row[0] in users: if row[0] in users:
users[row[0]]["created"] = datetime.fromisoformat(row[1][:-3]) users[row[0]]["created"] = fromisoformat(row[1][:-3])
users[row[0]]["origin"] = row[2] users[row[0]]["origin"] = row[2]
# Add copy created column from last_update if they are not in csv file # Add copy created column from last_update if they are not in csv file
for username in users: for username in users:
if "created" not in users[username]: if "created" not in users[username]:
users[username]["created"] = datetime.fromisoformat( users[username]["created"] = fromisoformat(
users[username]["last_update"] users[username]["last_update"]
) )
users[username]["origin"] = "last_update" users[username]["origin"] = "last_update"
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment