habitmove/migrate.py

73 lines
1.9 KiB
Python
Raw Normal View History

2021-08-26 19:57:49 +00:00
#!/usr/bin/env python
import json
2021-08-26 21:13:56 +00:00
import pandas as pd
2021-08-27 08:11:00 +00:00
import toloop as loop
2021-08-26 19:57:49 +00:00
def load_file(filename):
with open(filename) as f:
nomie_data = json.load(f)
return nomie_data
2021-08-26 21:13:56 +00:00
def get_dataframe(nomie_data):
return pd.DataFrame(nomie_data["events"])
# generate a yes/no cli question with a default answer
2021-08-26 19:57:49 +00:00
def confirmation_question(question, default_no=True):
choices = " [y/N]: " if default_no else " [Y/n]: "
default_answer = "n" if default_no else "y"
reply = str(input(question + choices)).lower().strip() or default_answer
if reply[0] == "y":
return True
if reply[0] == "n":
return False
else:
return False if default_no else True
2021-08-26 21:13:56 +00:00
# display stats and ask user to confirm if they seem okay
2021-08-26 19:57:49 +00:00
def verify_continue(data):
trackers = ""
for t in data["trackers"]:
trackers += t + ", "
2021-08-27 08:11:00 +00:00
print(f"Exporting from nomie {data['nomie']['number']}:")
2021-08-26 19:57:49 +00:00
print(f"Found trackers: {trackers}")
2021-08-27 08:11:00 +00:00
print(f"Found events: {len(data['events'])} entries.")
2021-08-26 19:57:49 +00:00
if not confirmation_question("Do you want to continue?", default_no=False):
print("Aborted.")
exit(0)
2021-08-27 08:11:00 +00:00
# better way to do this
# def create_connection(db_file):
# """create a database connection to the SQLite database
# specified by db_file
# :param db_file: database file
# :return: Connection object or None
# """
# conn = None
# try:
# conn = sqlite3.connect(db_file)
# return conn
# except Error as e:
# print(e)
# return conn
2021-08-26 19:57:49 +00:00
if __name__ == "__main__":
# load nomie json
nomie_data = load_file("input.json")
2021-08-26 21:13:56 +00:00
# DISABLED FOR DEBUGGING
2021-08-26 19:57:49 +00:00
verify_continue(nomie_data)
2021-08-27 08:11:00 +00:00
trackers = nomie_data["trackers"]
events = nomie_data["events"]
2021-08-26 19:57:49 +00:00
2021-08-26 21:13:56 +00:00
# bring up dataframe
df = get_dataframe(nomie_data)
2021-08-27 08:11:00 +00:00
# give it over to loop migrator
loop.migrate(trackers, events)