maintainance

dev.py

@@ -1,4 +1,4 @@
-import os
+import argparse
 import subprocess
 from pathlib import Path
 from typing import Optional
@@ -42,7 +42,7 @@ def generate_certificates(domain="localhost", cert_file="localhost.pem", key_fil
     ('localhost.pem', 'localhost-key.pem')
     """
     # Проверяем, существуют ли сертификаты
-    if os.path.exists(cert_file) and os.path.exists(key_file):
+    if Path(cert_file).exists() and Path(key_file).exists():
         logger.info(f"Сертификаты уже существуют: {cert_file}, {key_file}")
         return cert_file, key_file
 
@@ -76,7 +76,7 @@ def generate_certificates(domain="localhost", cert_file="localhost.pem", key_fil
         return None, None
 
 
-def run_server(host="0.0.0.0", port=8000, workers=1) -> None:
+def run_server(host="localhost", port=8000, use_https=False, workers=1, domain="localhost") -> None:
     """
     Запускает сервер Granian с поддержкой HTTPS при необходимости
 
@@ -85,6 +85,7 @@ def run_server(host="0.0.0.0", port=8000, workers=1) -> None:
         port: Порт для запуска сервера
         use_https: Флаг использования HTTPS
         workers: Количество рабочих процессов
+        domain: Домен для сертификата
 
     >>> run_server(use_https=True)  # doctest: +SKIP
     """
@@ -94,31 +95,49 @@ def run_server(host="0.0.0.0", port=8000, workers=1) -> None:
         logger.warning("Многопроцессорный режим может вызвать проблемы сериализации приложения. Использую 1 процесс.")
         workers = 1
 
-    # При проблемах с ASGI можно попробовать использовать Uvicorn как запасной вариант
     try:
-        # Генерируем сертификаты с помощью mkcert
-        cert_file, key_file = generate_certificates()
+        if use_https:
+            # Генерируем сертификаты с помощью mkcert
+            cert_file, key_file = generate_certificates(domain=domain)
 
-        if not cert_file or not key_file:
-            logger.error("Не удалось сгенерировать сертификаты для HTTPS")
-            return
+            if not cert_file or not key_file:
+                logger.error("Не удалось сгенерировать сертификаты для HTTPS")
+                return
 
-        logger.info(f"Запуск HTTPS сервера на https://{host}:{port} с использованием Granian")
-        # Запускаем Granian сервер с явным указанием ASGI
-        server = Granian(
-            address=host,
-            port=port,
-            workers=workers,
-            interface=Interfaces.ASGI,
-            target="main:app",
-            ssl_cert=Path(cert_file),
-            ssl_key=Path(key_file),
-        )
+            logger.info(f"Запуск HTTPS сервера на https://{host}:{port} с использованием Granian")
+            # Запускаем Granian сервер с явным указанием ASGI
+            server = Granian(
+                address=host,
+                port=port,
+                workers=workers,
+                interface=Interfaces.ASGI,
+                target="main:app",
+                ssl_cert=Path(cert_file),
+                ssl_key=Path(key_file),
+            )
+        else:
+            logger.info(f"Запуск HTTP сервера на http://{host}:{port} с использованием Granian")
+            server = Granian(
+                address=host,
+                port=port,
+                workers=workers,
+                interface=Interfaces.ASGI,
+                target="main:app",
+            )
         server.serve()
     except Exception as e:
-        # В случае проблем с Granian, пробуем запустить через Uvicorn
+        # В случае проблем с Granian, логируем ошибку
         logger.error(f"Ошибка при запуске Granian: {e!s}")
 
 
 if __name__ == "__main__":
-    run_server()
+    parser = argparse.ArgumentParser(description="Запуск сервера разработки с поддержкой HTTPS")
+    parser.add_argument("--https", action="store_true", help="Использовать HTTPS")
+    parser.add_argument("--workers", type=int, default=1, help="Количество рабочих процессов")
+    parser.add_argument("--domain", type=str, default="localhost", help="Домен для сертификата")
+    parser.add_argument("--port", type=int, default=8000, help="Порт для запуска сервера")
+    parser.add_argument("--host", type=str, default="localhost", help="Хост для запуска сервера")
+
+    args = parser.parse_args()
+
+    run_server(host=args.host, port=args.port, use_https=args.https, workers=args.workers, domain=args.domain)