Created the ldap-sync command

Author: JoelEager

packet/commands.py

@@ -5,10 +5,11 @@
 from secrets import token_hex
 from datetime import datetime, time, timedelta
 import csv
+from itertools import chain
 import click
 
 from . import app, db
-from .models import Freshman, Packet, UpperSignature, FreshSignature
+from .models import *
 from .ldap import ldap_get_eboard, ldap_get_live_onfloor
 
 @app.cli.command("create-secret")
@@ -119,3 +120,35 @@ def create_packets(freshmen_csv):
 
     db.session.commit()
     print("Done!")
+
+@app.cli.command("ldap-sync")
+def ldap_sync():
+    print("Fetching data from LDAP...")
+    eboard = set(member.uid for member in ldap_get_eboard())
+    onfloor = set(member.uid for member in ldap_get_live_onfloor())
+    all_upper = eboard.union(onfloor)
+
+    print("Applying updates to the DB...")
+    for packet in Packet.query.filter(Packet.end > datetime.now()).all():
+        # Update the eboard state of all UpperSignatures
+        for sig in packet.upper_signatures:
+            sig.eboard = sig.member in eboard
+
+        # Migrate UpperSignatures that are from accounts that are not eboard or onfloor anymore
+        for sig in filter(lambda sig: sig.member not in all_upper, packet.upper_signatures):
+            UpperSignature.query.filter_by(packet_id=packet.id, member=sig.member).delete()
+            if sig.signed:
+                db.session.add(MiscSignature(packet=packet, member=sig.member))
+
+        # Migrate MiscSignatures that are from accounts that are now eboard or onfloor members
+        for sig in filter(lambda sig: sig.member in all_upper, packet.misc_signatures):
+            MiscSignature.query.filter_by(packet_id=packet.id, member=sig.member).delete()
+            db.session.add(UpperSignature(packet=packet, member=sig.member, eboard=sig.member in eboard, signed=True))
+
+        # Create UpperSignatures for any new eboard or onfloor members
+        upper_sigs = set(map(lambda sig: sig.member, packet.upper_signatures))
+        for member in filter(lambda member: member not in upper_sigs, all_upper):
+            db.session.add(UpperSignature(packet=packet, member=member, eboard=member in eboard))
+
+    db.session.commit()
+    print("Done!")