Skip to content

Commit 36c5a4b

Browse files
committed
lint fix
1 parent 70b5484 commit 36c5a4b

5 files changed

Lines changed: 31 additions & 8 deletions

File tree

selfservice/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@
7676
# limiter = Limiter(
7777
# get_remote_address, app=app, default_limits=["50 per day", "10 per hour"]
7878
# )
79-
#else:
79+
# else:
8080
# limiter = Limiter(get_remote_address, app=app, default_limits=[])
8181

8282
# Initialize QR Code Generator
@@ -115,7 +115,7 @@ def index():
115115

116116

117117
@app.route("/health")
118-
#@limiter.exempt
118+
# @limiter.exempt
119119
def health():
120120
"""
121121
Shows an ok status if the application is up and running

selfservice/blueprints/recovery.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
"""
22
Flask blueprint for handling identity verification and account recovery.
33
"""
4+
45
import datetime
56
import uuid
67
import logging
@@ -266,7 +267,12 @@ def admin():
266267
session_id = str(uuid.uuid4())
267268

268269
# Create the object in the database.
269-
session_data = RecoverySession(id=session_id, username=request.form["username"], expires=datetime.datetime.now()+datetime.timedelta(hours=int(request.form["expireTime"])))
270+
session_data = RecoverySession(
271+
id=session_id,
272+
username=request.form["username"],
273+
expires=datetime.datetime.now()
274+
+ datetime.timedelta(hours=int(request.form["expireTime"])),
275+
)
270276
db.session.add(session_data)
271277
db.session.commit()
272278

@@ -278,7 +284,8 @@ def admin():
278284
{
279285
"username": s.username,
280286
"session_expired": (
281-
s.session_expires < datetime.datetime.now() or s.token_expires < datetime.datetime.now()
287+
s.session_expires < datetime.datetime.now()
288+
or s.token_expires < datetime.datetime.now()
282289
),
283290
"token_exists": s.token_id is not None,
284291
"token_expires": s.token_expires,

selfservice/models.py

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
"""
22
SQLAlchemy Database Models
33
"""
4+
45
import datetime
56
from datetime import timedelta
67

@@ -16,6 +17,7 @@
1617
from sqlalchemy.sql.functions import now
1718
from selfservice import db
1819

20+
1921
class ResetToken(db.Model):
2022
"""
2123
Reset tokens are generated once an identity has been verified. They allow
@@ -25,14 +27,19 @@ class ResetToken(db.Model):
2527
__tablename__ = "token"
2628
id = Column(Integer, primary_key=True)
2729
username = Column(String(64), nullable=False)
28-
expires = Column(DateTime, default=func.timezone("UTC", now()+timedelta(minutes=10)), nullable=False)
30+
expires = Column(
31+
DateTime,
32+
default=func.timezone("UTC", now() + timedelta(minutes=10)),
33+
nullable=False,
34+
)
2935
token = Column(String(36))
3036
session = Column(String(36), ForeignKey("session.id"))
3137
used = Column(Boolean)
3238

3339
def is_expired(self) -> bool:
3440
return self.expires < datetime.datetime.now()
3541

42+
3643
class RecoverySession(db.Model):
3744
"""
3845
A recovery session is created once a user enters their username and
@@ -43,7 +50,11 @@ class RecoverySession(db.Model):
4350
__tablename__ = "session"
4451
id = Column(String(36), primary_key=True)
4552
username = Column(String(64), nullable=False)
46-
expires = Column(DateTime, default=func.timezone("UTC", now()+timedelta(minutes=30)), nullable=False)
53+
expires = Column(
54+
DateTime,
55+
default=func.timezone("UTC", now() + timedelta(minutes=30)),
56+
nullable=False,
57+
)
4758

4859
def is_expired(self) -> bool:
4960
return self.expires < datetime.datetime.now()

selfservice/utilities/general.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
from twilio.rest import Client
1111
from flask import current_app
1212

13+
1314
def email_recovery(username, address, token):
1415
"""
1516
Send verification emails based on below template.
@@ -62,4 +63,4 @@ def phone_recovery(phone, token):
6263
m = client.messages.create(
6364
to=phone, from_=from_number, body=body, messaging_service_sid=service_sid
6465
)
65-
print(m)
66+
print(m)

selfservice/utilities/reset.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,11 @@ def generate_token(session):
6363

6464
# Create the object in the database.
6565
reset = ResetToken(
66-
username=session.username, token=token, session=session.id, used=False, expires=session.expires
66+
username=session.username,
67+
token=token,
68+
session=session.id,
69+
used=False,
70+
expires=session.expires,
6771
)
6872
db.session.add(reset)
6973
db.session.commit()

0 commit comments

Comments
 (0)