def register():
if request.method == 'POST':
username = request.form['username']
password = request.form['password']
#db = get_db()
error = None
if not username:
error = 'Username is required.'
elif not password:
error = 'Password is required.'
#elif db.execute(
# 'SELECT id FROM user WHERE username = ?', (username,)
#).fetchone() is not None:
elif User.query.filter(User.name == username).first() is not None:
error = 'User {} is already registered.'.format(username)
if error is None:
#db.execute(
# 'INSERT INTO user (username, password) VALUES (?, ?)',
# (username, generate_password_hash(password))
#)
#db.commit()
u = User(username, password)
db_session.add(u)
db_session.commit()
return redirect(url_for('auth.login'))
flash(error)
return render_template('auth/register.html')
def register():
"""Register a new user.
Validates that the username is not already taken. Hashes the
password for security.
"""
if request.method == 'POST':
username = request.form['username']
password = request.form['password']
error = None
if not username:
error = 'Username is required.'
elif not password:
error = 'Password is required.'
elif User.query.filter(User.username == username).first() is not None:
error = 'User {0} is already registered.'.format(username)
if error is None:
# the name is available, store it in the database and go to
# the login page
u = User(username, generate_password_hash(password))
db_session.add(u)
db_session.commit()
return redirect(url_for('auth.login'))
flash(error)
return render_template('auth/register.html')
def create():
if request.method == 'POST':
title = request.form['title']
body = request.form['body']
error = None
if not title:
error = 'Title is required.'
if error is not None:
flash(error)
else:
# mysql
post = Post(title, body, g.user.id)
db_session.add(post)
db_session.commit()
# sqlite3
# db = get_db()
# db.execute(
# 'INSERT INTO post (title, body, author_id)'
# ' VALUES (?, ?, ?)',
# (title, body, g.user['id'])
# )
# db.commit()
return redirect(url_for('blog.index'))
return render_template('blog/create.html')
def add_entry():
if not session.get('logged_in'):
abort(401)
entry = Entry(request.form['title'], request.form['text'], request.form['pw'])
db_session.add(entry)
db_session.commit()
flash('New entry was successfully posted')
return redirect(url_for('show_list'))
def add_user(cls, password, username, email):
"""Add user to the database"""
hashed_password = generate_password_hash(password, method='sha256')
new_user = User(username=username,
email=email,
password=hashed_password)
db_session.add(new_user)
db_session.commit()
def new():
form = RegistrationForm(request.form)
if request.method == 'POST' and form.validate():
u = User()
u.firstName = request.form['firstName']
u.lastName = request.form['lastName']
u.email = request.form['email']
db_session.add(u)
db_session.commit()
flash('New user created.')
return redirect(url_for('index'))
else:
return render_template('new.html', form=form)
def signup():
form = RegisterForm()
if form.validate_on_submit():
print('test')
hashed_password = generate_password_hash(
form.password.data, method='sha256')
new_user = User(username=form.username.data,
email=form.email.data, password=hashed_password)
db_session.add(new_user)
db_session.commit()
return redirect(url_for('login', error="Signup successfully"))
return render_template('signup.html', form=form)
def insert_project_info(project_id, json_data, project_info):
project_info.project_id = project_id # 查询出来
project_info.total_quantity = json_data['total_quantity'] # 数据总量
project_info.crawler_quantity = json_data['crawler_quantity'] # 抓取总量
project_info.result_quantity = json_data['result_quantity'] # 入库总量
project_info.crawler_start_time = datetime.datetime.now() # 开始抓取时间
project_info.crawler_end_time = datetime.datetime.now() # 结束时间
project_info.author = json_data['author'] # 作者
project_info.partner = json_data['partner'] # 协助者
db_session.add(project_info)
db_session.commit()
def upload():
"""Upload a document."""
if request.method == 'POST' and 'doc' in request.files:
try:
filename = docs.save(request.files['doc'])
filename = secure_filename(filename)
print(filename)
url = docs.url(filename)
user = User.query.filter_by(id=current_user.id).first()
user.file_name = filename
user.file_url = url
db_session.add(user)
db_session.commit()
flash("Doc saved.", "success")
except UploadNotAllowed:
flash("The format is not correct.", "danger")
return render_template('upload.html')
def add_opinion(tweet_id, flag, opinion_value):
"""Add user's opinion to rumor."""
print("tweet_id", tweet_id)
print("opinion", opinion_value)
print("user_id", current_user.id)
rumor = Rumor.query.filter_by(tweet_id=tweet_id).first()
rumor_id = rumor.id
# check if the record existance
existed = db_session.query(exists().where(and_(
Opinion.flag == flag,
Opinion.stance == opinion_value,
Opinion.rumor_id == rumor_id,
Opinion.user_name == current_user.username
))).scalar()
if existed:
# record existed
print("record existed")
print("Opinion", Opinion.query.all())
r = make_response('existed', 200)
return r
else:
# check if record updation
print("checking update")
q = db_session.query(Opinion).filter(and_(
Opinion.flag == flag,
Opinion.rumor_id == rumor_id,
Opinion.user_name == current_user.username
))
if q.all():
# update record
print("updating record...")
q.update({"stance": opinion_value})
db_session.commit()
r = make_response('updated', 200)
return r
# insert new record
print("insert record...")
opinion = Opinion(flag=flag, stance=opinion_value)
opinion.rumor = rumor
current_user.rumors.append(opinion)
db_session.add(current_user)
db_session.commit()
print("Opinion", Opinion.query.all())
r = make_response('inserted', 200)
return r
def register():
""" user register """
# 1. check username and password
if request.method == 'POST':
username = request.form['username']
password = request.form['password']
db = get_db()
error = None
# check if username is empty
if not username:
error = 'Username is required!'
# check if password is empty
elif not password:
error = 'Password is required!'
# check if username exists in database
# mysql 数据库
elif User.query.filter(User.username == username).first() is not None:
error = f'User {username} is registered!'
# sqllite3 数据库
# elif db.execute(
# 'SELECT id FROM user WHERE username = ?', (username, )
# ).fetchone() is not None:
# error = f'User {username} is registered!'
# everthing is allright, reigster user into database
if error is None:
db_session.add(User(username, generate_password_hash(password)))
db_session.commit()
# db.execute(
# 'INSERT INTO user (username, password) VALUES (?, ?)',
# (username, generate_password_hash(password))
# )
# db.commit()
return redirect(url_for('auth.login'))
flash(error)
return render_template('auth/register.html')
def insert_project():
"""
插入一条数据
{
"project_name": "ad",
"total_quantity" : 1,
"crawler_quantity":1,
"result_quantity":1,
"crawler_start_time":1,
"crawler_end_time":1,
"author":"小武神",
"partner":"锦鲤"
}
:return:
"""
json_data = request.get_json()
project = db_session.query(MonitorProject).filter(
MonitorProject.project_name == json_data['project_name']).first()
print("project={}".format(project))
if not project:
# 数据库不存在project id
m = MonitorProject()
m.project_name = json_data['project_name']
db_session.add(m)
db_session.flush()
db_session.commit()
project_info = MonitorProjectInfo()
insert_project_info(m.id, json_data, project_info)
print(json_data)
else:
# 已经存在，跟新其他信息
project_info = MonitorProjectInfo()
insert_project_info(project.id, json_data, project_info)
print(project.id)
return jsonify({'success': 'true'})
def add_comment():
comments = Comment(request.args['id'], request.form['author'], request.form['comment'], request.form['addCommentPassword'])
db_session.add(comments)
db_session.commit()
flash('New comment was successfully posted')
return redirect(url_for('show_entry', id=request.args['id']))
def add_data(self, eventName):
"""Initialize the data in the database."""
# load events
print("adding {} data...".format(eventName))
eventPath = self.dataRootPath / eventName
terms = [
term for term in eventPath.name.split('_') if not term.isdigit()
]
edited_event = " ".join(terms)
tableEvent = CreateDatabaseTable.create_event(edited_event)
origin_statement_id = edited_event + "_" + "origin_statement"
origin_statement = Helper.getEventClaims(eventPath)
tableOriginStatement = CreateDatabaseTable.create_origin_statement(
origin_statement_id, origin_statement)
tableEvent.origin_statement = tableOriginStatement
clustersPath = eventPath / 'clusterData'
clusters = [
cluster for cluster in clustersPath.iterdir() if cluster.is_dir()
]
for cluster in clusters:
clusterTable = CreateDatabaseTable.create_cluster(cluster)
tableEvent.clusters.append(clusterTable)
event_cluster_id = edited_event + "_" + cluster.name
# tableEvent_Cluster = CreateDatabaseTable.combine_event_cluster(
# event_cluster_id, tableEvent, clusterTable)
# get statements
statementClusters = [
statementCluster
for statementCluster in (cluster / "classification").iterdir()
if statementCluster.is_dir()
]
for statementCluster in statementClusters:
# load data
statement = Helper.getRepresentativeClaim(statementCluster)
rumors = Helper.getClusterClaims(statementCluster)
# snippets = Helper.getNews(cluster, statementCluster.name)
snippets = Helper.getSnippets(statementCluster)
# check all data
if not (statement and rumors and snippets):
continue
print("successfully load cluster data ", statementCluster)
# print("statement ", statement)
statement_id = event_cluster_id + "_" + statementCluster.name
tableStatement = CreateDatabaseTable.create_statement(
statement_id, statement)
if not tableStatement:
continue
# continue
# tableRumor = CreateDatabaseTable.create_rumor(rumor, rumor_id)
# tableEvent_Cluster.rumors.append(tableRumor)
# get rumors
for indexRumor, rumor in enumerate(rumors):
rumor_id = statement_id + "_" + str(indexRumor)
tableRumor = CreateDatabaseTable.create_rumor(
rumor, rumor_id)
if not tableRumor:
continue
# tableEvent_Cluster.rumors.append(tableRumor)
tableStatement.rumors.append(tableRumor)
# # news
# for indexSnippet, snippet in enumerate(snippets):
# snippet_id = statement_id + "_" + str(indexSnippet)
# tableSnippet = CreateDatabaseTable.create_snippet(
# snippet_id, snippet)
# if not tableSnippet:
# continue
# tableStatement.snippets.append(tableSnippet)
# snippets
for indexSnippet, snippet in enumerate(snippets):
snippet_id = statement_id + "_" + str(indexSnippet)
tableSnippet = CreateDatabaseTable.create_snippet(
snippet_id, snippet)
if not tableSnippet:
continue
tableStatement.snippets.append(tableSnippet)
# tableEvent_Cluster.statements.append(tableStatement)
tableEvent.cluster_associations[-1].statements.append(
tableStatement)
db_session.add(clusterTable)
# print("commit")
db_session.commit()