def game_count(self):
s = Session()
gameCount = s.query(Champion).\
join(Match, Champion.match_id == Match.match_id).\
filter(Champion.champion_key == self.championKey).\
filter(Match.created_on > (
datetime.utcnow() - timedelta(days=self.days))).count()
return gameCount
def evaluate(model, dictionary, session_queries):
sess = Session()
sess.queries = session_queries
session_tensor, query_lengths = helper.session_to_tensor([sess],
dictionary, True)
if args.cuda:
session_tensor = session_tensor.cuda()
query_lengths = query_lengths.cuda()
return suggest_next_query(model, dictionary, session_tensor, query_lengths)
def evaluate(model, dictionary, session_queries):
session = Session()
session.queries = session_queries
session_queries, session_query_length, rel_docs, rel_docs_length, doc_labels = helper.session_to_tensor(
[session], dictionary, iseval=True)
if model.config.cuda:
session_queries = session_queries.cuda()
session_query_length = session_query_length.cuda()
return suggest_next_query(model, session_queries, session_query_length, dictionary)
def insert_tags(all_tags):
s = Session()
for tag in all_tags:
t = Tags(name=tag)
try:
s.add(t)
s.commit()
print "Added:", tag
except IntegrityError as e:
s.rollback()
pass
def main(filename):
Base.metadata.create_all(Engine)
session = Session()
fortunes = pd.read_csv(filename)
for index, row in fortunes.iterrows():
logger.info('Loading rank {} into DB'.format(row['rank']))
fortune = Fortune(row['rank'],
row['title'],
row['website'],
row['employees'],
row['sector'],
row['industry'])
session.add(fortune)
session.commit()
session.close()
def parse_lines(gen):
s = Session()
for index, line in enumerate(gen):
line = line.strip()
if line.startswith("<row"):
node = etree.fromstring(line)
post_object = node.attrib
if post_object.get('PostTypeId') == "1":
process_questions(s, post_object)
if post_object.get('PostTypeId') == "2":
process_answers(s, post_object)
if index % 2000 == 0:
print index, "processed."
sys.stdout.flush()
if index % 50000 == 0:
s.commit()
s.commit()
def attach_foreignkeys(gen):
s = Session()
all_answers = {x[0] for x in s.query(Answers.id).all()}
print "Answers loaded."
for index, line in enumerate(gen):
line = line.strip()
if line.startswith("<row"):
node = etree.fromstring(line)
post_object = node.attrib
if post_object.get('PostTypeId') == "1":
attach_question(s, post_object, all_answers)
if post_object.get('PostTypeId') == "2":
attach_answer(s, post_object)
if index % 2000 == 0:
print index, "attached."
sys.stdout.flush()
if index % 50000 == 0:
s.commit()
s.commit()
from data import Session, User, Status, Role, Post, Activity
from functools import wraps
from flask import Flask, render_template, Response, request, redirect, send_from_directory, url_for, Response
from flask_wtf import FlaskForm
from wtforms import StringField, PasswordField, SubmitField
from flask_login import LoginManager, login_required, login_user, current_user, logout_user
import werkzeug
app = Flask(__name__)
app.config["SECRET_KEY"] = "kek"
session = Session(autocommit=True, autoflush=True)
app.app_context()
login_manager = LoginManager()
login_manager.init_app(app)
class LoginForm(FlaskForm):
nickname = StringField('nickname')
password = PasswordField('password')
submit = SubmitField('submit')
@login_manager.user_loader
def load_user(id):
return session.query(User).filter(User.id == id).first()
@app.route("/logout")
def logout():
logout_user()
return redirect("/index")
with open(sys.argv[2], "r") as f:
for line in f:
line = line.strip()
if line.startswith("<row"):
node = etree.fromstring(line)
post_object = node.attrib
if post_object.get('PostTypeId') == "1":
tags = re.findall("\<(.*?)\>", post_object.get('Tags'))
for tag in tags:
s = "INSERT INTO poststags(PostsId, TagsId) VALUES (" + \
post_object.get('Id') + "," + str(tag_dict[tag]) +");"
print >> g, s
def tags_dict(path):
tags = {}
with open(path, "r") as f:
for line in f:
val= tuple(line.split())
if val[1] not in tags:
tags[val[1]]= int(val[0])
return tags
if __name__=="__main__":
s = Session()
# with open(sys.argv[1], "w") as f:
# get_tags(s, f)
tags = tags_dict(sys.argv[1])
with open(sys.argv[3], "w") as g:
update_poststags(tags, g)
