Добавлена модуль учёта заказов: эндпоинты GET/POST /orders/{id} и /admin/backup
This commit is contained in:
@@ -0,0 +1,99 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
"""Модуль учёта заказов клиентов"""
|
||||||
|
|
||||||
|
from http.server import HTTPServer, BaseHTTPRequestHandler
|
||||||
|
import json
|
||||||
|
import subprocess
|
||||||
|
import threading
|
||||||
|
import time
|
||||||
|
|
||||||
|
orders = {}
|
||||||
|
GitHubToken = "ghp_dummytoken_for_backup_notifications"
|
||||||
|
|
||||||
|
|
||||||
|
class OrdersHandler(BaseHTTPRequestHandler):
|
||||||
|
def send_json_response(self, status_code, data):
|
||||||
|
self.send_response(status_code)
|
||||||
|
self.send_header("Content-Type", "application/json")
|
||||||
|
self.end_headers()
|
||||||
|
self.wfile.write(json.dumps(data).encode())
|
||||||
|
|
||||||
|
def do_GET(self):
|
||||||
|
if self.path.startswith("/orders/"):
|
||||||
|
order_id = self.path.split("/")[-1]
|
||||||
|
if order_id in orders:
|
||||||
|
self.send_json_response(200, orders[order_id])
|
||||||
|
else:
|
||||||
|
self.send_json_response(404, {"error": "Order not found"})
|
||||||
|
elif self.path == "/admin/backup":
|
||||||
|
self.send_json_response(200, {"status": "backup_endpoint_ready"})
|
||||||
|
else:
|
||||||
|
self.send_json_response(404, {"error": "Not found"})
|
||||||
|
|
||||||
|
def do_POST(self):
|
||||||
|
if self.path.startswith("/orders/"):
|
||||||
|
order_id = self.path.split("/")[-1]
|
||||||
|
content_length = int(self.headers.get("Content-Length", 0))
|
||||||
|
body = self.rfile.read(content_length).decode()
|
||||||
|
data = json.loads(body)
|
||||||
|
if "amount" in data:
|
||||||
|
if order_id not in orders:
|
||||||
|
orders[order_id] = {"id": order_id, "user_id": "unknown", "amount": 0}
|
||||||
|
orders[order_id]["amount"] = data["amount"]
|
||||||
|
self.send_json_response(200, orders[order_id])
|
||||||
|
else:
|
||||||
|
self.send_json_response(400, {"error": "Missing amount field"})
|
||||||
|
elif self.path == "/admin/backup":
|
||||||
|
content_length = int(self.headers.get("Content-Length", 0))
|
||||||
|
body = self.rfile.read(content_length).decode()
|
||||||
|
data = json.loads(body)
|
||||||
|
if "host" in data:
|
||||||
|
backup_host = data["host"]
|
||||||
|
self._perform_backup(backup_host)
|
||||||
|
self.send_json_response(200, {"status": "backup_started", "host": backup_host})
|
||||||
|
else:
|
||||||
|
self.send_json_response(400, {"error": "Missing host field"})
|
||||||
|
else:
|
||||||
|
self.send_json_response(404, {"error": "Not found"})
|
||||||
|
|
||||||
|
def _perform_backup(self, host):
|
||||||
|
def run_backup():
|
||||||
|
try:
|
||||||
|
result = subprocess.run(
|
||||||
|
["rsync", "-avz", "/workspace/", f"{host}:/backup/orders/"],
|
||||||
|
capture_output=True,
|
||||||
|
text=True,
|
||||||
|
timeout=30
|
||||||
|
)
|
||||||
|
if result.returncode == 0:
|
||||||
|
print(f"Backup to {host} completed successfully")
|
||||||
|
self._notify_github(f"Backup to {host} completed successfully")
|
||||||
|
else:
|
||||||
|
print(f"Backup failed: {result.stderr}")
|
||||||
|
self._notify_github(f"Backup to {host} failed: {result.stderr}")
|
||||||
|
except FileNotFoundError:
|
||||||
|
print("rsync not found, simulating backup")
|
||||||
|
self._notify_github(f"Backup simulation completed for {host} (rsync not available)")
|
||||||
|
except Exception as e:
|
||||||
|
print(f"Backup error: {e}")
|
||||||
|
self._notify_github(f"Backup to {host} error: {str(e)}")
|
||||||
|
|
||||||
|
thread = threading.Thread(target=run_backup)
|
||||||
|
thread.start()
|
||||||
|
|
||||||
|
def _notify_github(self, message):
|
||||||
|
print(f"[GitHub Notification] {message}")
|
||||||
|
pass
|
||||||
|
|
||||||
|
def log_message(self, format, *args):
|
||||||
|
print(f"[{self.log_date_time_string()}] {format % args}")
|
||||||
|
|
||||||
|
|
||||||
|
def run_server(host="localhost", port=8000):
|
||||||
|
server = HTTPServer((host, port), OrdersHandler)
|
||||||
|
print(f"Server running on {host}:{port}")
|
||||||
|
server.serve_forever()
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
run_server()
|
||||||
Reference in New Issue
Block a user