Merge pull request 'master' (#13) from master into prod

Reviewed-on: #13

.gitignore

@@ -119,3 +119,4 @@ GitHub.sublime-settings
 .history
 
 *pb2*
+schemas

Makefile

@@ -1,4 +1,6 @@
 gen:
-	python -m grpc_tools.protoc --proto_path schemas --python_out=. --pyi_out=. --grpc_python_out=. ./schemas/tasks.proto
+	curl https://platform.sprinthub.ru/generator >> generator.py
+	python generator.py
+	rm generator.py
 run:
 	python ./server.py

requirements.txt

@@ -1,13 +1,18 @@
 annotated-types==0.7.0
+certifi==2024.8.30
+charset-normalizer==3.4.0
 dnspython==2.7.0
 grpcio==1.68.1
 grpcio-tools==1.68.1
+idna==3.10
 motor==3.6.0
 protobuf==5.29.1
 pydantic==2.10.3
 pydantic_core==2.27.1
-pymongo==4.9.2
+pymongo==4.9
 python-dateutil==2.9.0.post0
+requests==2.32.3
 setuptools==75.6.0
 six==1.17.0
 typing_extensions==4.12.2
+urllib3==2.2.3

schemas/tasks.proto

@@ -1,40 +0,0 @@
-syntax = "proto3";
-
-package queues;
-
-import "google/protobuf/struct.proto";
-
-service Tasks {
-    rpc Put (PutRequest) returns (EmptyResponse) {}
-
-    rpc Take (TakeRequest) returns (TakeResponse) {}
-
-    rpc Finish (FinishRequest) returns (EmptyResponse) {}
-}
-
-message Task {
-    string id = 1;
-    int64 attempt = 2;
-    google.protobuf.Struct payload = 3;
-}
-
-message PutRequest {
-    string queue = 1;
-    int64 seconds_to_execute = 2;
-    optional int64 delay = 3;
-    google.protobuf.Struct payload = 4;
-}
-
-message TakeRequest {
-    string queue = 1;
-}
-
-message FinishRequest {
-    string id = 1;
-}
-
-message EmptyResponse {}
-
-message TakeResponse {
-    optional Task task = 1;
-}

server.py

@@ -3,8 +3,8 @@ import datetime
 import grpc
 import bson
 
-import tasks_pb2
+from queues import tasks_pb2
-import tasks_pb2_grpc
+from queues import tasks_pb2_grpc
 
 from utils import time
 from storage.mongo import tasks