diff --git a/acm_url/routes.py b/acm_url/routes.py index fe68d76..3720611 100644 --- a/acm_url/routes.py +++ b/acm_url/routes.py @@ -1,3 +1,4 @@ +from datetime import datetime from flask import Flask, render_template, request, url_for, redirect, session from acm_url import app, db import string @@ -82,6 +83,7 @@ def vanity(vanity): if entry is None: return render_template('404.html') + entry.last_visited = datetime.now() entry.visit_count = entry.visit_count + 1 db.session.commit() diff --git a/acm_url/schema.py b/acm_url/schema.py index f8909fb..b8ee5a0 100644 --- a/acm_url/schema.py +++ b/acm_url/schema.py @@ -9,6 +9,7 @@ class URL(db.Model): url = db.Column(db.Text, nullable=False) created = db.Column(db.DateTime, server_default=func.now(), nullable=False) visit_count = db.Column(db.Integer, default=0, nullable=False) + last_visited = db.Column(db.DateTime, nullable=True) def __repr__(self): - return ''.format(self.vanity) \ No newline at end of file + return ''.format(self.vanity) diff --git a/migrations/versions/2d3fd3b013f4_.py b/migrations/versions/2d3fd3b013f4_.py new file mode 100644 index 0000000..20e2bdc --- /dev/null +++ b/migrations/versions/2d3fd3b013f4_.py @@ -0,0 +1,30 @@ +"""empty message + +Revision ID: 2d3fd3b013f4 +Revises: ed69a26bebbe +Create Date: 2022-02-27 16:17:41.435362 + +""" +from alembic import op +import sqlalchemy as sa + + +# revision identifiers, used by Alembic. +revision = '2d3fd3b013f4' +down_revision = 'ed69a26bebbe' +branch_labels = None +depends_on = None + + +def upgrade(): + # ### commands auto generated by Alembic - please adjust! ### + op.add_column('URL', sa.Column('last_visited', sa.DateTime(), nullable=True)) + op.drop_column('URL', 'last_visted') + # ### end Alembic commands ### + + +def downgrade(): + # ### commands auto generated by Alembic - please adjust! ### + op.add_column('URL', sa.Column('last_visted', sa.DATETIME(), nullable=True)) + op.drop_column('URL', 'last_visited') + # ### end Alembic commands ### diff --git a/migrations/versions/36772f67d5f0_.py b/migrations/versions/ed69a26bebbe_.py similarity index 89% rename from migrations/versions/36772f67d5f0_.py rename to migrations/versions/ed69a26bebbe_.py index 68da90b..d707cd5 100644 --- a/migrations/versions/36772f67d5f0_.py +++ b/migrations/versions/ed69a26bebbe_.py @@ -1,6 +1,6 @@ """empty message -Revision ID: 36772f67d5f0 +Revision ID: ed69a26bebbe Revises: Create Date: 2022-01-11 19:29:41.527717 @@ -10,7 +10,7 @@ # revision identifiers, used by Alembic. -revision = '36772f67d5f0' +revision = 'ed69a26bebbe' down_revision = None branch_labels = None depends_on = None @@ -24,6 +24,7 @@ def upgrade(): sa.Column('url', sa.Text(), nullable=False), sa.Column('created', sa.DateTime(), server_default=sa.text('(CURRENT_TIMESTAMP)'), nullable=False), sa.Column('visit_count', sa.Integer(), nullable=False), + sa.Column('last_visited', sa.DateTime(), nullable=True), sa.PrimaryKeyConstraint('id') ) op.create_index(op.f('ix_URL_vanity'), 'URL', ['vanity'], unique=True)