diff --git a/app/main/views.py b/app/main/views.py index 4e2d07c99cdf91a123559f149ab80738f5543479..990f60d5fd6ee37e8a40638dff7fd836e3337ef4 100644 --- a/app/main/views.py +++ b/app/main/views.py @@ -7,7 +7,6 @@ from . import main from .forms import CreateCorpusForm from ..decorators import admin_required from .. import db -import hashlib import os @@ -28,14 +27,11 @@ def dashboard(): if create_corpus_form.validate_on_submit(): app = current_app._get_current_object() - corpus = {'description': create_corpus_form.description.data, - 'id': id, - 'creator': current_user.id, - 'title': create_corpus_form.title.data - } - corpus = Corpus(description=create_corpus_form.description.data, - title=create_corpus_form.title.data, - user_id=current_user.id) + corpus = Corpus() + corpus.creation_date = datetime.utcnow() + corpus.description = create_corpus_form.description.data, + corpus.title = create_corpus_form.title.data, + corpus.user_id = current_user.id db.session.add(corpus) db.session.commit() dir = os.path.join(app.config['OPAQUE_STORAGE'], diff --git a/app/models.py b/app/models.py index 4a6c423abbf7aaac2005ff7b7cab18d3e145f8cf..d17eb0fde7008b2a8dfde95059d17eb77a5a3201 100644 --- a/app/models.py +++ b/app/models.py @@ -107,9 +107,9 @@ class User(UserMixin, db.Model): # Primary key id = db.Column(db.Integer, primary_key=True) confirmed = db.Column(db.Boolean, default=False) - registration_date = db.Column(db.DateTime(), default=datetime.utcnow) email = db.Column(db.String(64), unique=True, index=True) password_hash = db.Column(db.String(128)) + registration_date = db.Column(db.DateTime(), default=datetime.utcnow) role_id = db.Column(db.Integer, db.ForeignKey('roles.id')) username = db.Column(db.String(64), unique=True, index=True) # Relationships @@ -259,6 +259,7 @@ class Corpus(db.Model): __tablename__ = 'corpora' # Primary key id = db.Column(db.Integer, primary_key=True) + creation_data = db.Column(db.DateTime(), default=datetime.utcnow) description = db.Column(db.String(64)) title = db.Column(db.String(32)) user_id = db.Column(db.Integer, db.ForeignKey('users.id')) diff --git a/migrations/versions/c425353fb4dc_add_creation_data_to_corpus.py b/migrations/versions/c425353fb4dc_add_creation_data_to_corpus.py new file mode 100644 index 0000000000000000000000000000000000000000..d580a9bab7b302b273ce23ba7b13ad82abae593f --- /dev/null +++ b/migrations/versions/c425353fb4dc_add_creation_data_to_corpus.py @@ -0,0 +1,28 @@ +"""Add creation_data to corpus + +Revision ID: c425353fb4dc +Revises: 3f2aee5e6a45 +Create Date: 2019-08-06 15:38:33.930052 + +""" +from alembic import op +import sqlalchemy as sa + + +# revision identifiers, used by Alembic. +revision = 'c425353fb4dc' +down_revision = '3f2aee5e6a45' +branch_labels = None +depends_on = None + + +def upgrade(): + # ### commands auto generated by Alembic - please adjust! ### + op.add_column('corpora', sa.Column('creation_data', sa.DateTime(), nullable=True)) + # ### end Alembic commands ### + + +def downgrade(): + # ### commands auto generated by Alembic - please adjust! ### + op.drop_column('corpora', 'creation_data') + # ### end Alembic commands ###