コード例 #1
0
ファイル: database_helper.py プロジェクト: TriFaceDude/TDDD24
def set_user_signed_in(email, token):
	db = connect_db()
	cur = db.cursor()
	cur.execute("CREATE TABLE IF NOT EXISTS online_users(user_email TEXT, user_token TEXT)")
	cur.execute("INSERT INTO online_users (user_email, user_token) VALUES(?,?)", (email, token))
	db.commit()
	db.close()
コード例 #2
0
ファイル: database_helper.py プロジェクト: TriFaceDude/TDDD24
def set_user_password(email, new_password):
	
	db = connect_db()
	cur = db.cursor()
	cur.execute("UPDATE users SET user_pass=? WHERE user_email=?", (new_password,email))
	db.commit()
	db.close()
コード例 #3
0
ファイル: database_helper.py プロジェクト: TriFaceDude/TDDD24
def create_user(email, password, firstname, lastname, gender, city, country):

	user = (email, password, firstname, lastname, gender, city, country)
	db = connect_db()
	cur = db.cursor()
	cur.execute("INSERT INTO users VALUES(?, ?, ?, ?, ?, ?, ?)", user)
	db.commit()
	db.close()
コード例 #4
0
ファイル: database_helper.py プロジェクト: TriFaceDude/TDDD24
def get_users():
	db = connect_db()
	cur = db.execute('SELECT * FROM users')
	users = {}
	
	for row in cur.fetchall():
		users[row[0]] = {'email' : row[0], 'password' : row[1], 'firstname' : row[2], 'lastname' : row[3], 'gender' : row[4], 'city' : row[5], 'country' : row[6]}
	db.close()
	
	return users
コード例 #5
0
ファイル: database_helper.py プロジェクト: TriFaceDude/TDDD24
def retrive_messages(email):

	db = connect_db()	
	cur = db.cursor()
	cur.execute("SELECT msg_index, sender_email, msg_content FROM user_messages WHERE receiver_email=?", (email,))
	
	messages = []
	for row in cur.fetchall():
		messages.append({'msg_id' : row[0], 'sender_email' : row[1], 'msg_content' : row[2]})
	
	return messages	
コード例 #6
0
ファイル: database_helper.py プロジェクト: TriFaceDude/TDDD24
def write_message(token, email, msg):

	db = connect_db()	
	cur = db.cursor()	
	cur.execute("CREATE TABLE IF NOT EXISTS user_messages (receiver_email TEXT, msg_index INTEGER, sender_email TEXT, msg_content TEXT)")
	
	cur.execute("SELECT COUNT(*) FROM user_messages WHERE receiver_email=?", (email,))
	cur.execute("INSERT INTO user_messages VALUES(?, ?, ?, ?)", (email, msg_index, token_to_email(token), msg))
	db.commit()
	cur.close()
	
	return online_users
コード例 #7
0
ファイル: database_helper.py プロジェクト: TriFaceDude/TDDD24
def get_online_users():
	db = connect_db()
	cur = db.execute("SELECT name FROM sqlite_master WHERE type='table' AND tbl_name='online_users'")

	online_users = {}

	if cur.fetchall() != []:
		cur = db.execute("SELECT user_email, user_token FROM online_users")
		for row in cur.fetchall():
			online_users[row[1]] = row[0]

			db.close()

	return online_users
コード例 #8
0
ファイル: database_helper.py プロジェクト: TriFaceDude/TDDD24
def set_user_offline(token):
	db = connect_db()
	db.execute('DELETE FROM online_users WHERE user_token=?', (token,))
	db.commit()
	db.close()
コード例 #9
0
from server import connect_db

tuples = [
    ("test", "alpha"),
    ("test", "beta"),
    ("test", "gamma"),
    ("test", "theta"),
    ("test", "epsilon"),
]

if __name__ == '__main__':
    with open('schema.sql') as fp:
        setup = fp.read()
    with connect_db() as db:
        db.executescript(setup)
        db.executemany(
            """insert into submissions(group_id, content) values (?,?)""",
            tuples)
コード例 #10
0
import sys
from lxml.etree import tostring
from itertools import chain
from flask import Flask
from operator import itemgetter
from flask import render_template, json, jsonify, g, request, send_from_directory
from flask.json import JSONEncoder
import psycopg2
import psycopg2.extras
import datetime
import os
import elasticsearch
import re
import os
from server import connect_db, get_act_exact, tohtml


def requires_contents(act):
	return len(act.xpath('.//toc'))

def create_contents(act):
	if requires_contents(act):
		print etree.tostring(tohtml(act, 'contents.xslt'), encoding='UTF-8', pretty_print=True)


if __name__ == "__main__":
	db = connect_db()
	create_contents(get_act_exact('Companies Act 1993', db=db))


コード例 #11
0
ファイル: init_database.py プロジェクト: halv00rsen/Scorez3.0
def init():
	if __name__ != "__main__":
		return "ERROR: Must start from terminal."
	print("Initializes the server config and database.")
	reset = False
	conf = config_file_ok()
	if conf:
		inp = input("Config file looks ok. Do you want to reset it (y/n)?")
		reset = inp == "y"
		if reset:
			print("Will create a new config file.")
	if reset or not conf:
		# db = input("Database name: ")
		# if not db:
		# 	print("No database name. Aborting...")
		# 	return None
		secret_key = input("Super secret key: ")
		if not secret_key:
			print("Need a secret key. Aborting...")
			return
		with open("server_config.py", "w") as f:
			f.write("from os import path\n")
			f.write("DEBUG = True\n")
			f.write("SECRET_KEY = '{}'\n".format(secret_key))
			f.write("DATABASE = path.join('db', 'database.db')")

	if path.isdir("db") and path.exists(path.join("db", "database.db")):
		print("A database already exists.")
		print("1. Reset current database.")
		print("2. Transfer data to new database.")
		print("3. Keep old database.")
		inp = input("Action (1,2,3): ")
		# inp = input("A database already exists. Do you want to reset current database? (y/n) ")
		# if inp != "y":
		# 	print("Database was not reseted.")
		# 	return None
		if inp == "2":
			pass
		elif inp != "1":
			print("Old database keeped.")
			return None
	print("Will initialize a new database.")
	print("You will need an admin account.")
	username = input("Username: "******"Password: "******"Retype password: "******"Error! Username or password is empty. Aborting...")
	else:
		if not path.isdir("db"):
			makedirs("db")
		with open(path.join("db", "database.db"), "w") as f:
			pass
		from server import init_db, hash_password, connect_db
		init_db()
		db = connect_db()
		try:
			r = db.execute("insert into User values (?,?,?)", [username, hash_password(password), True])
			# db.execute("insert into Beer_type values (?)", ["Ingen"])
			db.commit()
			print("Database was successfully created.")
		except Exception as e:
			print("Database not created, error.")
			print(e)
			pass
		finally:
			if db:
				db.close()