From 8cf64464ba094df60fe81442dd0d728075368344 Mon Sep 17 00:00:00 2001 From: Dirk Alders Date: Sun, 30 Aug 2020 19:52:03 +0200 Subject: [PATCH] Import and Export for Models added --- models.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/models.py b/models.py index f97cacd..bf0fbf1 100644 --- a/models.py +++ b/models.py @@ -15,9 +15,31 @@ def get_userprofile(user): return profile +def UserprofilerExist(username, data): + try: + profile = UserProfile.objects.get(user__username=username) + except UserProfile.DoesNotExist: + return False + return profile.timezone == data['timezone'] and profile.language_code == data['language_code'] + + # USERPROFILE Model # class UserProfile(models.Model): user = models.OneToOneField(User, unique=True, on_delete=models.CASCADE) timezone = models.CharField(max_length=150, default='UTC', choices=[(t, t) for t in pytz.common_timezones]) language_code = models.CharField(max_length=150, default='en', choices=settings.LANGUAGES) + + def export_key(self): + return self.user.username + + def export_data(self): + rv = {} + rv['timezone'] = self.timezone + rv['language_code'] = self.language_code + return rv + + def import_data(self, data): + self.timezone = data['timezone'] + self.language_code = data['language_code'] + self.save()