Skip to content

Commit 803f865

Browse files
committed
Code cleanup
1 parent 123e3cb commit 803f865

File tree

5 files changed

+31
-11
lines changed

5 files changed

+31
-11
lines changed

packet/commands.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
from .models import Freshman, Packet, FreshSignature, UpperSignature, MiscSignature
1212
from .ldap import ldap_get_eboard, ldap_get_live_onfloor
1313

14+
1415
@app.cli.command("create-secret")
1516
def create_secret():
1617
"""
@@ -19,15 +20,18 @@ def create_secret():
1920
print("Here's your random secure token:")
2021
print(token_hex())
2122

23+
2224
packet_start_time = time(hour=19)
2325
packet_end_time = time(hour=21)
2426

27+
2528
class CSVFreshman:
2629
def __init__(self, row):
2730
self.name = row[0]
2831
self.rit_username = row[3]
2932
self.onfloor = row[1] == "TRUE"
3033

34+
3135
def parse_csv(freshmen_csv):
3236
print("Parsing file...")
3337
try:
@@ -37,6 +41,7 @@ def parse_csv(freshmen_csv):
3741
print("Failure while parsing CSV")
3842
raise e
3943

44+
4045
@app.cli.command("sync-freshmen")
4146
@click.argument("freshmen_csv")
4247
def sync_freshmen(freshmen_csv):
@@ -82,6 +87,7 @@ def sync_freshmen(freshmen_csv):
8287
db.session.commit()
8388
print("Done!")
8489

90+
8591
@app.cli.command("create-packets")
8692
@click.argument("freshmen_csv")
8793
def create_packets(freshmen_csv):
@@ -126,6 +132,7 @@ def create_packets(freshmen_csv):
126132
db.session.commit()
127133
print("Done!")
128134

135+
129136
@app.cli.command("ldap-sync")
130137
def ldap_sync():
131138
"""
@@ -162,6 +169,7 @@ def ldap_sync():
162169
db.session.commit()
163170
print("Done!")
164171

172+
165173
@app.cli.command("fetch-results")
166174
def fetch_results():
167175
"""

packet/context_processors.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
from packet.models import Freshman
1010
from packet import app
1111

12+
1213
# pylint: disable=bare-except
1314
@lru_cache(maxsize=128)
1415
def get_csh_name(username):
@@ -18,6 +19,7 @@ def get_csh_name(username):
1819
except:
1920
return username
2021

22+
2123
# pylint: disable=bare-except
2224
@lru_cache(maxsize=128)
2325
def get_rit_name(username):
@@ -27,12 +29,14 @@ def get_rit_name(username):
2729
except:
2830
return username
2931

32+
3033
def log_time(label):
3134
"""
3235
Used during debugging to log timestamps while rendering templates
3336
"""
3437
print(label, datetime.now())
3538

39+
3640
@app.context_processor
3741
def utility_processor():
3842
return dict(get_csh_name=get_csh_name, get_rit_name=get_rit_name, log_time=log_time)

packet/debug_utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77

88
from packet import context_processors
99

10+
1011
def log_time(func):
1112
"""
1213
Decorator for logging the execution time of a function
@@ -24,6 +25,7 @@ def wrapped_function(*args, **kwargs):
2425

2526
return wrapped_function
2627

28+
2729
def log_cache():
2830
"""
2931
Utility call for logging cache info

packet/models.py

Lines changed: 13 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -43,12 +43,6 @@ class Freshman(db.Model):
4343
# One freshman can have multiple packets if they repeat the intro process
4444
packets = relationship("Packet", order_by="desc(Packet.id)")
4545

46-
def current_packet(self):
47-
"""
48-
:return: The most recent packet for this freshman
49-
"""
50-
return next(iter(self.packets), None)
51-
5246

5347
class Packet(db.Model):
5448
__tablename__ = "packet"
@@ -61,23 +55,29 @@ class Packet(db.Model):
6155
info_achieve = Column(Text, nullable=True) # Used to fulfil the technical achievements list requirement
6256

6357
freshman = relationship("Freshman", back_populates="packets")
64-
upper_signatures = relationship("UpperSignature", order_by="UpperSignature.signed.desc(), UpperSignature.updated",
65-
lazy="subquery")
66-
fresh_signatures = relationship("FreshSignature", order_by="FreshSignature.signed.desc(), FreshSignature.updated",
67-
lazy="subquery")
68-
misc_signatures = relationship("MiscSignature", order_by="MiscSignature.updated", lazy="subquery")
58+
upper_signatures = relationship("UpperSignature", lazy="subquery",
59+
order_by="UpperSignature.signed.desc(), UpperSignature.updated")
60+
fresh_signatures = relationship("FreshSignature", lazy="subquery",
61+
order_by="FreshSignature.signed.desc(), FreshSignature.updated")
62+
misc_signatures = relationship("MiscSignature", lazy="subquery", order_by="MiscSignature.updated")
6963

7064
def is_open(self):
7165
return self.start < datetime.now() < self.end
7266

7367
def signatures_required(self):
68+
"""
69+
:return: A SigCounts instance with the fields set to the number of signatures received by this packet
70+
"""
7471
eboard = sum(map(lambda sig: 1 if sig.eboard else 0, self.upper_signatures))
7572
upper = len(self.upper_signatures) - eboard
7673
fresh = len(self.fresh_signatures)
7774

7875
return SigCounts(eboard, upper, fresh, REQUIRED_MISC_SIGNATURES)
7976

8077
def signatures_received(self):
78+
"""
79+
:return: A SigCounts instance with the fields set to the number of required signatures for this packet
80+
"""
8181
eboard = sum(map(lambda sig: 1 if sig.eboard and sig.signed else 0, self.upper_signatures))
8282
upper = sum(map(lambda sig: 1 if not sig.eboard and sig.signed else 0, self.upper_signatures))
8383
fresh = sum(map(lambda sig: 1 if sig.signed else 0, self.fresh_signatures))
@@ -86,7 +86,9 @@ def signatures_received(self):
8686

8787
def did_sign(self, username, is_csh):
8888
"""
89+
:param username: The CSH or RIT username to check for
8990
:param is_csh: Set to True for CSH accounts and False for freshmen
91+
:return: Boolean value for if the given account signed this packet
9092
"""
9193
if is_csh:
9294
for sig in filter(lambda sig: sig.member == username, chain(self.upper_signatures, self.misc_signatures)):

packet/routes/shared.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,7 @@
1+
"""
2+
Routes available to both freshmen and CSH users
3+
"""
4+
15
from datetime import datetime
26
from flask import render_template, redirect
37

0 commit comments

Comments
 (0)