almet/ihatemoney
1
0
Fork 0
mirror of https://github.com/spiral-project/ihatemoney.git synced 2025-05-06 13:01:50 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Replace duplicate if with property method

Browse source
This commit is contained in:
Andrew Dickinson 2020-04-18 12:53:33 -04:00
parent dc9a420bfa
commit 895ba61d74
1 changed files with 13 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
ihatemoney/forms.py
View file

@ -92,25 +92,28 @@ class EditProjectForm(FlaskForm):
project_history = BooleanField(_("Enable project history")) project_history = BooleanField(_("Enable project history"))
ip_recording = BooleanField(_("Use IP tracking for project history")) ip_recording = BooleanField(_("Use IP tracking for project history"))
@property
def logging_preference(self):
"""Get the LoggingMode object corresponding to current form data."""
if not self.project_history.data:
return LoggingMode.DISABLED
else:
if self.ip_recording.data:
return LoggingMode.RECORD_IP
else:
return LoggingMode.ENABLED
def save(self): def save(self):
"""Create a new project with the information given by this form. """Create a new project with the information given by this form.
Returns the created instance Returns the created instance
""" """
if not self.project_history.data:
new_logging_preference = LoggingMode.DISABLED
else:
if self.ip_recording.data:
new_logging_preference = LoggingMode.RECORD_IP
else:
new_logging_preference = LoggingMode.ENABLED
project = Project( project = Project(
name=self.name.data, name=self.name.data,
id=self.id.data, id=self.id.data,
password=generate_password_hash(self.password.data), password=generate_password_hash(self.password.data),
contact_email=self.contact_email.data, contact_email=self.contact_email.data,
logging_preference=new_logging_preference, logging_preference=self.logging_preference,
) )
return project return project
@ -123,16 +126,7 @@ class EditProjectForm(FlaskForm):
project.password = generate_password_hash(self.password.data) project.password = generate_password_hash(self.password.data)
project.contact_email = self.contact_email.data project.contact_email = self.contact_email.data
project.logging_preference = self.logging_preference
if not self.project_history.data:
new_logging_preference = LoggingMode.DISABLED
else:
if self.ip_recording.data:
new_logging_preference = LoggingMode.RECORD_IP
else:
new_logging_preference = LoggingMode.ENABLED
project.logging_preference = new_logging_preference
return project return project