recovery: handle error when user doesn't exist
Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org>
This commit is contained in:
parent
8da00585e3
commit
a515afae0b
|
@ -37,7 +37,11 @@ class Command(BaseCommand):
|
||||||
duration = int(options.get("duration", 1))
|
duration = int(options.get("duration", 1))
|
||||||
_now = now()
|
_now = now()
|
||||||
expiry = _now + timedelta(days=duration * 365.2425)
|
expiry = _now + timedelta(days=duration * 365.2425)
|
||||||
user = User.objects.get(username=options.get("user"))
|
users = User.objects.filter(username=options.get("user"))
|
||||||
|
if not users.exists():
|
||||||
|
self.stderr.write(f"User '{options.get('user')}' not found.")
|
||||||
|
return
|
||||||
|
user = users.first()
|
||||||
token = Token.objects.create(
|
token = Token.objects.create(
|
||||||
expires=expiry,
|
expires=expiry,
|
||||||
user=user,
|
user=user,
|
||||||
|
|
Reference in a new issue