From 40a2e158e2877ec892c6b3d2276b68f483385e8f Mon Sep 17 00:00:00 2001
From: zhouchengrong <Zchengrong@yeah.net>
Date: Tue, 4 Jun 2024 09:39:40 +0800
Subject: [PATCH] =?UTF-8?q?feat=20generate=20single=20logo=20=E9=83=A8?=
 =?UTF-8?q?=E7=BD=B2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/api/api_slogan.py         | 24 -----------------------
 app/schemas/slogan.py         |  7 -------
 app/service/slogan/service.py | 36 -----------------------------------
 3 files changed, 67 deletions(-)
 delete mode 100644 app/api/api_slogan.py
 delete mode 100644 app/schemas/slogan.py
 delete mode 100644 app/service/slogan/service.py

diff --git a/app/api/api_slogan.py b/app/api/api_slogan.py
deleted file mode 100644
index 31459ba..0000000
--- a/app/api/api_slogan.py
+++ /dev/null
@@ -1,24 +0,0 @@
-import logging
-import time
-from fastapi import APIRouter, BackgroundTasks
-
-from app.schemas.slogan import SloganModel
-from app.service.slogan.service import Slogan
-
-router = APIRouter()
-logger = logging.getLogger()
-
-
-@router.post("/slogan")
-def slogan(request_item: SloganModel, background_tasks: BackgroundTasks):
-    try:
-        logger.info(f"request data ### : {request_item}")
-        service = Slogan(request_item)
-        background_tasks.add_task(service.get_result)
-        code = 200
-        message = "access"
-    except Exception as e:
-        code = 400
-        message = e
-        logger.warning(e)
-    return {"code": code, "message": message}
diff --git a/app/schemas/slogan.py b/app/schemas/slogan.py
deleted file mode 100644
index e80423d..0000000
--- a/app/schemas/slogan.py
+++ /dev/null
@@ -1,7 +0,0 @@
-from pydantic import BaseModel
-
-
-class SloganModel(BaseModel):
-    prompt: str
-    svg: str
-    tasks_id: str
diff --git a/app/service/slogan/service.py b/app/service/slogan/service.py
deleted file mode 100644
index 5a330d6..0000000
--- a/app/service/slogan/service.py
+++ /dev/null
@@ -1,36 +0,0 @@
-import json
-import logging
-
-import redis
-
-from app.core.config import *
-
-logger = logging.getLogger()
-
-
-class Slogan:
-    def __init__(self, request_data):
-        self.tasks_id = request_data.tasks_id
-        self.prompt = request_data.prompt
-        self.svg = request_data.svg
-        self.redis_client = redis.StrictRedis(host=REDIS_HOST, port=REDIS_PORT, db=REDIS_DB, decode_responses=True)
-        self.slogan_data = {'tasks_id': self.tasks_id, 'status': 'PENDING', 'message': "pending", 'image_url': ''}
-        self.redis_client.set(self.tasks_id, json.dumps(self.slogan_data))
-        self.redis_client.expire(self.tasks_id, 600)
-
-        # if DEBUG is False:
-        self.connection = pika.BlockingConnection(pika.ConnectionParameters(**RABBITMQ_PARAMS))
-        self.channel = self.connection.channel()
-        self.result_image_url = "test/slogan/init_img.png"
-
-    def read_tasks_status(self):
-        status_data = self.redis_client.get(self.tasks_id)
-        return json.loads(status_data), status_data
-
-    def get_result(self):
-        self.slogan_data['status'] = "SUCCESS"
-        self.slogan_data['message'] = "success"
-        self.slogan_data['image_url'] = "test/slogan/init_img.png"
-        dict_slogan_data, str_slogan_data = self.read_tasks_status()
-        self.channel.basic_publish(exchange='', routing_key=SLOGAN_RABBITMQ_QUEUES, body=str_slogan_data)
-        logger.info(f" [x] Sent {json.dumps(dict_slogan_data, indent=4)}")