add aurweb.models.ban.Ban ORM mapping

Signed-off-by: Kevin Morris <kevr@0cost.org>
This commit is contained in:
Kevin Morris 2020-12-26 19:29:19 -08:00
parent a836892cde
commit adc9fccb7d
2 changed files with 82 additions and 0 deletions

19
aurweb/models/ban.py Normal file
View file

@ -0,0 +1,19 @@
from fastapi import Request
from sqlalchemy.orm import mapper
from aurweb.schema import Bans
class Ban:
def __init__(self, **kwargs):
self.IPAddress = kwargs.get("IPAddress")
self.BanTS = kwargs.get("BanTS")
def is_banned(request: Request):
from aurweb.db import session
ip = request.client.host
return session.query(Ban).filter(Ban.IPAddress == ip).first() is not None
mapper(Ban, Bans)