1 files changed, 8 insertions, 6 deletions
diff --git a/fbin/db.py b/fbin/db.py
index 53b0d8b..9493f65 100644
--- a/fbin/db.py
+++ b/fbin/db.py
@@ -1,15 +1,15 @@
-from contextlib import contextmanager
 import datetime
 import mimetypes
 import os
 
-from flask import current_app
-from sqlalchemy.orm.exc import NoResultFound
-from sqlalchemy.exc import IntegrityError
+# FIXME: reimported from elsewhere
+from sqlalchemy.orm.exc import NoResultFound  # noqa: F401
+from sqlalchemy.exc import IntegrityError  # noqa: F401
 from flask_sqlalchemy import SQLAlchemy
 
 db = SQLAlchemy()
 
+
 class User(db.Model):
     __tablename__ = 'users'
 
@@ -23,6 +23,7 @@ class User(db.Model):
         self.username = username
         self.jab_id = jab_id
 
+
 class UserSession(db.Model):
     __tablename__ = 'sessions'
 
@@ -37,6 +38,7 @@ class UserSession(db.Model):
         self.access_token = access_token
         self.refresh_token = refresh_token
 
+
 class File(db.Model):
     __tablename__ = 'files'
 
@@ -51,7 +53,7 @@ class File(db.Model):
     scanned = db.Column(db.Boolean, nullable=False, default=False)
     blocked_reason = db.Column(db.JSON)
 
-    def __init__(self, hash, filename, size, date, user_id = None, ip = None):
+    def __init__(self, hash, filename, size, date, user_id=None, ip=None):
         self.hash = hash
         self.filename = filename
         self.size = size
@@ -80,7 +82,7 @@ class File(db.Model):
         return self.date.strftime('%Y-%m-%d %H:%M:%S UTC')
 
     def get_mime_type(self):
-        return mimetypes.guess_type(self.filename, strict = False)[0] or 'application/octet-stream'
+        return mimetypes.guess_type(self.filename, strict=False)[0] or 'application/octet-stream'
 
     def is_image(self):
         return self.get_mime_type().startswith('image')