add session management api; add more unit tests for account module

utils/api/_serializers.py

@@ -1,11 +1,9 @@
-from django.utils import timezone
 from rest_framework import serializers
 
 
 class DateTimeTZField(serializers.DateTimeField):
     def to_representation(self, value):
-        # self.format = "%Y-%m-%d %H:%M:%S %Z"
-        value = timezone.localtime(value)
+        # value = timezone.localtime(value)
         return super(DateTimeTZField, self).to_representation(value)
 
 

utils/api/tests.py

@@ -3,12 +3,14 @@ from django.test.testcases import TestCase
 from rest_framework.test import APIClient
 
 from account.models import AdminType, ProblemPermission, User, UserProfile
+from conf.models import WebsiteConfig
 
 
 class APITestCase(TestCase):
     client_class = APIClient
 
-    def create_user(self, username, password, admin_type=AdminType.REGULAR_USER, login=True, problem_permission=ProblemPermission.NONE):
+    def create_user(self, username, password, admin_type=AdminType.REGULAR_USER, login=True,
+                    problem_permission=ProblemPermission.NONE):
         user = User.objects.create(username=username, admin_type=admin_type, problem_permission=problem_permission)
         user.set_password(password)
         UserProfile.objects.create(user=user)
@@ -18,13 +20,17 @@ class APITestCase(TestCase):
         return user
 
     def create_admin(self, username="admin", password="admin", login=True):
-        return self.create_user(username=username, password=password, admin_type=AdminType.ADMIN, problem_permission=ProblemPermission.OWN,
+        return self.create_user(username=username, password=password, admin_type=AdminType.ADMIN,
+                                problem_permission=ProblemPermission.OWN,
                                 login=login)
 
     def create_super_admin(self, username="root", password="root", login=True):
         return self.create_user(username=username, password=password, admin_type=AdminType.SUPER_ADMIN,
                                 problem_permission=ProblemPermission.ALL, login=login)
 
+    def create_website_config(self):
+        return WebsiteConfig.objects.create()
+
     def reverse(self, url_name):
         return reverse(url_name)
 

utils/shortcuts.py

@@ -69,3 +69,12 @@ def img2base64(img):
     img_prefix = "data:image/png;base64,"
     b64_str = img_prefix + b64encode(buf_str).decode("utf-8")
     return b64_str
+
+
+def datetime2str(value, format="iso-8601"):
+    if format.lower() == "iso-8601":
+        value = value.isoformat()
+        if value.endswith("+00:00"):
+            value = value[:-6] + "Z"
+        return value
+    return value.strftime(format)