22from .models import Freshman , UpperSignature , FreshSignature , MiscSignature , db
33
44
5- def sign (member_username , freshman_username ):
5+ def sign (signer_username , freshman_username ):
66 freshman = Freshman .query .filter_by (rit_username = freshman_username ).first ()
77 if freshman is None :
88 return False
9-
109 packet = freshman .current_packet ()
1110 if packet is None :
1211 return False
1312 if not packet .is_open ():
1413 return False
1514
16- upper_signature = UpperSignature .query .filter_by (member = member_username ).first ()
17- fresh_signature = FreshSignature .query .filter_by (freshman_username = member_username ).first ()
15+ upper_signature = UpperSignature .query .filter (UpperSignature .member == signer_username ).first ()
16+ fresh_signature = FreshSignature .query .filter (FreshSignature .freshman_username == signer_username ).first ()
17+
1818 if upper_signature :
1919 upper_signature .signed = True
2020 elif fresh_signature :
2121 fresh_signature .signed = True
2222 else :
23- db .session .add (MiscSignature (packet .id , member_username , datetime .now (), packet ))
23+ db .session .add (MiscSignature (packet .id , signer_username , datetime .now (), packet ))
2424 db .session .commit ()
2525
2626 return True
@@ -37,8 +37,8 @@ def get_signatures(freshman_username):
3737
3838
3939def get_number_signed (freshman_username ):
40- return Freshman .query .filter_by (rit_username = freshman_username )[ 0 ] .current_packet ().signatures_received ()
40+ return Freshman .query .filter_by (rit_username = freshman_username ). first () .current_packet ().signatures_received ()
4141
4242
4343def get_number_required (freshman_username ):
44- return Freshman .query .filter_by (rit_username = freshman_username )[ 0 ] .current_packet ().signatures_required ()
44+ return Freshman .query .filter_by (rit_username = freshman_username ). first () .current_packet ().signatures_required ()
0 commit comments