.gitignore

@@ -116,6 +116,4 @@ GitHub.sublime-settings
 !.vscode/tasks.json
 !.vscode/launch.json
 !.vscode/extensions.json
-.history
+.history
-
-*pb2*

.gitlab-ci.yml

@@ -0,0 +1,51 @@
+stages:
+  - build
+  - deploy-dev
+  - deploy-prod
+
+build:
+  stage: build
+  tags:
+    - dev
+  before_script:
+    - docker login -u mathwave -p $DOCKERHUB_PASSWORD
+  script:
+    - docker build -t mathwave/sprint-repo:pizda-bot .
+
+push:
+  stage: build
+  tags:
+    - dev
+  before_script:
+    - docker login -u mathwave -p $DOCKERHUB_PASSWORD
+  script:
+    - docker push mathwave/sprint-repo:pizda-bot
+
+.deploy:
+  before_script:
+    - docker login -u mathwave -p $DOCKERHUB_PASSWORD
+
+deploy-dev:
+  extends:
+    - .deploy
+  stage: deploy-dev
+  tags:
+    - dev
+  rules:
+    - if: '$CI_COMMIT_BRANCH == "master"'
+      when: on_success
+    - when: manual
+  script:
+    - docker stack deploy --with-registry-auth -c ./.deploy/deploy-dev.yaml pizda-bot
+
+deploy-prod:
+  extends:
+    - .deploy
+  stage: deploy-prod
+  tags:
+    - prod
+  only:
+    - master
+  when: manual
+  script:
+    - docker stack deploy --with-registry-auth -c ./.deploy/deploy-prod.yaml pizda-bot

api.py

@@ -2,7 +2,7 @@ from collections import defaultdict
 
 from flask import Flask
 
-from utils.mongo import mongo
+from mongo import mongo
 
 app = Flask(__name__)
 

daemons/base.py

@@ -1,13 +1,3 @@
-import os
+class BaseDaemon:
-
-
-stage = os.getenv("STAGE", 'local')
-if stage == 'local':
-    QUEUES_URL = 'localhost:50051'
-else:
-    QUEUES_URL = 'queues-grpc:50051'
-
-
-class Daemon:
     def execute(self):
-        raise NotImplemented
+        raise NotImplementedError

daemons/worker.py

@@ -6,9 +6,9 @@ from utils import queues
 from telebot.types import Message
 import json
 
-from utils.mongo import mongo
+from mongo import mongo
-from utils.sprint_platform import PlatformClient
+from sprint_platform import PlatformClient
-from utils.storage import set_values, get_chat_info
+from storage import set_values, get_chat_info
 
 security_token = os.getenv("PLATFORM_SECURITY_TOKEN")
 stage = os.getenv("STAGE", 'local')
@@ -39,7 +39,7 @@ def get_replies():
     return client.get_config('replies')
 
 
-class Daemon(base.Daemon, queues.TasksHandlerMixin):
+class Daemon(base.BaseDaemon, queues.TasksHandlerMixin):
     @property
     def queue_name(self):
         return 'pizda_bot_worker'

local_platform.json

@@ -0,0 +1,5 @@
+{
+    "configs": {},
+    "experiments": {},
+    "platform_staff": {}
+}

main.py

@@ -6,7 +6,7 @@ if arg == 'worker':
     from daemons import worker
     daemon = worker.Daemon()
 else:
-    from daemons.api import app
+    from api import app
     app.run(host="0.0.0.0", port=1238)
 
 daemon.execute()

mongo.py

@@ -1,14 +1,10 @@
 import pymongo
-import os
+import settings
-
-MONGO_USER = os.getenv("MONGO_USER", "mongo")
-MONGO_PASSWORD = os.getenv("MONGO_PASSWORD", "password")
-MONGO_HOST = os.getenv("MONGO_HOST", "localhost")
 
 
 class Mongo:
     def __init__(self):
-        url = f"mongodb://{MONGO_USER}:{MONGO_PASSWORD}@{MONGO_HOST}:27017/"
+        url = f"mongodb://{settings.MONGO_USER}:{settings.MONGO_PASSWORD}@{settings.MONGO_HOST}:27017/"
         self.client = pymongo.MongoClient(url)
         self.database = self.client.get_database("pizda-bot")
         self.chats_collection.create_index([

settings.py

@@ -0,0 +1,10 @@
+import os
+import sys
+import zoneinfo
+
+MONGO_USER = os.getenv("MONGO_USER", "mongo")
+MONGO_PASSWORD = os.getenv("MONGO_PASSWORD", "password")
+MONGO_HOST = os.getenv("MONGO_HOST", "localhost")
+
+CACHE_SIZE = int(os.getenv("CACHE_SIZE", 1000))
+CACHE_TTL = int(os.getenv("CACHE_TTL", 3600))

storage.py

@@ -1,12 +1,9 @@
 from cachetools import TTLCache
-import os
 
-from utils.mongo import mongo
+import settings
+from mongo import mongo
 
-CACHE_SIZE = int(os.getenv("CACHE_SIZE", 1000))
+cache = TTLCache(settings.CACHE_SIZE, settings.CACHE_TTL)
-CACHE_TTL = int(os.getenv("CACHE_TTL", 3600))
-
-cache = TTLCache(CACHE_SIZE, CACHE_TTL)
 
 
 def get_chat_info(chat_id: int) -> dict:

updater.py

@@ -2,7 +2,7 @@ import datetime
 import random
 from time import sleep
 
-from utils.mongo import mongo
+from mongo import mongo
 
 
 def update():

utils/queues.py

@@ -1,3 +1,4 @@
+import json
 import os
 import requests
 import time