diff --git a/tiny0/database.db b/tiny0/database.db new file mode 100644 index 0000000..d195296 Binary files /dev/null and b/tiny0/database.db differ diff --git a/tiny0/models.py b/tiny0/models.py index f3c14c9..f40229d 100644 --- a/tiny0/models.py +++ b/tiny0/models.py @@ -1,7 +1,9 @@ from tiny0 import db -''' +class URL(db.Model): + id = db.Column(db.Integer, primary_key=True) + token = db.Column(db.String(8), index=True, unique=True, nullable=False) + url = db.Column(db.String(2000), nullable=False) -Declaration of models - -''' + def __repr__(self): + return f"'{self.id}' '{self.token}' '{self.url}'" diff --git a/tiny0/routes.py b/tiny0/routes.py index 7a34f1a..0f7bb1b 100644 --- a/tiny0/routes.py +++ b/tiny0/routes.py @@ -1,7 +1,7 @@ from flask import render_template, redirect, request, url_for -from tiny0 import app +from tiny0 import app, db from tiny0.forms import URLForm -#from tiny0.models import URLs +from tiny0.models import URL #from token import gen_valid_token # Index Page diff --git a/tiny0/token.py b/tiny0/token.py index d489b0d..29420ab 100644 --- a/tiny0/token.py +++ b/tiny0/token.py @@ -1,4 +1,4 @@ -#from tiny0.models import URLs +#from tiny0.models import URL from secrets import choice '''