diff --git a/bumper/__main__.py b/bumper/__main__.py index 3d89911..a5a5c1a 100644 --- a/bumper/__main__.py +++ b/bumper/__main__.py @@ -1,4 +1,6 @@ #!/usr/bin/env python3 +"""Main module.""" + import bumper if __name__ == "__main__": diff --git a/bumper/web/auth_util.py b/bumper/web/auth_util.py index 596f6f3..8904d93 100644 --- a/bumper/web/auth_util.py +++ b/bumper/web/auth_util.py @@ -36,6 +36,7 @@ _logger = get_logger("confserver") def generate_token(user): + """Generate token.""" try: tmpaccesstoken = uuid.uuid4().hex user_add_token(user["userid"], tmpaccesstoken) @@ -46,6 +47,7 @@ def generate_token(user): def generate_authcode(user, countrycode, token): + """Generate auth token.""" try: tmpauthcode = f"{countrycode}_{uuid.uuid4().hex}" user_add_authcode(user["userid"], token, tmpauthcode) @@ -56,6 +58,7 @@ def generate_authcode(user, countrycode, token): async def login(request): + """Perform login.""" try: user_devid = request.match_info.get("devid", "") countrycode = request.match_info.get("country", "us") @@ -124,6 +127,7 @@ async def login(request): async def get_authcode(request: Request) -> Response: + """Get auth code.""" try: user_devid = request.match_info.get("devid", None) # Ecovacs if not user_devid: diff --git a/bumper/web/server.py b/bumper/web/server.py index dc4a582..fe6aa68 100644 --- a/bumper/web/server.py +++ b/bumper/web/server.py @@ -6,7 +6,6 @@ import json import logging import os import ssl -from typing import Union import aiohttp_jinja2 import jinja2 diff --git a/requirements.txt b/requirements.txt index fa3e80e..1240c5b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ aiohttp==3.8.1 aiohttp-jinja2==1.5 -git+https://github.com/Yakifo/amqtt@master#amqtt==11.0.0 cachetools==5.1.0 +git+https://github.com/Yakifo/amqtt@master#amqtt==11.0.0 gmqtt==0.6.11 Jinja2==3.1.2 tinydb==4.7.0