forked from Mirrors/cloudflare-ddns-docker
56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
import re
|
|
|
|
import requests
|
|
from requests import Request
|
|
from requests.auth import AuthBase
|
|
|
|
from .constants import VERIFY_TOKEN
|
|
|
|
DURATION_REGEX = re.compile(
|
|
r"((?P<days>\d+?) ?(days|day|D|d) ?)?"
|
|
r"((?P<hours>\d+?) ?(hours|hour|H|h) ?)?"
|
|
r"((?P<minutes>\d+?) ?(minutes|minute|min|M|m) ?)?"
|
|
r"((?P<seconds>\d+?) ?(seconds|second|sec|S|s))?"
|
|
)
|
|
UNIT_TO_SECONDS = {
|
|
"days": 86400,
|
|
"hours": 3600,
|
|
"minutes": 60,
|
|
"seconds": 1
|
|
}
|
|
|
|
|
|
def parse_duration(duration: str) -> int:
|
|
"""
|
|
Parameter type for durations.
|
|
|
|
The converter supports the following symbols for each unit of time:
|
|
- days: `d`, `D`, `day`, `days`
|
|
- hours: `H`, `h`, `hour`, `hours`
|
|
- minutes: `M`, `m`, `minute`, `minutes`, `min`
|
|
- seconds: `S`, `s`, `second`, `seconds`, `sec`
|
|
The units need to be provided in descending order of magnitude.
|
|
"""
|
|
match = DURATION_REGEX.fullmatch(duration)
|
|
if not match:
|
|
raise ValueError(f"{duration} isn't a valid duration.")
|
|
|
|
duration = 0
|
|
for unit, time_value in match.groupdict().items():
|
|
if time_value:
|
|
duration += int(time_value) * UNIT_TO_SECONDS[unit]
|
|
|
|
return duration
|
|
|
|
|
|
class BearerAuth(AuthBase):
|
|
"""Bearer based authentication."""
|
|
|
|
def __init__(self, token: str) -> None:
|
|
self.token = token
|
|
|
|
def __call__(self, r: Request) -> Request:
|
|
"""Attach the Authorization header to the request."""
|
|
r.headers["Authorization"] = f"Bearer {self.token}"
|
|
return r
|