def position(uid):
if uid in staff_hours['staff-positions']:
return staff_hours['staff-positions'][uid]
elif is_staff(uid, group='ocfroot'):
return 'Technical Manager'
else:
return 'Staff Member'
def position(uid):
if uid in staff_hours['staff-positions']:
return staff_hours['staff-positions'][uid]
elif is_staff(uid, group='ocfroot'):
return 'Technical Manager'
else:
return 'Staff Member'
def get_quota(c, user):
"""Return a UserQuota representing the user's quota."""
if is_staff(user, 'opstaff'):
return UserQuota(user, 500, 500)
if not user_exists(user) or user_is_group(user):
return UserQuota(user, 0, 0)
c.execute('SELECT `today`, `semester` FROM `printed` WHERE `user` = %s',
(user, ))
row = c.fetchone()
if not row:
row = {'today': 0, 'semester': 0}
semesterly = max(0, SEMESTERLY_QUOTA - int(row['semester']))
return UserQuota(
user=user,
daily=max(0, min(semesterly,
daily_quota() - int(row['today']))),
semesterly=semesterly,
)
def get_quota(c, user):
"""Return a UserQuota representing the user's quota."""
if user == 'pubstaff' or is_staff(user, 'opstaff'):
return UserQuota(user, 500, 500)
if not user_exists(user) or user_is_group(user):
return UserQuota(user, 0, 0)
c.execute(
'SELECT `today`, `semester` FROM `printed` WHERE `user` = %s',
(user,)
)
row = c.fetchone()
if not row:
row = {'today': 0, 'semester': 0}
semesterly = max(0, SEMESTERLY_QUOTA - int(row['semester']))
return UserQuota(
user=user,
daily=max(0, min(semesterly, daily_quota() - int(row['today']))),
semesterly=semesterly,
)
def test_is_staff(user, group, expected):
kwargs = {} if not group else {'group': group}
assert is_staff(user, **kwargs) is expected
def test_is_staff(user, group, expected):
kwargs = {} if not group else {'group': group}
assert is_staff(user, **kwargs) is expected