Merge pull request #49 from aclark4life/main

Misc updates

Author: aclark4life

.gitignore

@@ -1,5 +1,5 @@
 django_mongodb_cli.egg-info/
-django_mongodb_cli/__pycache__/
+__pycache__
 /src/
 .idea
 server.log

demo/__init__.py

@@ -0,0 +1,7 @@
+from django.contrib import admin
+from .models import Patient
+
+
+@admin.register(Patient)
+class PatientAdmin(admin.ModelAdmin):
+    pass

demo/medical_records/__init__.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class MedicalRecordsConfig(AppConfig):
+    default_auto_field = "django_mongodb_backend.fields.ObjectIdAutoField"
+    name = "demo.medical_records"

demo/medical_records/admin.py

@@ -0,0 +1,77 @@
+import os
+import random
+from django.core.management.base import BaseCommand
+from faker import Faker
+
+from django_mongodb_demo.models import Patient, PatientRecord, Billing
+
+
+class Command(BaseCommand):
+    help = "Create patients with embedded patient records and billing using Faker. Optionally set MONGODB_URI."
+
+    def add_arguments(self, parser):
+        parser.add_argument(
+            "num_patients", type=int, help="Number of patients to create"
+        )
+        parser.add_argument(
+            "--flush",
+            action="store_true",
+            help="Delete all existing patients before creating new ones",
+        )
+        parser.add_argument(
+            "--mongodb-uri",
+            type=str,
+            help="MongoDB connection URI to set as MONGODB_URI env var",
+        )
+
+    def handle(self, *args, **options):
+        fake = Faker()
+
+        num_patients = options["num_patients"]
+
+        # Set MONGODB_URI if provided
+        if options.get("mongodb_uri"):
+            os.environ["MONGODB_URI"] = options["mongodb_uri"]
+            self.stdout.write(
+                self.style.SUCCESS(f"MONGODB_URI set to: {options['mongodb_uri']}")
+            )
+
+        # Optionally flush
+        if options["flush"]:
+            Patient.objects.all().delete()
+            self.stdout.write(self.style.WARNING("Deleted all existing patients."))
+
+        for _ in range(num_patients):
+            # Create a Billing object
+            billing = Billing(
+                cc_type=fake.credit_card_provider(), cc_number=fake.credit_card_number()
+            )
+
+            # Create a PatientRecord object
+            record = PatientRecord(
+                ssn=fake.ssn(),
+                billing=billing,
+                bill_amount=round(random.uniform(50.0, 5000.0), 2),
+            )
+
+            # Create Patient
+            patient = Patient(
+                patient_name=fake.name(),
+                patient_id=random.randint(100000, 999999),
+                patient_record=record,
+            )
+            patient.save()
+
+            self.stdout.write(
+                self.style.SUCCESS(
+                    f"Created Patient: {patient.patient_name} ({patient.patient_id})"
+                )
+            )
+            self.stdout.write(f"  SSN: {record.ssn}")
+            self.stdout.write(f"  Billing CC Type: {billing.cc_type}")
+            self.stdout.write(f"  Billing CC Number: {billing.cc_number}")
+            self.stdout.write(f"  Bill Amount: ${record.bill_amount}")
+
+        self.stdout.write(
+            self.style.SUCCESS(f"Successfully created {num_patients} patient(s).")
+        )

demo/medical_records/apps.py

@@ -0,0 +1,27 @@
+from django.db import models
+from django_mongodb_backend.models import EmbeddedModel
+from django_mongodb_backend.fields import (
+    EmbeddedModelField,
+    EncryptedEmbeddedModelField,
+    EncryptedCharField,
+)
+
+
+class Patient(models.Model):
+    patient_name = models.CharField(max_length=255)
+    patient_id = models.BigIntegerField()
+    patient_record = EmbeddedModelField("PatientRecord")
+
+    def __str__(self):
+        return f"{self.patient_name} ({self.patient_id})"
+
+
+class PatientRecord(EmbeddedModel):
+    ssn = EncryptedCharField(max_length=11)
+    billing = EncryptedEmbeddedModelField("Billing")
+    bill_amount = models.DecimalField(max_digits=10, decimal_places=2)
+
+
+class Billing(EmbeddedModel):
+    cc_type = models.CharField(max_length=50)
+    cc_number = models.CharField(max_length=20)

demo/medical_records/management/commands/create_patient.py

@@ -0,0 +1,14 @@
+from django.test import TestCase
+from .models import Author, Article
+
+
+class DemoTest(TestCase):
+    def test_create_author_and_article(self):
+        author = Author.objects.create(name="Alice", email="alice@example.com")
+        article = Article.objects.create(
+            title="Hello MongoDB",
+            slug="hello-mongodb",
+            author=author,
+            content="Testing MongoDB backend.",
+        )
+        self.assertEqual(article.author.name, "Alice")

demo/medical_records/models.py

@@ -0,0 +1,7 @@
+from django.urls import path
+from . import views
+
+urlpatterns = [
+    path("", views.article_list, name="article_list"),
+    path("article/<slug:slug>/", views.article_detail, name="article_detail"),
+]

demo/medical_records/tests.py

@@ -0,0 +1,15 @@
+from django.shortcuts import render, get_object_or_404
+from .models import Article
+
+
+def article_list(request):
+    articles = Article.objects.all().order_by("-published_at")
+    return render(request, "demo/article_list.html", {"articles": articles})
+
+
+def article_detail(request, slug):
+    article = get_object_or_404(Article, slug=slug)
+    comments = article.comments.all().order_by("-created_at")
+    return render(
+        request, "demo/article_detail.html", {"article": article, "comments": comments}
+    )