feat: switch all timezone naive datetimes to timezone aware

This commit is contained in:
Iain Learmonth 2024-12-06 16:08:48 +00:00
parent 41fc0a73a5
commit e22abb383c
30 changed files with 322 additions and 226 deletions

View file

@ -1,20 +1,22 @@
import urllib.parse
from datetime import datetime
from typing import Optional, List
from datetime import datetime, timezone
from typing import List, Optional
import requests
import sqlalchemy
from flask import flash, redirect, url_for, render_template, Response, Blueprint
from flask import (Blueprint, Response, flash, redirect, render_template,
url_for)
from flask.typing import ResponseReturnValue
from flask_wtf import FlaskForm
from sqlalchemy import exc
from wtforms import StringField, SelectField, SubmitField, BooleanField, IntegerField
from wtforms import (BooleanField, IntegerField, SelectField, StringField,
SubmitField)
from wtforms.validators import DataRequired
from app.extensions import db
from app.models.base import Group
from app.models.mirrors import Origin, Country
from app.portal.util import response_404, view_lifecycle, LifecycleForm
from app.models.mirrors import Country, Origin
from app.portal.util import LifecycleForm, response_404, view_lifecycle
bp = Blueprint("origin", __name__)
@ -64,8 +66,8 @@ def origin_new(group_id: Optional[int] = None) -> ResponseReturnValue:
origin.auto_rotation = form.auto_rotate.data
origin.smart = form.smart_proxy.data
origin.assets = form.asset_domain.data
origin.created = datetime.utcnow()
origin.updated = datetime.utcnow()
origin.added = datetime.now(tz=timezone.utc)
origin.updated = datetime.now(tz=timezone.utc)
try:
db.session.add(origin)
db.session.commit()
@ -106,7 +108,7 @@ def origin_edit(origin_id: int) -> ResponseReturnValue:
origin.risk_level_override = form.risk_level_override_number.data
else:
origin.risk_level_override = None
origin.updated = datetime.utcnow()
origin.updated = datetime.now(tz=timezone.utc)
try:
db.session.commit()
flash(f"Saved changes for origin {origin.domain_name}.", "success")