CMS/memberPlumbing
Archived
2
0
Fork 1
Code Issues 8 Pull Requests 2 Releases Wiki Activity

sqlExport: Get folder membership and apply into memberflags

Browse Source
This commit is contained in:
Adam Goldsmith 2023-01-03 16:20:52 -05:00
parent f85c26a844
commit a449b81e46
1 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
memberPlumbing/sqlExport.py
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,10 +50,11 @@ 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 member[flag]: if (type == "folders" and member["Account ID"] in folders[id]) or (
type != "folders" and member[flag]
):
ml.magic_save() ml.magic_save()
else: else:
ml.delete_instance() ml.delete_instance()