110 lines
3.5 KiB
Python
110 lines
3.5 KiB
Python
import asyncio
|
|
from aiohttp import web
|
|
from damination import DaminationAPI
|
|
|
|
|
|
routes = web.RouteTableDef()
|
|
toonat = DaminationAPI()
|
|
app = web.Application()
|
|
|
|
|
|
@routes.get('/get_subscriber/{widget_id}')
|
|
async def get_widgets(request):
|
|
widget_id = request.match_info['widget_id']
|
|
subscribers = toonat.get_subscribers(widget_id)
|
|
return web.json_response(subscribers)
|
|
|
|
@routes.get('/subscribe/{widget_id}')
|
|
async def subscribe(request):
|
|
"""
|
|
주어진 widget_id와 url로 구독을 추가합니다.
|
|
"""
|
|
widget_id = request.match_info['widget_id']
|
|
url = request.query['url']
|
|
success = toonat.add_subscriber(widget_id, url)
|
|
if success:
|
|
return web.Response(text=f"Subscribed to widget ID {widget_id} with URL {url}.")
|
|
else:
|
|
return web.Response(text=f"Already subscribed to widget ID {widget_id} with URL {url}.")
|
|
|
|
@routes.get('/unsubscribe/{widget_id}')
|
|
async def unsubscribe(request):
|
|
"""
|
|
주어진 widget_id와 url로 구독을 취소합니다.
|
|
"""
|
|
widget_id = request.match_info['widget_id']
|
|
url = request.query['url']
|
|
success = toonat.remove_subscriber(widget_id, url)
|
|
if success:
|
|
return web.Response(text=f"Unsubscribed from widget ID {widget_id} with URL {url}.")
|
|
else:
|
|
return web.Response(text=f"Not subscribed to widget ID {widget_id} with URL {url}.")
|
|
|
|
@routes.get('/get_widgets')
|
|
async def get_widgets(request):
|
|
"""
|
|
현재 설정된 모든 위젯 ID 목록을 반환하는 웹 API 엔드포인트입니다.
|
|
"""
|
|
widget_ids = toonat.get_widget_ids()
|
|
return web.json_response(widget_ids)
|
|
|
|
|
|
@routes.get('/add_widget')
|
|
async def add_widget(request):
|
|
"""
|
|
주어진 widget_id를 위젯 ID 목록에 추가하고 알림을 가져오는 작업을 시작하는 웹 API 엔드포인트입니다.
|
|
widget_id: 추가할 위젯 ID
|
|
"""
|
|
widget_id = request.query['widget_id']
|
|
await toonat.add_widget_and_start_fetching_notifications(widget_id)
|
|
return web.Response(text=f"Widget ID {widget_id} added and fetching started.")
|
|
|
|
@routes.get('/remove_widget')
|
|
async def remove_widget(request):
|
|
"""
|
|
주어진 widget_id를 위젯 ID 목록에서 제거하고 알림을 가져오는 작업을 중단하는 웹 API 엔드포인트입니다.
|
|
widget_id: 제거할 위젯 ID
|
|
"""
|
|
widget_id = request.query['widget_id']
|
|
await toonat.remove_widget_and_stop_fetching_notifications(widget_id)
|
|
return web.Response(text=f"Widget ID {widget_id} removed and fetching stopped.")
|
|
|
|
@routes.get('/status')
|
|
async def status(request):
|
|
"""
|
|
서버 상태를 확인하는 웹 API 엔드포인트입니다. 이 엔드포인트는 서버가 정상적으로 작동 중임을 확인하는 데 사용됩니다.
|
|
"""
|
|
return web.HTTPOk()
|
|
|
|
|
|
async def web_server(app):
|
|
"""
|
|
aiohttp 웹 서버를 설정하고 시작하는 비동기 함수입니다.
|
|
"""
|
|
app.add_routes(routes)
|
|
runner = web.AppRunner(app)
|
|
await runner.setup()
|
|
site = web.TCPSite(runner, '0.0.0.0', 80)
|
|
await site.start()
|
|
|
|
|
|
async def main():
|
|
|
|
# DaminationAPI를 실행합니다.
|
|
api_task = asyncio.create_task(toonat.run())
|
|
|
|
# 웹 서버를 실행합니다.
|
|
web_server_task = asyncio.create_task(web_server(app))
|
|
|
|
await api_task
|
|
await web_server_task
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if not toonat.WIDGET_IDS:
|
|
print("No widget ID found in settings.ini.")
|
|
print("Please enter the widget ID via web endpoint (http://localhost/add_widget/YOUR_WIDGET_ID)")
|
|
|
|
asyncio.run(main())
|