add feed in / out button page and also button to redirect to button page

This commit is contained in:
Alberto-Audrix committed 2026-09-02 13:33:41 +07:00
1 parent 07b8e2dc25
commit b86bd7a2f4
17 files changed
+719 -28

No files matched your search

@@ -0,0 +1,15 @@
from django.db import migrations, models
class Migration(migrations.Migration):
dependencies = [
("farms", "0012_cycle_feed_initial_balance_date"),
]
operations = [
migrations.AddField(
model_name="kandang",
name="feed_in_button_url",
field=models.URLField(blank=True, default="", max_length=500),
),
]
@@ -0,0 +1,29 @@
from django.db import migrations, models
def migrate_single_url_to_list(apps, schema_editor):
Kandang = apps.get_model("farms", "Kandang")
for kandang in Kandang.objects.exclude(feed_in_button_url="").exclude(feed_in_button_url__isnull=True):
url = (kandang.feed_in_button_url or "").strip()
if url:
kandang.feed_in_button_urls = [{"label": "", "url": url}]
kandang.save(update_fields=["feed_in_button_urls"])
class Migration(migrations.Migration):
dependencies = [
("farms", "0013_kandang_feed_in_button_url"),
]
operations = [
migrations.AddField(
model_name="kandang",
name="feed_in_button_urls",
field=models.JSONField(blank=True, default=list),
),
migrations.RunPython(migrate_single_url_to_list, migrations.RunPython.noop),
migrations.RemoveField(
model_name="kandang",
name="feed_in_button_url",
),
]
+1
View File
@@ -24,6 +24,7 @@ class Site(models.Model):
class Kandang(models.Model):
kandang_id = models.BigAutoField(primary_key=True)
kandang_name = models.CharField(max_length=30)
feed_in_button_urls = models.JSONField(default=list, blank=True)
site = models.ForeignKey(Site, on_delete=models.CASCADE, related_name="kandangs")
created_at = models.DateTimeField(auto_now_add=True)
updated_at = models.DateTimeField(auto_now=True)
+45 -1
View File
@@ -11,12 +11,56 @@ class SiteSerializer(PkAsIdMixin, serializers.ModelSerializer):
read_only_fields = ["id", "created_at", "updated_at"]
class FeedInButtonUrlSerializer(serializers.Serializer):
label = serializers.CharField(max_length=50, required=False, allow_blank=True, default="")
url = serializers.URLField(max_length=500)
class KandangSerializer(PkAsIdMixin, serializers.ModelSerializer):
feed_in_button_urls = FeedInButtonUrlSerializer(many=True, required=False)
class Meta:
model = Kandang
fields = ["id", "kandang_name", "site", "created_at", "updated_at"]
fields = [
"id",
"kandang_name",
"feed_in_button_urls",
"site",
"created_at",
"updated_at",
]
read_only_fields = ["id", "created_at", "updated_at"]
def validate_feed_in_button_urls(self, value):
cleaned = []
for item in value or []:
url = str(item.get("url", "")).strip()
if not url:
continue
label = str(item.get("label", "") or "").strip()[:50]
cleaned.append({"label": label, "url": url})
return cleaned
def update(self, instance, validated_data):
urls = validated_data.pop("feed_in_button_urls", None)
instance = super().update(instance, validated_data)
if urls is not None:
instance.feed_in_button_urls = urls
instance.save(update_fields=["feed_in_button_urls", "updated_at"])
return instance
def to_representation(self, instance):
data = super().to_representation(instance)
urls = data.get("feed_in_button_urls") or []
if not isinstance(urls, list):
urls = []
data["feed_in_button_urls"] = [
{"label": str(item.get("label", "") or ""), "url": str(item.get("url", "") or "")}
for item in urls
if isinstance(item, dict) and str(item.get("url", "") or "").strip()
]
return data
class FlockSerializer(PkAsIdMixin, serializers.ModelSerializer):
class Meta:
@@ -0,0 +1,77 @@
from django.test import Client, TestCase
from apps.accounts.models import User
from apps.farms.models import Kandang, Site
class KandangFeedInButtonUrlTests(TestCase):
def setUp(self):
self.owner = User.objects.create_user(user_name="owner", password="secret")
self.admin = User.objects.create_user(
user_name="admin",
password="admin123",
status=User.STATUS_SUPERADMIN,
is_superuser=True,
)
self.staff = User.objects.create_user(
user_name="staff",
password="staff123",
status=User.STATUS_ACTIVE,
is_staff=True,
)
self.site = Site.objects.create(site_name="Site A", user=self.owner)
self.kandang = Kandang.objects.create(kandang_name="Kandang 1", site=self.site)
self.payload = {
"feed_in_button_urls": [{"label": "Gudang A", "url": "http://192.168.1.10:8080/"}]
}
def _csrf_headers(self, client: Client) -> dict[str, str]:
token = client.cookies["csrftoken"].value
return {"HTTP_X_CSRFTOKEN": token}
def _login(self, client: Client, user_name: str, password: str) -> None:
client.get("/api/v1/auth/csrf/")
response = client.post(
"/api/v1/auth/login/",
{"user_name": user_name, "password": password},
content_type="application/json",
**self._csrf_headers(client),
)
self.assertEqual(response.status_code, 200, response.content)
def test_superadmin_can_save_feed_in_button_urls(self):
client = Client(enforce_csrf_checks=True)
self._login(client, "admin", "admin123")
response = client.patch(
f"/api/v1/kandangs/{self.kandang.pk}/",
self.payload,
content_type="application/json",
**self._csrf_headers(client),
)
self.assertEqual(response.status_code, 200, response.content)
self.kandang.refresh_from_db()
self.assertEqual(self.kandang.feed_in_button_urls, self.payload["feed_in_button_urls"])
def test_staff_cannot_save_feed_in_button_urls(self):
client = Client(enforce_csrf_checks=True)
self._login(client, "staff", "staff123")
response = client.patch(
f"/api/v1/kandangs/{self.kandang.pk}/",
self.payload,
content_type="application/json",
**self._csrf_headers(client),
)
self.assertEqual(response.status_code, 403, response.content)
self.kandang.refresh_from_db()
self.assertEqual(self.kandang.feed_in_button_urls, [])
def test_staff_cannot_rename_kandang(self):
client = Client(enforce_csrf_checks=True)
self._login(client, "staff", "staff123")
response = client.patch(
f"/api/v1/kandangs/{self.kandang.pk}/",
{"kandang_name": "Renamed"},
content_type="application/json",
**self._csrf_headers(client),
)
self.assertEqual(response.status_code, 403, response.content)