Lots more functionality - working towards UI

1 files changed, 6 insertions, 2 deletions

diff --git a/models.py b/models.py
index 6061661..27f1436 100644
--- a/models.py
+++ b/models.py
@@ -6,6 +6,7 @@ from sqlalchemy import (
     Text,
     DateTime,
 )
+import datetime
 
 from database import Base
 
@@ -23,7 +24,10 @@ class Link(Base):
     link = Column(String, primary_key=True)
     owner = Column(Integer, ForeignKey("users.id"), nullable=False)
     redirect_link = Column(String, nullable=False)
-    expire_date = Column(DateTime, nullable=False)
+    expire_date = Column(
+        DateTime,
+        default=datetime.datetime.utcnow() + datetime.timedelta(days=30),
+    )
 
 
 class Log(Base):
@@ -31,7 +35,7 @@ class Log(Base):
     id = Column(Integer, primary_key=True)
     owner = Column(Integer, ForeignKey("users.id"), nullable=False)
     link = Column(String, ForeignKey("links.link"), nullable=False)
-    timestamp = Column(DateTime, nullable=False)
+    timestamp = Column(DateTime, default=datetime.datetime.utcnow())
     ip = Column(String, nullable=False)
     location = Column(String, nullable=False)
     browser = Column(String, nullable=False)