def setUp(self):
# Necessary Currency Objects
usd = Currency.objects.create()
USA = Country()
USA.PrimaryCurrency = usd
USA.Name = "United States"
USA.Code = "US"
USA.save()
# Types
u1 = UserType.objects.create(Name="Type 1")
u2 = UserType.objects.create(Name="Type 2")
u3 = UserType.objects.create(Name="Type 3")
brian = User.objects.create(username="******",
email="*****@*****.**",
first_name="Brian",
last_name="Katchmar") # Passes
settings_brian = UserSettings()
settings_brian = settings_brian.get_settings_based_on_user(brian)
settings_brian.BusinessName = "MIAC"
settings_brian.Region = "New York"
settings_brian.save()
UserAssociatedTypes.objects.create(UserAccount=brian,
UserFreelanceType=u1)
josh = User.objects.create(username="******",
email="*****@*****.**",
first_name="Josh",
last_name="Blank") # No User Settings
kenny = User.objects.create(
username="******",
email="*****@*****.**",
first_name="Kenny",
last_name="Kim"
) # Has User Settings, has associated types, neither of the two fields we check
kenny.set_password("Thing5Ar3Gr34t")
kenny.save()
settings_kenny = UserSettings()
settings_kenny = settings_brian.get_settings_based_on_user(kenny)
settings_kenny.save()
UserAssociatedTypes.objects.create(UserAccount=kenny,
UserFreelanceType=u2)
UserAssociatedTypes.objects.create(UserAccount=kenny,
UserFreelanceType=u3)
clara = User.objects.create(
username="******",
email="*****@*****.**",
first_name="Clara",
last_name="Chang") # Has User Settings, no associated types
clara.set_password("Thing5Ar3Gr34t")
clara.save()
settings_clara = UserSettings()
settings_clara = settings_brian.get_settings_based_on_user(clara)
settings_clara.BusinessName = "InFlow"
settings_clara.Region = "New York"
settings_clara.save()
def setUp(self):
# Necessary Currency Objects
usd = Currency.objects.create()
USA = Country()
USA.PrimaryCurrency = usd
USA.Name = "United States"
USA.Code = "US"
USA.save()
kenny = User.objects.create(username="******",
email="*****@*****.**",
first_name="Kenny",
last_name="Kim")
kenny.set_password("Thing5Ar3Gr34t")
kenny.save()
clara = User.objects.create(username="******",
email="*****@*****.**",
first_name="Clara",
last_name="Chang")
clara.set_password("Thing5Ar3Gr34t")
clara.save()
settings_kenny = UserSettings()
settings_kenny = settings_kenny.get_settings_based_on_user(kenny)
settings_kenny.save()
settings_clara = UserSettings()
settings_clara = settings_clara.get_settings_based_on_user(clara)
settings_clara.BusinessName = "InFlow"
settings_clara.Region = "New York"
settings_clara.PhoneNumber = "2125555555"
settings_clara.StripeConnectAccountKey = "1234567890"
settings_clara.save()
def setUp(self):
# Necessary Currency Objects
usd = Currency.objects.create()
USA = Country()
USA.PrimaryCurrency = usd
USA.Name = "United States"
USA.Code = "US"
USA.save()
# Types
u1 = UserType.objects.create(Name="Clara Type 1")
u2 = UserType.objects.create(Name="Clara Type 2")
u3 = UserType.objects.create(Name="Clara Type 3")
clara = User.objects.create(
username="******",
email="*****@*****.**",
first_name="Clara",
last_name="Chang") # Has User Settings, no associated types
clara.set_password("Thing5Ar3Gr34t")
clara.save()
settings_clara = UserSettings()
settings_clara = settings_clara.get_settings_based_on_user(clara)
settings_clara.BusinessName = "InFlow"
settings_clara.Region = "New York"
settings_clara.save()
def setUp(self):
# Necessary Currency Objects
usd = Currency.objects.create()
USA = Country()
USA.PrimaryCurrency = usd
USA.Name = "United States"
USA.Code = "US"
USA.save()
# Users
user_1 = User.objects.create(username="******",
email="*****@*****.**",
first_name="Brian",
last_name="Katchmar")
user_1.set_password("Th3L10nK1ng15Fun")
user_1.save()
settings_user_1 = UserSettings()
settings_user_1 = settings_user_1.get_settings_based_on_user(user_1)
settings_user_1.BusinessName = "MIAC"
settings_user_1.Region = "New York"
settings_user_1.save()
user_2 = User.objects.create(username="******",
email="*****@*****.**",
first_name="Brian",
last_name="Katchmar")
user_2.set_password("Thing5Ar3Gr34t")
user_2.save()
settings_user_2 = UserSettings()
settings_user_2 = settings_user_2.get_settings_based_on_user(user_2)
settings_user_2.save()
user_3 = User.objects.create(username="******",
email="*****@*****.**",
first_name="Brian",
last_name="Katchmar")
user_3.set_password("Thing5Ar3Gr34t")
user_3.save()
settings_user_3 = UserSettings()
settings_user_3 = settings_user_3.get_settings_based_on_user(user_3)
settings_user_3.BusinessName = "InFlow"
settings_user_3.Region = "New York"
settings_user_3.PhoneNumber = "2125555555"
settings_user_3.StripeConnectAccountKey = "1234567890"
settings_user_3.save()
def post(self, request):
context = self.get_context_data(request)
# Gather Data From Post
business_name = request.POST.get("business-name", "")
region = request.POST.get("region", "")
zip_code = request.POST.get("zip-code", "")
# Build up user settings or fetch previously saved ones, then save what was selected as the freelancer work with and interested in
usersettings = UserSettings()
usersettings = usersettings.get_settings_based_on_user(request.user)
usersettings.BusinessName = business_name
usersettings.Region = region
usersettings.ZipCode = zip_code
usersettings.save()
return redirect(reverse("base:dashboard"))
def setUp(self):
# Necessary Currency Objects
usd = Currency.objects.create()
USA = Country()
USA.PrimaryCurrency = usd
USA.Name = "United States"
USA.Code = "US"
USA.save()
# Users
brian = User.objects.create(username="******",
email="*****@*****.**",
first_name="Brian",
last_name="Katchmar")
brian.set_password("Th3L10nK1ng15Fun")
brian.save()
settings_brian = UserSettings()
settings_brian = settings_brian.get_settings_based_on_user(brian)
settings_brian.BusinessName = "MIAC"
settings_brian.Region = "New York"
settings_brian.save()
kenny = User.objects.create(
username="******",
email="*****@*****.**",
first_name="Kenny",
last_name="Kim"
) # Has User Settings, has associated types, neither of the two fields we check
kenny.set_password("Thing5Ar3Gr34t")
kenny.save()
settings_kenny = UserSettings()
settings_kenny = settings_kenny.get_settings_based_on_user(kenny)
settings_kenny.save()
clara = User.objects.create(username="******",
email="*****@*****.**",
first_name="Clara",
last_name="Chang")
clara.set_password("Thing5Ar3Gr34t")
clara.save()
settings_clara = UserSettings()
settings_clara = settings_clara.get_settings_based_on_user(clara)
settings_clara.BusinessName = "InFlow"
settings_clara.Region = "New York"
settings_clara.PhoneNumber = "2125555555"
settings_clara.StripeConnectAccountKey = "1234567890"
settings_clara.save()
if NotificationSetting.objects.filter(
SettingName=self.setting_string_1).exists():
ns1 = NotificationSetting.objects.get(
SettingName=self.setting_string_1)
else:
ns1 = NotificationSetting.objects.create(
SettingName=self.setting_string_1)
if NotificationSetting.objects.filter(
SettingName=self.setting_string_2).exists():
ns2 = NotificationSetting.objects.get(
SettingName=self.setting_string_2)
else:
ns2 = NotificationSetting.objects.create(
SettingName=self.setting_string_2)
if NotificationSetting.objects.filter(
SettingName=self.setting_string_3).exists():
ns3 = NotificationSetting.objects.get(
SettingName=self.setting_string_3)
else:
ns3 = NotificationSetting.objects.create(
SettingName=self.setting_string_3)
if NotificationSetting.objects.filter(
SettingName=self.setting_string_4).exists():
ns4 = NotificationSetting.objects.get(
SettingName=self.setting_string_4)
else:
ns4 = NotificationSetting.objects.create(
SettingName=self.setting_string_4)
# Purge the DB Entries, we're rebuild what we're expecting
UserNotificationSettings.objects.filter(UserAccount=brian).delete()
UserNotificationSettings.objects.filter(UserAccount=kenny).delete()
UserNotificationSettings.objects.filter(UserAccount=clara).delete()
# Brian
UserNotificationSettings.objects.create(UserAccount=brian,
Setting=ns1,
Selected=True)
UserNotificationSettings.objects.create(UserAccount=brian,
Setting=ns2,
Selected=True)
UserNotificationSettings.objects.create(UserAccount=brian,
Setting=ns3,
Selected=True)
UserNotificationSettings.objects.create(UserAccount=brian,
Setting=ns4,
Selected=True)
# Kenny
UserNotificationSettings.objects.create(UserAccount=kenny,
Setting=ns1,
Selected=True)
UserNotificationSettings.objects.create(UserAccount=kenny,
Setting=ns2,
Selected=False)
UserNotificationSettings.objects.create(UserAccount=kenny,
Setting=ns3,
Selected=False)
UserNotificationSettings.objects.create(UserAccount=kenny,
Setting=ns4,
Selected=True)
# Clara
UserNotificationSettings.objects.create(UserAccount=clara,
Setting=ns1,
Selected=False)
UserNotificationSettings.objects.create(UserAccount=clara,
Setting=ns2,
Selected=True)
UserNotificationSettings.objects.create(UserAccount=clara,
Setting=ns3,
Selected=False)
UserNotificationSettings.objects.create(UserAccount=clara,
Setting=ns4,
Selected=True)