Coverage for app/models/user.py: 93%

1from app.models import * 

2 

3class User(baseModel): 

4 username = CharField(primary_key=True) 

5 bnumber = CharField(unique=True) 

6 email = CharField() 

7 phoneNumber = CharField(null=True) 

8 firstName = CharField() 

9 lastName = CharField() 

10 cpoNumber = CharField(null=False) 

11 isStudent = BooleanField(default=False) 

12 major = CharField(null=True) 

13 rawClassLevel = CharField(null=True) 

14 isFaculty = BooleanField(default=False) 

15 isStaff = BooleanField(default=False) 

16 isCeltsAdmin = BooleanField(default=False) 

17 isCeltsStudentStaff = BooleanField(default=False) 

18 dietRestriction = TextField(null=True) 

19 minorInterest = BooleanField(default=False) 

20 hasGraduated = BooleanField(default=False) 

21 declaredMinor = BooleanField(default=False) 

22 

23 # override BaseModel's __init__ so that we can set up an instance attribute for cache 

24 def __init__(self,*args, **kwargs): 

25 super().__init__(*args,**kwargs) 

26 

27 self._pmCache = {} 

28 self._bsCache = None 

29 

30 @property 

31 def processedClassLevel(self): 

32 if not self.rawClassLevel: 

33 return "" 

34 return "Graduated" if (self.hasGraduated) else self.rawClassLevel 

35 

36 @property 

37 def isAdmin(self): 

38 return (self.isCeltsAdmin or self.isCeltsStudentStaff) 

39 

40 @property 

41 def isBonnerScholar(self): 

42 from app.models.bonnerCohort import BonnerCohort 

43 if self._bsCache is None: 

44 # TODO should we exclude users who are banned from Bonner here? 

45 self._bsCache = BonnerCohort.select().where(BonnerCohort.user == self).exists() 

46 

47 return self._bsCache 

48 

49 @property 

50 def fullName(self): 

51 return f"{self.firstName} {self.lastName}" 

52 

53 def addProgramManager(self, program): 

54 # Makes a user a Program Manager 

55 from app.models.programManager import ProgramManager 

56 ProgramManager.create(user = self, program = program) 

57 

58 return (f' {self} added as Program Manager') 

59 

60 def removeProgramManager(self, program): 

61 # Removes an existing Program Manager from being a Program Manager 

62 from app.models.programManager import ProgramManager 

63 ProgramManager.delete().where(ProgramManager.user == self, ProgramManager.program == program).execute() 

64 

65 return (f'{self} removed from Program Manager') 

66 

67 def isProgramManagerFor(self, program): 

68 # Looks to see who is the Program Manager for a program 

69 from app.models.programManager import ProgramManager # Must defer import until now to avoid circular reference 

70 if not program: 

71 return False 

72 

73 if program not in self._pmCache: 

74 self._pmCache[program] = ProgramManager.select().where(ProgramManager.user == self, ProgramManager.program == program).exists() 

75 

76 return self._pmCache[program] 

77 

78 def isProgramManagerForEvent(self, event): 

79 # Looks to see who the Program Manager for a specific event is 

80 return self.isProgramManagerFor(event.program) 

81