Skip to content
Snippets Groups Projects
Commit 5a67164d authored by Patrick Jentsch's avatar Patrick Jentsch
Browse files

Add corpus to database after creation.

parent 4ec0acc0
No related branches found
No related tags found
No related merge requests found
from datetime import datetime
from flask import current_app, flash, redirect, render_template, url_for
from flask_login import current_user, login_required
from ..models import User
from ..models import User, Corpus
from ..tables import AdminUserTable, AdminUserItem
from . import main
from .forms import CreateCorpusForm
from ..decorators import admin_required
from .. import db
import hashlib
import os
......@@ -27,15 +28,17 @@ def dashboard():
if create_corpus_form.validate_on_submit():
app = current_app._get_current_object()
id = hashlib.md5(
(current_user.username + '_' + datetime.now().isoformat()).encode()
).hexdigest()
corpus = {'description': create_corpus_form.description.data,
'id': id,
'creator': current_user.id,
'title': create_corpus_form.title.data
}
dir = os.path.join(app.config['OPAQUE_FILES'], 'corpora', id)
corpus = Corpus(description=create_corpus_form.description.data,
title=create_corpus_form.title.data,
user_id=current_user.id)
db.session.add(corpus)
db.session.commit()
dir = os.path.join(app.config['OPAQUE_STORAGE'], 'corpora', str(corpus.id))
try:
os.makedirs(dir)
......
......@@ -260,7 +260,7 @@ class Corpus(db.Model):
user_id = db.Column(db.Integer, db.ForeignKey('users.id'))
def __init__(self, **kwargs):
super(Job, self).__init__(**kwargs)
super(Corpus, self).__init__(**kwargs)
def __repr__(self):
"""
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment