first commit

api/admin.py

@@ -0,0 +1,7 @@
+from django.contrib import admin
+
+from .models import User, Prediction, UserPrediction
+
+admin.site.register(User)
+admin.site.register(Prediction)
+admin.site.register(UserPrediction)

api/apps.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class ApiConfig(AppConfig):
+    default_auto_field = 'django.db.models.BigAutoField'
+    name = 'api'

api/build_qstring.py

@@ -0,0 +1,38 @@
+import urllib.parse
+
+def build_query_string(data: dict) -> str:
+    required_keys = [
+        "profile",
+        "pred_type",
+        "launch_datetime",
+        "launch_latitude",
+        "launch_longitude",
+        "launch_altitude",
+        "ascent_rate",
+        "burst_altitude",
+        "descent_rate"
+    ]
+
+    # Проверяем, что все ключи на месте
+    missing_keys = [key for key in required_keys if key not in data]
+    if missing_keys:
+        raise ValueError(f"Missing required keys: {', '.join(missing_keys)}")
+
+    # Собираем строку запроса
+    return urllib.parse.urlencode({k: data[k] for k in required_keys})
+
+# Пример:
+json_data = {
+    "profile": "standard_profile",
+    "pred_type": "single",
+    "launch_datetime": "2025-03-16T08:47:00Z",
+    "launch_latitude": "56.6992",
+    "launch_longitude": "38.8247",
+    "launch_altitude": "0",
+    "ascent_rate": "5",
+    "burst_altitude": "30000",
+    "descent_rate": "5"
+}
+
+query_string = build_query_string(json_data)
+print(query_string)

api/migrations/0001_initial.py

@@ -0,0 +1,22 @@
+# Generated by Django 5.1.7 on 2025-03-15 08:54
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Todo',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('title', models.CharField(max_length=100)),
+                ('completed', models.BooleanField(default=False)),
+            ],
+        ),
+    ]

api/migrations/0002_prediction_user_userprediction_delete_todo.py

@@ -0,0 +1,46 @@
+# Generated by Django 5.1.7 on 2025-03-31 10:14
+
+import django.db.models.deletion
+import uuid
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('api', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Prediction',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
+                ('result', models.JSONField()),
+                ('deleted_at', models.DateTimeField(blank=True, null=True)),
+            ],
+        ),
+        migrations.CreateModel(
+            name='User',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+            ],
+        ),
+        migrations.CreateModel(
+            name='UserPrediction',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('prediction', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='api.prediction')),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='api.user')),
+            ],
+            options={
+                'unique_together': {('user', 'prediction')},
+            },
+        ),
+        migrations.DeleteModel(
+            name='Todo',
+        ),
+    ]

api/models.py

@@ -0,0 +1,20 @@
+import uuid
+from django.db import models
+
+class User(models.Model):
+    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
+
+class Prediction(models.Model):
+    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
+    created_at = models.DateTimeField(auto_now_add=True)
+    updated_at = models.DateTimeField(auto_now=True)
+    result = models.JSONField()
+    deleted_at = models.DateTimeField(null=True, blank=True)
+
+class UserPrediction(models.Model):
+    user = models.ForeignKey(User, on_delete=models.CASCADE)
+    prediction = models.ForeignKey(Prediction, on_delete=models.CASCADE)
+    created_at = models.DateTimeField(auto_now_add=True)
+
+    class Meta:
+        unique_together = ('user', 'prediction')

api/serializers.py

@@ -0,0 +1,7 @@
+from rest_framework import serializers
+from .models import Prediction
+
+class PredictionSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Prediction
+        fields = ['id', 'created_at', 'updated_at', 'result']

api/tests.py

@@ -0,0 +1,10 @@
+from django.test import TestCase
+from rest_framework.test import APIClient
+from .models import User
+
+class PredictionTest(TestCase):
+    def setUp(self):
+        self.client = APIClient()
+        self.user = User.objects.create()
+
+

api/urls.py

@@ -0,0 +1,10 @@
+from django.urls import path
+from .views import PredictionCreateView, PredictionListView, PredictionDeleteView
+from rest_framework.authtoken.views import obtain_auth_token
+
+urlpatterns = [
+    path('predictions', PredictionCreateView.as_view(), name='create_prediction'),
+    path('predictions', PredictionListView.as_view(), name='get_predictions'),
+    path('predictions/<uuid:pk>', PredictionDeleteView.as_view(), name='delete_prediction'),
+    path('api/token/', obtain_auth_token),
+]

api/views.py

@@ -0,0 +1,63 @@
+from rest_framework import status, generics
+from rest_framework.response import Response
+from rest_framework.views import APIView
+from django.utils import timezone
+from .models import Prediction, User, UserPrediction
+from .serializers import PredictionSerializer
+from rest_framework.permissions import IsAuthenticated
+import requests
+
+def get_prediction_from_tawhiri(params):
+    base_url = "https://fly.stratonautica.ru/api/v2"
+    response = requests.get(base_url, params=params)
+    
+    if response.status_code == 200:
+        return response.json()  # получаем результат предсказания
+    else:
+        raise Exception(f"Tawhiri error: {response.status_code} {response.text}")
+
+
+class PredictionCreateView(APIView):
+    def post(self, request):
+        user_id = request.data.get('user_id')
+        user = User.objects.get(id=user_id)
+
+        # Передаём остальные параметры (кроме user_id) в Tawhiri
+        tawhiri_params = {k: v for k, v in request.data.items() if k != 'user_id'}
+        
+        try:
+            prediction_result = get_prediction_from_tawhiri(tawhiri_params)
+        except Exception as e:
+            return Response({"error": str(e)}, status=500)
+
+        prediction = Prediction.objects.create(result=prediction_result)
+        UserPrediction.objects.create(user=user, prediction=prediction)
+
+        return Response(PredictionSerializer(prediction).data)
+
+class PredictionListView(APIView):
+    def get(self, request):
+        user_id = request.query_params.get('user_id')
+        created_from = request.query_params.get('created_from')
+        created_till = request.query_params.get('created_till')
+
+        predictions = Prediction.objects.filter(
+            id__in=UserPrediction.objects.filter(user_id=user_id).values_list('prediction_id'),
+            created_at__gte=created_from,
+            created_at__lte=created_till,
+            deleted_at__isnull=True
+        )
+        return Response(PredictionSerializer(predictions, many=True).data)
+
+class PredictionDeleteView(APIView):
+    def delete(self, request, pk):
+        try:
+            prediction = Prediction.objects.get(pk=pk)
+            prediction.deleted_at = timezone.now()
+            prediction.save()
+            return Response({"deleted": True})
+        except Prediction.DoesNotExist:
+            return Response({"error": "Not found"}, status=404)
+
+class PredictionCreateView(APIView):
+    permission_classes = [IsAuthenticated]