Merge pull request #434 from HebaruSan/feature/flask-2

Author: DasSkelett

KerbalStuff/app.py

@@ -15,7 +15,8 @@
 from flask_login import LoginManager, current_user
 from flaskext.markdown import Markdown
 from sqlalchemy import desc
-from werkzeug.exceptions import HTTPException, InternalServerError
+from werkzeug.exceptions import HTTPException, InternalServerError, NotFound
+from flask.typing import ResponseReturnValue
 from jinja2 import ChainableUndefined
 
 from .blueprints.accounts import accounts
@@ -129,8 +130,8 @@ def load_user(username: str) -> User:
 # 1         | 1   | 1             | 4XX in API -> jsonified_exception(e)
 
 
-@app.errorhandler(404)
-def handle_404(e: HTTPException) -> Union[Tuple[str, int], werkzeug.wrappers.Response]:
+@app.errorhandler(NotFound)
+def handle_404(e: NotFound) -> ResponseReturnValue:
     # Switch out the default message
     if e.description == werkzeug.exceptions.NotFound.description:
         e.description = "Requested page not found. Looks like this was deleted, or maybe was never here."
@@ -143,7 +144,7 @@ def handle_404(e: HTTPException) -> Union[Tuple[str, int], werkzeug.wrappers.Res
 
 # This one handles the remaining 4XX errors. JSONified for XHR requests, otherwise the user gets a nice error screen.
 @app.errorhandler(HTTPException)
-def handle_http_exception(e: HTTPException) -> Union[Tuple[str, int], werkzeug.wrappers.Response]:
+def handle_http_exception(e: HTTPException) -> ResponseReturnValue:
     if e.code and e.code >= 500:
         return handle_generic_exception(e)
     if request.path.startswith("/api/") \
@@ -155,7 +156,7 @@ def handle_http_exception(e: HTTPException) -> Union[Tuple[str, int], werkzeug.w
 # And this one handles everything leftover, that means, real otherwise unhandled exceptions.
 # https://flask.palletsprojects.com/en/1.1.x/errorhandling/#unhandled-exceptions
 @app.errorhandler(Exception)
-def handle_generic_exception(e: Union[Exception, HTTPException]) -> Union[Tuple[str, int], werkzeug.wrappers.Response]:
+def handle_generic_exception(e: Exception) -> ResponseReturnValue:
     site_logger.exception(e)
     try:
         db.rollback()
@@ -179,7 +180,7 @@ def handle_generic_exception(e: Union[Exception, HTTPException]) -> Union[Tuple[
 
 
 @app.teardown_request
-def teardown_request(exception: Optional[Exception]) -> None:
+def teardown_request(exception: Optional[BaseException]) -> None:
     db.close()
 
 

KerbalStuff/blueprints/login_oauth.py

@@ -66,8 +66,6 @@ def get_github_oath() -> Tuple[str, OAuthRemoteApp]:
     if resp is None:
         raise Exception(
             f"Access denied: reason={request.args['error']} error={request.args['error_description']}")
-    if 'error' in resp:
-        return jsonify(resp)
     session['github_token'] = (resp['access_token'], '')
     gh_info = github.get('user').data
     return gh_info['login'], github

KerbalStuff/common.py

@@ -127,25 +127,6 @@ def wrapper(*args: str, **kwargs: int) -> werkzeug.wrappers.Response:
     return wrapper
 
 
-def cors(f: Callable[..., Any]) -> Callable[..., Any]:
-    @wraps(f)
-    def wrapper(*args: str, **kwargs: int) -> Tuple[str, int]:
-        res = f(*args, **kwargs)
-        if request.headers.get('x-cors-status', False):
-            if isinstance(res, tuple):
-                json_text = res[0].data
-                code = res[1]
-            else:
-                json_text = res.data
-                code = 200
-            o = json.loads(json_text)
-            o['x-status'] = code
-            return jsonify(o)
-        return res
-
-    return wrapper
-
-
 def paginate_query(query: Query, page_size: int = 30) -> Tuple[List[Mod], int, int]:
     total_pages = math.ceil(query.count() / page_size)
     page = get_page()

KerbalStuff/database.py

@@ -1,6 +1,5 @@
 from sqlalchemy import create_engine
-from sqlalchemy.ext.declarative import declarative_base
-from sqlalchemy.orm import scoped_session, sessionmaker
+from sqlalchemy.orm import scoped_session, sessionmaker, declarative_base
 
 from .config import _cfg
 

KerbalStuff/kerbdown.py

@@ -1,11 +1,11 @@
 import urllib.parse
 from urllib.parse import parse_qs, urlparse
-from typing import Dict, Any, Match, Tuple
+from typing import Dict, Any, Match, Tuple, Optional
 
 from markdown import Markdown
 from markdown.extensions import Extension
 from markdown.inlinepatterns import InlineProcessor
-from markdown.util import etree
+from xml.etree import ElementTree
 
 
 class EmbedInlineProcessor(InlineProcessor):
@@ -23,18 +23,19 @@ def __init__(self, md: Markdown, configs: Dict[str, Any]) -> None:
         super().__init__(self.EMBED_RE, md)
         self.config = configs
 
-    def handleMatch(self, m: Match[str], data: str) -> Tuple[etree.Element, int, int]:  # type: ignore[override]
+    def handleMatch(self, m: Match[str], data: str) -> Tuple[ElementTree.Element, int, int]:  # type: ignore[override]
         d = m.groupdict()
         url = d.get('url')
+        el: Optional[ElementTree.Element]
         if not url:
-            el = etree.Element('span')
+            el = ElementTree.Element('span')
             el.text = "[[]]"
             return el, m.start(0), m.end(0)
         try:
             link = urlparse(url)
             host = link.hostname
         except:
-            el = etree.Element('span')
+            el = ElementTree.Element('span')
             el.text = "[[" + url + "]]"
             return el, m.start(0), m.end(0)
         el = None
@@ -44,16 +45,16 @@ def handleMatch(self, m: Match[str], data: str) -> Tuple[etree.Element, int, int
         except:
             pass
         if el is None:
-            el = etree.Element('span')
+            el = ElementTree.Element('span')
             el.text = "[[" + url + "]]"
         return el, m.start(0), m.end(0)
 
     def _get_youtube_id(self, link: urllib.parse.ParseResult) -> str:
         return (link.path if link.netloc == 'youtu.be'
                 else parse_qs(link.query)['v'][0])
 
-    def _embed_youtube(self, vid_id: str) -> etree.Element:
-        el = etree.Element('iframe')
+    def _embed_youtube(self, vid_id: str) -> ElementTree.Element:
+        el = ElementTree.Element('iframe')
         el.set('width', '100%')
         el.set('height', '600')
         el.set('frameborder', '0')

KerbalStuff/stubs/jinja2.pyi

@@ -10,8 +10,7 @@
 
 from packaging import version
 from sqlalchemy import orm, Column, Integer, Unicode, DateTime, String, ForeignKey
-from sqlalchemy.ext.declarative import declarative_base
-from sqlalchemy.orm import relationship, backref
+from sqlalchemy.orm import relationship, backref, declarative_base
 
 # revision identifiers, used by Alembic.
 revision = '85be165bc5dc'

alembic/versions/2020_06_23_17_49_36-85be165bc5dc.py

@@ -14,7 +14,7 @@
 from alembic import op
 import sqlalchemy as sa
 
-Base = sa.ext.declarative.declarative_base()
+Base = sa.orm.declarative_base()
 
 
 class ModVersion(Base):  # type: ignore

alembic/versions/2020_07_20_19_00_00-73c9d707134b.py

@@ -13,7 +13,7 @@
 from alembic import op
 import sqlalchemy as sa
 
-Base = sa.ext.declarative.declarative_base()
+Base = sa.orm.declarative_base()
 
 
 class User(Base):  # type: ignore

alembic/versions/2021_06_28_19_00_00-17fbd4ff8193.py

@@ -6,14 +6,14 @@ celery
 click
 dnspython
 flameprof
-Flask<2 # Needs testing before upgrading
+Flask
 Flask-Login
 Flask-Markdown
 Flask-OAuthlib
 future
 GitPython
 gunicorn
-Jinja2<3 # See Flask
+Jinja2
 Markdown
 MarkupSafe
 oauthlib