Added filetype support

files/models.py

@@ -41,11 +41,13 @@ class File(db.Model):
     filename = db.Column(db.String, nullable=False)
     private = db.Column(db.Integer, nullable=False)
     category_id = db.Column(db.Integer, db.ForeignKey('categories.id'), nullable=False)
+    file_type_id = db.Column(db.Integer, db.ForeignKey('file_types.id'), nullable=False)
 
-    def __init__(self, filename=None, private=None, category_id=None):
+    def __init__(self, filename=None, private=None, category_id=None, file_type_id=None):
         self.filename = filename
         self.private = private
         self.category_id = category_id
+        self.file_type_id = file_type_id 
 
     def __repr__(self):
         return f'<File name:{self.filename}>'
@@ -62,3 +64,15 @@ class Category(db.Model):
 
     def __repr__(self):
         return f'<Category name:{self.name}>'
+
+class FileType(db.Model):
+    __tablename__ = 'file_types'
+    id = db.Column(db.Integer, primary_key=True)
+    name = db.Column(db.String, nullable=False)
+    files = db.relationship('File', backref='type', lazy=True)
+
+    def __init__(self, name=None):
+        self.name = name
+
+    def __repr__(self):
+        return f'<FileType name:{self.name}>'