mirror of
https://github.com/tgbot-collection/YYeTsBot.git
synced 2025-11-25 11:29:38 +08:00
106 lines
2.8 KiB
Python
106 lines
2.8 KiB
Python
#!/usr/bin/env python3
|
|
# coding: utf-8
|
|
import logging
|
|
import os
|
|
from http import HTTPStatus
|
|
from pathlib import Path
|
|
|
|
from tornado import gen
|
|
from tornado.concurrent import run_on_executor
|
|
|
|
from handlers import cf
|
|
from handlers.base import BaseHandler
|
|
|
|
filename = Path(__file__).name.split(".")[0]
|
|
|
|
|
|
class ResourceHandler(BaseHandler):
|
|
filename = filename
|
|
|
|
@run_on_executor()
|
|
def get_resource_data(self):
|
|
resource_id = int(self.get_query_argument("id"))
|
|
username = self.get_current_user()
|
|
if str(resource_id) in os.getenv("HIDDEN_RESOURCE", "").split(","):
|
|
self.set_status(HTTPStatus.NOT_FOUND)
|
|
return {"status": 0, "info": "资源已隐藏"}
|
|
data = self.instance.get_resource_data(resource_id, username)
|
|
if not data:
|
|
self.ban()
|
|
self.set_status(HTTPStatus.NOT_FOUND)
|
|
data = {}
|
|
|
|
return data
|
|
|
|
@run_on_executor()
|
|
def search_resource(self):
|
|
referer = self.request.headers.get("referer")
|
|
if not referer:
|
|
cf.ban_new_ip(self.get_real_ip())
|
|
kw = self.get_query_argument("keyword").lower()
|
|
search_type = self.get_query_argument("type", "default")
|
|
self.set_header(
|
|
"search-engine", "Meilisearch" if os.getenv("MEILISEARCH") else "MongoDB"
|
|
)
|
|
return self.instance.search_resource(kw, search_type)
|
|
|
|
@gen.coroutine
|
|
def get(self):
|
|
if self.get_query_argument("id", None):
|
|
resp = yield self.get_resource_data()
|
|
elif self.get_query_argument("keyword", None):
|
|
resp = yield self.search_resource()
|
|
else:
|
|
resp = "error"
|
|
self.write(resp)
|
|
|
|
|
|
class ResourceLatestHandler(BaseHandler):
|
|
filename = filename
|
|
|
|
@run_on_executor()
|
|
def get_latest(self):
|
|
size = int(self.get_query_argument("size", "100"))
|
|
result = self.instance.get_latest_resource()
|
|
result["data"] = result["data"][:size]
|
|
return result
|
|
|
|
@gen.coroutine
|
|
def get(self):
|
|
resp = yield self.get_latest()
|
|
self.write(resp)
|
|
|
|
|
|
class TopHandler(BaseHandler):
|
|
filename = filename
|
|
|
|
def get_user_like(self) -> list:
|
|
username = self.get_current_user()
|
|
return self.instance.get_user_like(username)
|
|
|
|
def get_most(self) -> list:
|
|
return self.instance.get_most()
|
|
|
|
@run_on_executor()
|
|
def get_top_resource(self):
|
|
return self.instance.get_top_resource()
|
|
|
|
@gen.coroutine
|
|
def get(self):
|
|
resp = yield self.get_top_resource()
|
|
self.write(resp)
|
|
|
|
|
|
class NameHandler(BaseHandler):
|
|
filename = filename
|
|
|
|
@run_on_executor()
|
|
def get_names(self):
|
|
is_readable = self.get_query_argument("human", None)
|
|
return self.instance.get_names(is_readable)
|
|
|
|
@gen.coroutine
|
|
def get(self):
|
|
resp = yield self.get_names()
|
|
self.write(resp)
|