Day 5: Register and Login For 3 Role Users

Author: PTHARRISH

api/urls.py

@@ -0,0 +1,19 @@
+from django.urls import path
+
+from users.views import (
+    AdminRegisterView,
+    LoginView,
+    TechnicianRegisterView,
+    UserRegisterView,
+)
+
+urlpatterns = [
+    path("register/user/", UserRegisterView.as_view(), name="user-register"),
+    path(
+        "register/technician/",
+        TechnicianRegisterView.as_view(),
+        name="technician-register",
+    ),
+    path("register/admin/", AdminRegisterView.as_view(), name="admin-register"),
+    path("login/", LoginView.as_view(), name="login"),
+]

backend/settings.py

@@ -106,7 +106,8 @@
 
 AUTH_PASSWORD_VALIDATORS = [
     {
-        "NAME": "django.contrib.auth.password_validation.UserAttributeSimilarityValidator",
+        "NAME": "django.contrib.auth.password_validation."
+        "UserAttributeSimilarityValidator",
     },
     {
         "NAME": "django.contrib.auth.password_validation.MinimumLengthValidator",
@@ -153,7 +154,9 @@
 
 REST_FRAMEWORK = {
     "DEFAULT_SCHEMA_CLASS": "drf_spectacular.openapi.AutoSchema",
-    "DEFAULT_AUTHENTICATION_CLASSES": ("rest_framework_simplejwt.authentication.JWTAuthentication",),
+    "DEFAULT_AUTHENTICATION_CLASSES": (
+        "rest_framework_simplejwt.authentication.JWTAuthentication",
+    ),
 }
 
 AUTH_USER_MODEL = "users.User"
@@ -162,8 +165,8 @@
     "TITLE": " RO Water Purifier API",
     "DESCRIPTION": "RO Water Purifier is a web-based platform "
     "that allows users to browse and purchase RO water purifiers "
-    "while also providing a seamless way to book certified technicians for installation, "
-    "maintenance, and service.",
+    "while also providing a seamless way to book certified technicians "
+    "for installation, maintenance, and service.",
     "VERSION": "1.0.0",
     "SERVE_INCLUDE_SCHEMA": False,
 }

backend/urls.py

@@ -19,16 +19,29 @@
 
 from django.contrib import admin
 from django.urls import include, path
-from drf_spectacular.views import SpectacularAPIView, SpectacularRedocView, SpectacularSwaggerView
+from drf_spectacular.views import (
+    SpectacularAPIView,
+    SpectacularRedocView,
+    SpectacularSwaggerView,
+)
 from rest_framework_simplejwt.views import TokenObtainPairView, TokenRefreshView
 
 urlpatterns = [
     path("admin/", admin.site.urls),
     path("api/schema/", SpectacularAPIView.as_view(), name="schema"),
     path("api/token/", TokenObtainPairView.as_view(), name="token_obtain_pair"),
     path("api/token/refresh/", TokenRefreshView.as_view(), name="token_refresh"),
-    path("api/schema/swagger-ui/", SpectacularSwaggerView.as_view(url_name="schema"), name="swagger-ui"),
-    path("api/schema/redoc/", SpectacularRedocView.as_view(url_name="schema"), name="redoc"),
+    path(
+        "api/schema/swagger-ui/",
+        SpectacularSwaggerView.as_view(url_name="schema"),
+        name="swagger-ui",
+    ),
+    path(
+        "api/schema/redoc/",
+        SpectacularRedocView.as_view(url_name="schema"),
+        name="redoc",
+    ),
+    path("api/", include("api.urls")),
 ]
 
 urlpatterns += [path("silk/", include("silk.urls", namespace="silk"))]

pyproject.toml

@@ -13,6 +13,9 @@ dependencies = [
     "drf-spectacular>=0.29.0",
     "pillow>=12.0.0",
     "psycopg2>=2.9.11",
+    "pytest>=9.0.1",
+    "pytest-cov>=7.0.0",
+    "pytest-django>=4.11.1",
     "python-decouple>=3.8",
     "ruff>=0.14.5",
 ]
@@ -23,29 +26,48 @@ dev = [
     "ruff>=0.14.5",
 ]
 
+# ===============================
+# BLACK (FORMATTER ONLY)
+# ===============================
 [tool.black]
-line-length = 120
+line-length = 88
 target-version = ["py313"]
 
+# ===============================
+# RUFF - LINT + AUTOFIX
+# ===============================
 [tool.ruff]
-line-length = 120
+line-length = 88
 target-version = "py313"
-exclude = ["migrations"]
+exclude = ["migrations", ".venv", "env"]
+fix = true
 
 [tool.ruff.lint]
 select = ["E", "F", "I", "W", "D", "UP", "B", "DJ"]
 ignore = [
-    "D100", "D104", 
-    "D101", "D102", "D103",   
-    "D203", "D212","D105",
-   
+    "D100", 
+    "D101", 
+    "D102", 
+    "D103",
+    "D104",
+    "D105",
+    "D106",
+    "D107",
+    "D203",
+    "D212",
 ]
 
+# ===============================
+# RUFF FORMAT CONFIG
+# ===============================
 [tool.ruff.format]
 quote-style = "double"
 indent-style = "space"
 line-ending = "auto"
 
+# ===============================
+# RUFF ISORT (IMPORT MANAGEMENT)
+# ===============================
 [tool.ruff.lint.isort]
 combine-as-imports = true
 force-single-line = false

pytest.ini

@@ -0,0 +1,3 @@
+[pytest]
+DJANGO_SETTINGS_MODULE = backend.settings
+python_files = test_*.py *_test.py

users/models.py

@@ -9,10 +9,10 @@ class User(AbstractUser):
         ("admin", "Admin"),
         ("technician", "Technician"),
     ]
+    fullname = models.CharField(max_length=255)
     role = models.CharField(max_length=20, choices=ROLE_CHOICES)
     address = models.JSONField(blank=True, null=True)
-    phone_no = models.CharField(max_length=20, blank=True, default="")
-    mobile_no = models.CharField(max_length=20, blank=True, default="")
+    mobile = models.CharField(max_length=20, blank=True, default="")
 
 
 class Tag(models.Model):
@@ -37,23 +37,31 @@ class Profile(models.Model):
     years_experience = models.PositiveIntegerField(blank=True, null=True)
     months_experience = models.PositiveIntegerField(blank=True, null=True)
     tags = models.ManyToManyField(Tag, blank=True)  # Expertise categories
-    available_days = models.JSONField(blank=True, null=True)  # Calendar availability data
-    price_hour = models.DecimalField(max_digits=10, decimal_places=2, blank=True, null=True)
-    price_day = models.DecimalField(max_digits=10, decimal_places=2, blank=True, null=True)
+    available_days = models.JSONField(
+        blank=True, null=True
+    )  # Calendar availability data
+    price_hour = models.DecimalField(
+        max_digits=10, decimal_places=2, blank=True, null=True
+    )
+    price_day = models.DecimalField(
+        max_digits=10, decimal_places=2, blank=True, null=True
+    )
 
     def __str__(self):
-        return str(self.user.username)
+        return str(self.user)
 
 
 class Review(models.Model):
-    technician = models.ForeignKey(Profile, on_delete=models.CASCADE, related_name="reviews")
+    technician = models.ForeignKey(
+        Profile, on_delete=models.CASCADE, related_name="reviews"
+    )
     reviewer = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE)
     rating = models.PositiveSmallIntegerField()
     review_text = models.TextField()
     created_at = models.DateTimeField(auto_now_add=True)
 
     def __str__(self):
-        return str(self.technician.user.username) + " - " + str(self.rating)
+        return str(self.technician) + " - " + str(self.rating)
 
 
 class Booking(models.Model):
@@ -62,14 +70,20 @@ class Booking(models.Model):
         ("completed", "Completed"),
         ("cancelled", "Cancelled"),
     ]
-    technician = models.ForeignKey(Profile, on_delete=models.CASCADE, related_name="bookings")
-    user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE, related_name="bookings")
+    technician = models.ForeignKey(
+        Profile, on_delete=models.CASCADE, related_name="bookings"
+    )
+    user = models.ForeignKey(
+        settings.AUTH_USER_MODEL, on_delete=models.CASCADE, related_name="bookings"
+    )
     date_time_start = models.DateTimeField()
     date_time_end = models.DateTimeField()
     address = models.JSONField(blank=True, null=True)
     payment_done = models.BooleanField(default=False)
     price = models.DecimalField(max_digits=10, decimal_places=2)
-    service_status = models.CharField(max_length=20, choices=SERVICE_STATUS, default="pending")
+    service_status = models.CharField(
+        max_length=20, choices=SERVICE_STATUS, default="pending"
+    )
 
     def __str__(self):
-        return f"Booking {self.id} - {self.technician.user.username} for {self.user.username}"
+        return f"Booking {self.pk} - {self.technician} for {self.user}"

users/serializers.py

@@ -0,0 +1,88 @@
+import re
+
+from django.contrib.auth import get_user_model
+from rest_framework import serializers
+
+User = get_user_model()
+
+
+class RegisterSerializer(serializers.ModelSerializer):
+    confirm_password = serializers.CharField(write_only=True)
+
+    class Meta:
+        model = User
+        fields = [
+            "fullname",
+            "username",
+            "email",
+            "mobile",
+            "password",
+            "confirm_password",
+            "role",
+        ]
+        extra_kwargs = {
+            "password": {"write_only": True},
+        }
+
+    # ===== VALIDATIONS =====
+    def validate_mobile(self, value):
+        if not value.isdigit() or len(value) < 10:
+            raise serializers.ValidationError(
+                "Mobile number must be at least 10 digits and numeric."
+            )
+        if User.objects.filter(mobile=value).exists():
+            raise serializers.ValidationError("Mobile number already exists.")
+        return value
+
+    def validate_email(self, value):
+        regex = r"^[a-zA-Z0-9_.+-]+@[a-zA-Z0-9-]+\.[a-zA-Z0-9-.]+$"
+        if User.objects.filter(email=value).exists():
+            raise serializers.ValidationError("Email already exists.")
+        if not re.match(regex, value):
+            raise serializers.ValidationError("Enter a valid email address.")
+        return value
+
+    def validate_password(self, value):
+        if len(value) < 8:
+            raise serializers.ValidationError(
+                "Password must be at least 8 characters long."
+            )
+        return value
+
+    def validate(self, attrs):
+        if attrs.get("password") != attrs.get("confirm_password"):
+            raise serializers.ValidationError(
+                {"confirm_password": "Passwords do not match."}
+            )
+        return attrs
+
+    def create(self, validated_data):
+        validated_data.pop("confirm_password")
+        password = validated_data.pop("password")
+        user = User(**validated_data)
+        user.set_password(password)
+        user.save()
+        return user
+
+
+class LoginSerializer(serializers.Serializer):
+    identifier = serializers.CharField(required=True, allow_blank=False)
+    password = serializers.CharField(required=True, allow_blank=False, min_length=8)
+
+    def validate_identifier(self, value):
+        if not value or not value.strip():
+            raise serializers.ValidationError(
+                "Username, email, or mobile number is required."
+            )
+        return value.strip()
+
+    def validate_password(self, value):
+        if not value or not value.strip():
+            raise serializers.ValidationError("Password is required.")
+        return value
+
+    def create(self, validated_data):
+        return validated_data  # Login does not create objects
+
+    def update(self, instance, validated_data):
+        return instance  # Login does not update objects