working locally

Author: raphaeltm

samples/django-channels-redis/app/Dockerfile

@@ -1,43 +1,34 @@
-# Use an official Python runtime as a parent image
-FROM python:3.10.4-slim-buster AS builder
+# Use a smaller base image
+FROM python:3.10.4-slim-buster
 
 # Set environment variables
 ENV PYTHONDONTWRITEBYTECODE 1
 ENV PYTHONUNBUFFERED 1
 
 # Set work directory
-WORKDIR /code
+WORKDIR /app
+
+RUN apt-get update && apt-get install -y \
+    gcc \
+    python3-dev \
+    libpq-dev
 
 # Install dependencies
-COPY requirements.txt /code/
+COPY requirements.txt /app/
+RUN pip install --upgrade pip
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy project
-COPY . /code/
+# Copy project files
+COPY . /app/
 
 # Collect static files
 RUN python manage.py collectstatic --noinput
 
-
-# Use a smaller base image for the final stage
-FROM python:3.10.4-slim-buster
-
-# Set environment variables
-ENV PYTHONDONTWRITEBYTECODE 1
-ENV PYTHONUNBUFFERED 1
-
-# Set work directory
-WORKDIR /code
-
-# Copy the dependencies from the builder stage
-COPY --from=builder /usr/local/lib/python3.10/site-packages /usr/local/lib/python3.10/site-packages
-COPY --from=builder /code /code
-
 # Create a non-root user
 RUN adduser --disabled-password --gecos '' django
 
 # Set the user to the non-root user
 USER django
 
 # Start server
-CMD ["sh", "-c", "python manage.py migrate && gunicorn defang_sample.wsgi:application --bind 0.0.0.0:8000"]
+CMD ["sh", "-c", "python manage.py migrate && gunicorn django_defang.wsgi:application --bind 0.0.0.0:8000"]

samples/django-channels-redis/app/chat/__init__.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

samples/django-channels-redis/app/chat/admin.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class ChatConfig(AppConfig):
+    default_auto_field = "django.db.models.BigAutoField"
+    name = "chat"

samples/django-channels-redis/app/chat/apps.py

@@ -0,0 +1,69 @@
+import json
+
+from asgiref.sync import async_to_sync
+from channels.generic.websocket import WebsocketConsumer
+from pydantic import BaseModel
+from .models import ChatMessage
+
+class MessagePayload(BaseModel):
+    user: str = "Anonymous"
+    message: str
+    timestamp: float = None
+
+class GroupPayload(BaseModel):
+    type: str
+    user: str = "Anonymous"
+    message: str
+    timestamp: float = None
+
+class ChatConsumer(WebsocketConsumer):
+    def connect(self):
+        self.room_name = self.scope["url_route"]["kwargs"]["room_name"]
+        self.room_group_name = f"chat_{self.room_name}"
+        # Join room group
+        async_to_sync(self.channel_layer.group_add)(
+            self.room_group_name, self.channel_name
+        )
+        self.accept()
+
+    def disconnect(self, close_code):
+        # Leave room group
+        async_to_sync(self.channel_layer.group_discard)(
+            self.room_group_name, self.channel_name
+        )
+
+    # Receive message from WebSocket
+    def receive(self, text_data):
+        text_data_json = json.loads(text_data)
+        payload = MessagePayload(**text_data_json)
+        message = payload.message
+
+        persisted = ChatMessage.objects.create(
+            user=payload.user,
+            message=message,
+            room=self.room_name,
+        )
+
+        group_payload = GroupPayload(
+            type="chat_message", 
+            message=message,
+            user=payload.user,
+            timestamp=int(persisted.timestamp.timestamp() * 1000)
+        )
+
+        # Send message to room group
+        async_to_sync(self.channel_layer.group_send)(
+            self.room_group_name, group_payload.model_dump()
+        )
+
+    # Receive message from room group
+    def chat_message(self, event):
+        payload = GroupPayload(**event)
+        message_payload = MessagePayload(
+            user=payload.user,
+            message=payload.message,
+            timestamp=payload.timestamp,
+        )
+
+        # Send message to WebSocket
+        self.send(text_data=message_payload.model_dump_json())

samples/django-channels-redis/app/chat/consumers.py

@@ -0,0 +1,31 @@
+# Generated by Django 5.1 on 2024-08-22 22:02
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = []
+
+    operations = [
+        migrations.CreateModel(
+            name="ChatMessage",
+            fields=[
+                (
+                    "id",
+                    models.BigAutoField(
+                        auto_created=True,
+                        primary_key=True,
+                        serialize=False,
+                        verbose_name="ID",
+                    ),
+                ),
+                ("user", models.CharField(max_length=50)),
+                ("message", models.TextField()),
+                ("room", models.CharField(max_length=50)),
+                ("timestamp", models.DateTimeField(auto_now_add=True)),
+            ],
+        ),
+    ]

samples/django-channels-redis/app/chat/migrations/0001_initial.py

@@ -0,0 +1,10 @@
+from django.db import models
+
+class ChatMessage(models.Model):
+    user = models.CharField(max_length=50)
+    message = models.TextField()
+    room = models.CharField(max_length=50)
+    timestamp = models.DateTimeField(auto_now_add=True)
+
+    def __str__(self):
+        return f"{self.user} - {self.timestamp.strftime('%Y-%m-%d %H:%M:%S')}"

samples/django-channels-redis/app/chat/migrations/__init__.py

@@ -0,0 +1,7 @@
+from django.urls import re_path
+
+from . import consumers
+
+websocket_urlpatterns = [
+    re_path(r"ws/(?P<room_name>\w+)/$", consumers.ChatConsumer.as_asgi()),
+]

samples/django-channels-redis/app/chat/models.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.