aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--jnfilter/__init__.py43
-rw-r--r--setup.py3
2 files changed, 19 insertions, 27 deletions
diff --git a/jnfilter/__init__.py b/jnfilter/__init__.py
index 2c7641a..35be6bd 100644
--- a/jnfilter/__init__.py
+++ b/jnfilter/__init__.py
@@ -1,15 +1,13 @@
import re
import httpx
-import uvicorn
from functools import reduce
from typing import List, Iterator, Union
from xml.etree.ElementTree import ElementTree, fromstring, tostring, register_namespace
-from fastapi import FastAPI
-from starlette.responses import Response, PlainTextResponse
+from flask import Flask, Response, request
-app = FastAPI()
+app = Flask(__name__)
URL = "https://jovemnerd.com.br/feed-nerdcast/"
@@ -33,9 +31,6 @@ register_namespace("itunes", ITUNES)
register_namespace("atom", ATOM)
-class XMLResponse(Response):
- media_type = "application/xml"
-
def match(title: str, series: List[str]) -> bool:
def _match(s):
@@ -74,36 +69,34 @@ def filter_titles_xml(xml_str) -> Iterator[str]:
yield item.find("title").text
-async def load_and_filter(series: str, tag: Union[bool, None] = False) -> str:
+def load_and_filter(series: str, tag: Union[bool, None] = False) -> str:
series = series or 'nerdcast'
series = series.split(',')
- async with httpx.AsyncClient() as client:
- response = await client.get(URL)
+ with httpx.Client() as client:
+ response =client.get(URL)
xml_str = response.content
return filter_xml(xml_str, series, tag)
-async def load_titles() -> Iterator[str]:
- async with httpx.AsyncClient() as client:
- response = await client.get(URL)
+def load_titles() -> Iterator[str]:
+ with httpx.Client() as client:
+ response = client.get(URL)
xml_str = response.content
return filter_titles_xml(xml_str)
-@app.head("/")
-@app.get("/", response_class=XMLResponse)
-async def root(q: str = '', tag: Union[bool, None] = False):
- return await load_and_filter(q, tag)
+@app.route("/", methods=['GET', 'HEAD'])
+def root(q: str = '', tag: Union[bool, None] = False):
+ q = request.args.get("q", "")
+ tag = request.args.get("tag", False)
+ return load_and_filter(q, tag), 200, {'Content-Type': 'application/xml'}
-@app.get("/titles", response_class=PlainTextResponse)
-async def titles():
- titles = await load_titles()
+@app.route("/titles", methods=['GET'])
+def titles():
+ titles = load_titles()
return "\n".join(titles)
-@app.get("/series")
-async def titles():
+@app.route("/series", methods=['GET'])
+def series():
return [i[0] for i in RegexCollection.items()]
-
-def run():
- uvicorn.run(app=app, host="0.0.0.0", port=32000)
diff --git a/setup.py b/setup.py
index 07add82..7692d27 100644
--- a/setup.py
+++ b/setup.py
@@ -2,8 +2,7 @@ from setuptools import setup
requirements = [
'httpx==0.21.1',
- 'fastapi==0.70.0',
- 'uvicorn==0.15.0'
+ 'flask==2.2.2',
]