sqlExport: Get folder membership and apply into memberflags

This commit is contained in:
Adam Goldsmith 2023-01-03 16:20:52 -05:00
parent f85c26a844
commit a449b81e46

View File

@ -22,6 +22,12 @@ def do_import(config):
] ]
).on_conflict(action="update", preserve=[Flag.name, Flag.type]).execute() ).on_conflict(action="update", preserve=[Flag.name, Flag.type]).execute()
print("Getting folder membership...")
folders = {
folder_id: membershipworks.get_member_ids([folder_name])
for folder_name, folder_id in membershipworks._parse_flags()["folders"].items()
}
print("Getting/Updating members...") print("Getting/Updating members...")
members = membershipworks.get_all_members() members = membershipworks.get_all_members()
for m in members: for m in members:
@ -44,13 +50,14 @@ def do_import(config):
# update member's flags # update member's flags
for type, flags in membershipworks._parse_flags().items(): for type, flags in membershipworks._parse_flags().items():
if type != "folders": # currently no way to retrieve this info for flag, id in flags.items():
for flag, id in flags.items(): ml = MemberFlag(uid=member["Account ID"], flag_id=id)
ml = MemberFlag(uid=member["Account ID"], flag_id=id) if (type == "folders" and member["Account ID"] in folders[id]) or (
if member[flag]: type != "folders" and member[flag]
ml.magic_save() ):
else: ml.magic_save()
ml.delete_instance() else:
ml.delete_instance()
print("Getting/Updating transactions...") print("Getting/Updating transactions...")
# Deduping these is hard, so just recreate the data every time # Deduping these is hard, so just recreate the data every time