tidy up new user instantiation
This commit is contained in:
parent
4e75ea61b5
commit
bb1f488761
@ -44,7 +44,7 @@ def main():
|
|||||||
controller = Controller(config)
|
controller = Controller(config)
|
||||||
|
|
||||||
# Init the user:
|
# Init the user:
|
||||||
user = User(controller, args.calling_station.upper())
|
user = User(config, controller)
|
||||||
|
|
||||||
# Init the UI console
|
# Init the UI console
|
||||||
console = Console(config, controller, user)
|
console = Console(config, controller, user)
|
||||||
|
|||||||
@ -23,15 +23,15 @@ import sqlalchemy
|
|||||||
from datetime import datetime, timezone
|
from datetime import datetime, timezone
|
||||||
|
|
||||||
from rsbbs import __version__
|
from rsbbs import __version__
|
||||||
from rsbbs import Controller
|
from rsbbs import Config, Controller
|
||||||
|
|
||||||
from rsbbs.models import User as SAUser
|
from rsbbs.models import User as SAUser
|
||||||
|
|
||||||
|
|
||||||
class User():
|
class User():
|
||||||
def __init__(self, controller: Controller, callsign: str):
|
def __init__(self, config: Config, controller: Controller):
|
||||||
self.controller = controller
|
self.controller = controller
|
||||||
self.callsign = callsign
|
self.callsign = config.args.calling_station.upper()
|
||||||
self.user = self.get_or_create_user()
|
self.user = self.get_or_create_user()
|
||||||
|
|
||||||
def __getattr__(self, __name: str) -> Any:
|
def __getattr__(self, __name: str) -> Any:
|
||||||
@ -42,7 +42,7 @@ class User():
|
|||||||
session.expire_on_commit = False
|
session.expire_on_commit = False
|
||||||
try:
|
try:
|
||||||
statement = sqlalchemy.select(SAUser).where(
|
statement = sqlalchemy.select(SAUser).where(
|
||||||
SAUser.callsign == self.callsign.upper())
|
SAUser.callsign == self.callsign)
|
||||||
exopts = {"prebuffer_rows": True}
|
exopts = {"prebuffer_rows": True}
|
||||||
result = session.execute(statement,
|
result = session.execute(statement,
|
||||||
execution_options=exopts)
|
execution_options=exopts)
|
||||||
@ -57,7 +57,7 @@ class User():
|
|||||||
else:
|
else:
|
||||||
logging.info(f"User not found.")
|
logging.info(f"User not found.")
|
||||||
user = SAUser(
|
user = SAUser(
|
||||||
callsign=self.callsign.upper(),
|
callsign=self.callsign,
|
||||||
login_count=1,
|
login_count=1,
|
||||||
)
|
)
|
||||||
session.add(user)
|
session.add(user)
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user