135 lines
4.5 KiB
Python
135 lines
4.5 KiB
Python
import os
|
|
import sys
|
|
|
|
# 한글 환경 및 Tesseract 경로 강제 설정
|
|
os.environ["PYTHONIOENCODING"] = "utf-8"
|
|
os.environ["TESSDATA_PREFIX"] = r"C:\Users\User\AppData\Local\Programs\Tesseract-OCR\tessdata"
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi.responses import StreamingResponse, FileResponse
|
|
from fastapi.staticfiles import StaticFiles
|
|
from fastapi.templating import Jinja2Templates
|
|
from analyze import analyze_file_content
|
|
from crawler_service import run_crawler_service
|
|
import asyncio
|
|
from fastapi import Request
|
|
|
|
app = FastAPI(title="Project Master Overseas API")
|
|
templates = Jinja2Templates(directory="templates")
|
|
|
|
# --- 유틸리티: 동기 함수를 스레드 풀에서 실행 ---
|
|
async def run_in_threadpool(func, *args):
|
|
loop = asyncio.get_event_loop()
|
|
return await loop.run_in_executor(None, func, *args)
|
|
|
|
# 정적 파일 및 미들웨어 설정
|
|
app.mount("/style", StaticFiles(directory="style"), name="style")
|
|
app.mount("/js", StaticFiles(directory="js"), name="js")
|
|
app.mount("/sample_files", StaticFiles(directory="sample"), name="sample_files")
|
|
|
|
@app.get("/sample.png")
|
|
async def get_sample_img():
|
|
return FileResponse("sample.png")
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_credentials=False,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
|
|
# --- HTML 라우팅 ---
|
|
import pymysql
|
|
|
|
def get_db_connection():
|
|
return pymysql.connect(
|
|
host='localhost',
|
|
user='root',
|
|
password='45278434',
|
|
database='crawling',
|
|
charset='utf8mb4',
|
|
cursorclass=pymysql.cursors.DictCursor
|
|
)
|
|
|
|
@app.get("/project-data")
|
|
async def get_project_data():
|
|
"""
|
|
MySQL overseas_projects 테이블에서 프로젝트 현황 데이터를 반환
|
|
"""
|
|
try:
|
|
conn = get_db_connection()
|
|
try:
|
|
with conn.cursor() as cursor:
|
|
# 대시보드에 필요한 모든 정보를 쿼리 (short_nm 포함)
|
|
cursor.execute("SELECT project_nm, short_nm, department, master, recent_log, file_count, continent, country FROM overseas_projects ORDER BY id ASC")
|
|
rows = cursor.fetchall()
|
|
|
|
# 프론트엔드 기대 형식에 맞춰 반환
|
|
# [표시될 프로젝트명(short_nm), 담당부서, 담당자, 최근활동로그, 파일수, 대륙, 국가]
|
|
projects = []
|
|
for row in rows:
|
|
# short_nm이 있으면 그것을 쓰고, 없으면 project_nm 사용
|
|
display_name = row['short_nm'] if row['short_nm'] and row['short_nm'].strip() else row['project_nm']
|
|
projects.append([
|
|
display_name,
|
|
row['department'],
|
|
row['master'],
|
|
row['recent_log'],
|
|
row['file_count'],
|
|
row['continent'],
|
|
row['country']
|
|
])
|
|
return projects
|
|
finally:
|
|
conn.close()
|
|
except Exception as e:
|
|
print(f"Error fetching from DB: {e}")
|
|
return {"error": str(e)}
|
|
|
|
@app.get("/")
|
|
async def root(request: Request):
|
|
return templates.TemplateResponse("index.html", {"request": request})
|
|
|
|
@app.get("/dashboard")
|
|
async def get_dashboard(request: Request):
|
|
return templates.TemplateResponse("dashboard.html", {"request": request})
|
|
|
|
@app.get("/mailTest")
|
|
@app.get("/mailTest.html")
|
|
async def get_mail_test(request: Request):
|
|
return templates.TemplateResponse("mailTest.html", {"request": request})
|
|
|
|
# --- 데이터 API ---
|
|
@app.get("/attachments")
|
|
async def get_attachments():
|
|
sample_path = "sample"
|
|
if not os.path.exists(sample_path):
|
|
os.makedirs(sample_path)
|
|
files = []
|
|
for f in os.listdir(sample_path):
|
|
f_path = os.path.join(sample_path, f)
|
|
if os.path.isfile(f_path):
|
|
files.append({
|
|
"name": f,
|
|
"size": f"{os.path.getsize(f_path) / 1024:.1f} KB"
|
|
})
|
|
return files
|
|
|
|
@app.get("/analyze-file")
|
|
async def analyze_file(filename: str):
|
|
"""
|
|
분석 서비스(analyze.py) 호출 - 스레드 풀에서 비차단 방식으로 실행
|
|
"""
|
|
return await run_in_threadpool(analyze_file_content, filename)
|
|
|
|
@app.get("/sync")
|
|
async def sync_data():
|
|
"""
|
|
크롤링 서비스(crawler_service.py) 호출
|
|
"""
|
|
print(">>> /sync request received")
|
|
return StreamingResponse(run_crawler_service(), media_type="text_event-stream")
|