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

1from app.models import * 

2 

3 

4class User(baseModel): 

5 username = CharField(primary_key=True) 

6 bnumber = CharField(unique=True) 

7 email = CharField() 

8 phoneNumber = CharField(null=True) 

9 firstName = CharField() 

10 lastName = CharField() 

11 cpoNumber = CharField(null=False) 

12 isStudent = BooleanField(default=False) 

13 major = CharField(null=True) 

14 rawClassLevel = CharField(null=True) 

15 isFaculty = BooleanField(default=False) 

16 isStaff = BooleanField(default=False) 

17 isCeltsAdmin = BooleanField(default=False) 

18 isCeltsStudentStaff = BooleanField(default=False) 

19 dietRestriction = TextField(null=True) 

20 minorInterest = BooleanField(default=False) 

21 hasGraduated = BooleanField(default=False) 

22 declaredMinor = BooleanField(default=False) 

23 

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

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

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

27 

28 self._pmCache = {} 

29 self._bsCache = None 

30 

31 @property 

32 def processedClassLevel(self): 

33 if not self.rawClassLevel: 

34 return "" 

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

36 

37 @property 

38 def isAdmin(self): 

39 return (self.isCeltsAdmin or self.isCeltsStudentStaff) 

40 

41 @property 

42 def isBonnerScholar(self): 

43 from app.models.bonnerCohort import BonnerCohort 

44 if self._bsCache is None: 

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

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

47 

48 return self._bsCache 

49 

50 @property 

51 def fullName(self): 

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

53 

54 def addProgramManager(self, program): 

55 # Makes a user a Program Manager 

56 from app.models.programManager import ProgramManager 

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

58 

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

60 

61 def removeProgramManager(self, program): 

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

63 from app.models.programManager import ProgramManager 

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

65 

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

67 

68 def isProgramManagerFor(self, program): 

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

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

71 if not program: 

72 return False 

73 

74 if program not in self._pmCache: 

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

76 

77 return self._pmCache[program] 

78 

79 def isProgramManagerForEvent(self, event): 

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

81 return self.isProgramManagerFor(event.program) 

82 

83