def main():
"""Main function for populating the database"""
donors = [('Toni Morrison', random.sample(range(100, 25000), 3)),
('Mike McHargue', random.sample(range(100, 25000), 3)),
("Flannery O'Connor", random.sample(range(100, 25000), 3)),
('Angelina Davis', random.sample(range(100, 25000), 3)),
('Bell Hooks', random.sample(range(100, 25000), 3))]
db.connect()
db.drop_tables([Donor, Donation])
db.create_tables([Donor, Donation])
bcrypt = Bcrypt()
for donor, donations in donors:
try:
Donor.create(name=donor,
email='.'.join(donor.lower().split()) + '@gmail.com',
password=bcrypt.generate_password_hash('password'),
total=sum(donations),
average=sum(donations) / len(donations))
except:
db.rollback()
for donor, donations in donors:
for donation in donations:
try:
Donation.create(donor=donor, amount=donation)
except:
db.rollback()
def initdb():
"""Initialize the database."""
click.echo('Init the db')
print("remove database")
if os.path.exists(DATABASE):
os.remove(DATABASE)
print("connect database")
db.connect()
print("create tables")
db.create_tables([User, Task, Record])
print("close database")
db.close()
def __init__(self, path="./config.json", time=86400, handle=False):
db.connect()
db.create_tables([App, Version, Config])
if self._first_run():
logging.debug("[+] - First run")
logging.debug("[+] - Loading config")
self.config = self._read_config(path)
logging.debug("[+] - Saving config")
self._save_config()
else:
logging.debug("[+] - Restoring config from database")
self.config = self._get_config().config
self.urls = self.config["apps"]
self.whitelist = self.config["whitelist"]
self.admin = self.config["admin"]
self.updater = Updater(self.config["token"])
if time:
self.time = time
self._automatic_update()
if handle:
dp = self.updater.dispatcher
dp.add_handler(CommandHandler('getid', self.get_chatid))
dp.add_handler(CommandHandler('get_whitelist', self.get_whitelist))
dp.add_handler(
CommandHandler('add_whitelist',
self.add_whitelist,
pass_args=True))
dp.add_handler(
CommandHandler('del_whitelist',
self.del_whitelist,
pass_args=True))
dp.add_handler(CommandHandler('get_apps', self.get_apps))
dp.add_handler(
CommandHandler('add_app', self.add_apps, pass_args=True))
dp.add_handler(
CommandHandler('del_app', self.del_apps, pass_args=True))
dp.add_handler(CommandHandler('help', self.show_help))
dp.add_handler(CommandHandler('update', self.manual_update))
self.updater.start_polling()
self.updater.idle()
else:
self.doit()
def work():
sql_execute(
'ALTER TABLE "topic" ADD COLUMN "update_time" BIGINT NULL DEFAULT NULL ;'
)
sql_execute(
'ALTER TABLE "user" ADD COLUMN "ip_registered" inet NULL DEFAULT NULL;'
)
sql_execute('drop table "notif";')
sql_execute('drop table "mention";')
sql_execute('drop table "user_notif_record";')
db.create_tables([UserNotifLastInfo], safe=True)
for i in UserModel.select().execute():
try:
UserNotifLastInfo.create(id=i.id, update_time=int(time.time()))
except peewee.IntegrityError as e:
print(e)
db.rollback()
from model import db from model.test import Test db.connect() db.create_tables([Test], safe=True)
def db_init():
db.create_tables([User, ApiKey, Article], safe=True)
# coding:utf-8
from model import db
from model.user import User
from model.topic import Topic
from model.board import Board, BoardAdmin
from model.wiki import WikiItem, WikiArticle, WikiHistory
from model.reply import Reply
from model.follow import Follow
def first_run():
if User.count() == 0:
u = User.new('root', '123')
Board.new('初学者', u, '你们啊，还是需要学习一个。')
Board.new('问与答', u, '但是你们问来问去的问题呀，都 too young, too simple.')
Board.new('游戏开发', u, '游戏制作的种种')
Board.new('WEB开发', u, '当下最热门的技术种类')
Board.new('发起项目', u, '已有创意，就差一个程序员/设计师/策划/运营了！')
Board.new('项目展示', u, '今天我作为一个长者，给你们分享一点人生的经验')
Board.new('水区', u, '社区的哪一个牛人我没有见过？我跟他们谈笑风生！')
db.connect()
db.create_tables([User, Topic, Board, BoardAdmin, WikiItem, WikiArticle, WikiHistory, Reply, Follow], safe=True)
first_run()
from model import db, Grade db.connect() db.create_tables([Grade])
from model import db, SavedTotal db.connect() db.create_tables([SavedTotal])
return render_template('register.html', error="User exists")
else:
if Cache().check_password(str(request.form['password'])) == "1":
return render_template('register.html',
error="Your password is too common")
db.execute_sql('insert into users values(NULL, "{}", "{}")'.format(
request.form['login'], request.form['password']))
Cache().validate((request.form['login'], request.form['password']))
return render_template('login.html',
message="Successfully registered")
else:
return render_template('register.html')
@app.route('/search', methods=['GET', 'POST'])
def search():
results = []
city = ""
if request.method == 'POST':
city = request.form['city']
results = [
r for r in Search().find(json.loads('{"city": "' + city + '"}'))
]
return render_template('search.html', forecasts=results, city=city)
if __name__ == '__main__':
db.connect()
db.create_tables([Users, Forecasts])
app.run(host='0.0.0.0', port=5000)
""" Scripts to run to set up our database """ from model import db, Order # Create the database tables for our model db.connect() db.create_tables([Order])
import socketio
from flask import Flask, request, jsonify, render_template, g
from model import db, MDevice
app = Flask("music-pi")
sio = socketio.Server(async_mode="threading")
app.wsgi_app = socketio.WSGIApp(sio, app.wsgi_app)
db.connect()
db.create_tables([MDevice])
connected_devices = {}
def _get_device(name, data={}):
"""
A bit overloaded for convinience
get device by name
if the device never register before
make a new entry in the database
if there are additional attributes passed in
assign these attributes and save it
"""
d, created = MDevice.get_or_create(name=name)
for k, v in data.items():
setattr(d, k, v)
if created or len(data) > 0:
d.save()
return d
for term in terms:
sumGrade += term.termGPA * term.termCredit
sumCredit += term.termCredit
year.yearGPA = float('%.2f' % (sumGrade / sumCredit))
year.yearCredit = sumCredit
year.save()
print("绩点计算完毕")
if __name__ == "__main__":
# 连接数据库，建立数据表
try:
db.connect()
db.create_tables([
Student, ClassSchedule, Class, YearGrade, TermGrade, OneLessonGrade
])
except:
pass
# 查找学生，若不存在则创建账号
try:
student = Student.get(Student.studentnumber == "xxxxxxxx")
except Exception as e:
student = Student(studentnumber="xxxxxxxx",
password="******") # 用自己的教务系统账号密码
student.save()
spider = ZhengFangSpider(student,
baseUrl="http://202.195.144.168/jndx") # 实例化爬虫
spider.loginWithOutCode()
from base import Gopay
from random import choice
from model import db, Transaction
email = input("Your gopay email address : ")
gopay = Gopay(email)
key = ""
seed = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789-"
for i in range(30):
key += choice(seed)
token = gopay.login()
with open("config.py", "w") as f:
to_be_written = 'TOKEN = "%s"\n' % token
to_be_written += 'API_KEY = "%s"\n' % key
f.write(to_be_written)
db.connect()
db.create_tables([Transaction])
db.close()
# coding:utf-8 from model import db from model.user import User,Board db.connect() db.create_tables([User, Board], safe=True)
# coding:utf-8
from model import db
from model.user import User
import model.game
db.connect()
try:
db.create_tables([User])
except:
pass
import os from model import db, Charades from playhouse.db_url import connect db.connect() db.create_tables([Charades])
''' Initialize database, such as tables, data '''
import peewee
from model import db
from model.account import Account
from model.account_ask import AccountAsk
try:
db.create_tables([Account, AccountAsk])
except peewee.OperationalError as event:
print('OperationalError', event)
pass
except peewee.ProgrammingError as event:
# 错误: 关系 "bloglabels" 已经存在
pass
def init_model():
tables = [Show, Season, Episode]
# db.drop_tables(tables)
db.create_tables(tables, True)
"""
Scripts to run to set up our database
"""
from datetime import datetime
from model import db, User, Task
from passlib.hash import pbkdf2_sha256
# Create the database tables for our model
db.connect()
db.drop_tables([User, Task])
db.create_tables([User, Task])
Task(name="Do the laundry.").save()
Task(name="Do the dishes.", performed=datetime.now()).save()
User(name="admin", password=pbkdf2_sha256.hash("password")).save()
User(name="bob", password=pbkdf2_sha256.hash("bobbob")).save()
from model import db, Participant, Prize_Structure, Team
db.connect()
# This line will allow you "upgrade" an existing database by
# dropping all existing tables from it.
db.drop_tables([Participant, Prize_Structure, Team])
db.create_tables([Participant, Prize_Structure, Team])
play_in = Prize_Structure(advanced_to_round=0, money_share=0).save()
first = Prize_Structure(advanced_to_round=1, money_share=0).save()
second = Prize_Structure(advanced_to_round=2, money_share=0).save()
sweet_sixteen = Prize_Structure(advanced_to_round=3, money_share=0.03).save()
elite_eight = Prize_Structure(advanced_to_round=4, money_share=0.05).save()
final_four = Prize_Structure(advanced_to_round=5, money_share=0.1).save()
championship = Prize_Structure(advanced_to_round=6, money_share=0.16).save()
winner = Prize_Structure(advanced_to_round=7, money_share=0.2).save()
teams = [('Villanova', 'East', 1, first), ('Michigan', 'West', 3, first),
('Kansas', 'Midwest', 1, first), ('Loyola (IL)', 'South', 11, first),
('Kansas State', 'South', 9, first),
('Florida State', 'West', 9, first), ('Duke', 'Midwest', 2, first),
('Texas Tech', 'East', 3, first), ('Kentucky', 'South', 5, first),
('Nevada', 'South', 7, first), ('Gonzaga', 'West', 4, first),
('Texas A&M', 'West', 7, first), ('Clemson', 'Midwest', 5, first),
('Syracuse', 'Midwest', 11, play_in), ('W. Virginia', 'East', 5,
first),
('Purdue', 'East', 2, first), ('Marshall', 'East', 13, first),
('UMBC', 'South', 16, first), ('Buffalo', 'South', 13, first),
('Tennessee', 'South', 3, first), ('Cincinnati', 'South', 2, first),
sql_execute('alter table "user" drop column key')
sql_execute('alter table "user" drop column key_time')
sql_execute('alter table "upload" add column filename text')
sql_execute('alter table "upload" add column source text')
db.connect()
work()
db.create_tables([Test, BoardModel, Follow, CommentModel, TopicModel, UserModel,
WikiArticleModel,
NotificationModel, UserNotifLastInfo,
UserOAuth,
UserToken,
UploadModel,
ManageLogModel,
Mention,
PostStatsModel,
StatsLog], safe=True)
work()
sql_execute("""
ALTER TABLE "board" ALTER COLUMN "id" SET DEFAULT int2bytea(nextval('id_gen_seq')::bigint);
ALTER TABLE "topic" ALTER COLUMN "id" SET DEFAULT int2bytea(nextval('id_gen_seq')::bigint);
ALTER TABLE "user" ALTER COLUMN "id" SET DEFAULT int2bytea(nextval('id_gen_seq')::bigint);
ALTER TABLE "user" ALTER COLUMN "number" SET DEFAULT nextval('user_count_seq')::bigint;
""")
WikiArticleModel.get_sidebar_article()
from model import db, SavedDonation db.connect() db.create_tables([SavedDonation])
def init_db():
"""Initializes the database."""
db = get_db()
db.create_tables([User, Task])
db.save()
# coding:utf-8 from model import db from model.user import User from model.todo import Todo db.connect() db.create_tables([User, Todo], safe=True)
def create_tables():
db.create_tables([Person,Consult,ConsultExt])
import random
from model import db, Donor, Donation, User
from passlib.hash import pbkdf2_sha256
db.connect()
# This line will allow you "upgrade" an existing database by
# dropping all existing tables from it.
db.drop_tables([Donor, Donation, User])
db.create_tables([Donor, Donation, User])
alice = Donor(name="Alice")
alice.save()
bob = Donor(name="Bob")
bob.save()
charlie = Donor(name="Charlie")
charlie.save()
donors = [alice, bob, charlie]
for x in range(30):
Donation(donor=random.choice(donors), value=random.randint(100,
10000)).save()
User(name='admin', password=pbkdf2_sha256.hash("password")).save()
from model import db, Todo,Vocab,Simple
import config
# db.connect()
if(not db.get_tables()):
db.create_tables([Todo, Vocab, Simple])
# new = Todo.create(id=2,content="selam naber",tags="Useless,")
# newV = Vocab.create(id=1,content="selam: naber", tags="Useless,")
# newC = Content.create(id=1,content="Selam naber.", tags="Useless,")
# todo Wrapper class
class todoW:
def select_all(): # todo select all
# return [x for x in Todo.select()]
return Todo.select()
def add(content, tags):
a = Todo.create(content=content, tags = tags) # this is an insert query
def update_is_done(id):
a = Todo.update(is_done=True).where(Todo.id == id)
return a.execute()
def delete_by_id(id):
a = Todo.select().where(Todo.id==id)
a.delete_instance()
def add_tag_by_id(id, _tags):
from model import db, Message db.connect() db.create_tables([Message])
# coding:utf-8 from model import db from model.test import Test from model.user import User from model.chat_log import ChatLog from model.file_log import FileLog from model.flag_log import FlagLog db.connect() db.create_tables([Test, User, ChatLog, FileLog, FlagLog], safe=True)
$$ LANGUAGE plpgsql;
CREATE SEQUENCE IF NOT EXISTS id_gen_seq NO MINVALUE NO MAXVALUE START 4096 NO CYCLE; /* 0x1000 */
CREATE SEQUENCE IF NOT EXISTS user_count_seq NO MINVALUE NO MAXVALUE START 1 NO CYCLE;
""")
db.execute_sql("""
CREATE EXTENSION IF NOT EXISTS hstore;
CREATE EXTENSION IF NOT EXISTS citext;
""")
except peewee.ProgrammingError:
# permission denied to create extension "hstore"
db.rollback()
db.create_tables([
Test, Board, Follow, Comment, Topic, User, WikiArticle, Notification,
UserNotifLastInfo, UserOAuth, Upload, ManageLog, Mention, Statistic,
Statistic24h
],
safe=True)
try:
db.execute_sql("""
ALTER TABLE "board" ALTER COLUMN "id" SET DEFAULT int2bytea(nextval('id_gen_seq')::bigint);
ALTER TABLE "topic" ALTER COLUMN "id" SET DEFAULT int2bytea(nextval('id_gen_seq')::bigint);
ALTER TABLE "user" ALTER COLUMN "id" SET DEFAULT int2bytea(nextval('id_gen_seq')::bigint);
ALTER TABLE "user" ALTER COLUMN "number" SET DEFAULT nextval('user_count_seq')::bigint;
""")
except peewee.ProgrammingError:
db.rollback()
WikiArticle.get_sidebar_root_article()
import random
from peewee import DoesNotExist
from passlib.hash import pbkdf2_sha256
from model import db, Donor, Donation, Admin
db.connect()
# db.execute_sql('PRAGMA foreign_keys = ON;')
# This line will allow you "upgrade" an existing database by
# dropping all existing tables from it.
db.drop_tables([Donor, Donation, Admin])
#
db.create_tables([Donor, Donation, Admin])
alice = Donor.create(name="Alice")
# alice.save()
bob = Donor.create(name="Bob")
# bob.save()
charlie = Donor.create(name="Charlie")
# charlie.save()
donors = [alice, bob, charlie]
for x in range(30):
Donation.create(donor=random.choice(donors),
value=random.randint(100, 10000))
user1 = Admin.create(user="******", password=pbkdf2_sha256.hash("password"))
user2 = Admin.create(user="******", password=pbkdf2_sha256.hash("123456"))
END;
$$ LANGUAGE plpgsql;
CREATE SEQUENCE IF NOT EXISTS id_gen_seq NO MINVALUE NO MAXVALUE START 4096 NO CYCLE; /* 0x1000 */
CREATE SEQUENCE IF NOT EXISTS user_count_seq NO MINVALUE NO MAXVALUE START 1 NO CYCLE;
""")
db.execute_sql("""
CREATE EXTENSION IF NOT EXISTS hstore;
CREATE EXTENSION IF NOT EXISTS citext;
""")
except peewee.ProgrammingError:
# permission denied to create extension "hstore"
db.rollback()
db.create_tables([
Test, Board, Follow, Comment, Topic, User, WikiItem, WikiArticle,
WikiHistory, Notification, UserNotifRecord, UserOAuth, Upload, ManageLog,
Statistic, Statistic24h
],
safe=True)
try:
db.execute_sql("""
ALTER TABLE "board" ALTER COLUMN "id" SET DEFAULT int2bytea(nextval('id_gen_seq')::bigint);
ALTER TABLE "topic" ALTER COLUMN "id" SET DEFAULT int2bytea(nextval('id_gen_seq')::bigint);
ALTER TABLE "user" ALTER COLUMN "id" SET DEFAULT int2bytea(nextval('id_gen_seq')::bigint);
""")
except peewee.ProgrammingError:
db.rollback()
# coding:utf-8 from model import db from model.test import Test from model.user import User db.connect() db.create_tables([Test, User], safe=True)
import random
from model import db, Donor, Donation, User
from passlib.hash import pbkdf2_sha256
db.connect()
# This line will allow you "upgrade" an existing database by
# dropping all existing tables from it.
db.drop_tables([User, Donor, Donation])
db.create_tables([User, Donor, Donation])
alice = Donor(name="Alice")
alice.save()
bob = Donor(name="Bob")
bob.save()
charlie = Donor(name="Charlie")
charlie.save()
donors = [alice, bob, charlie]
for x in range(30):
Donation(donor=random.choice(donors), value=random.randint(100,
10000)).save()
User(name="admin", password=pbkdf2_sha256.hash("password")).save()
User(name="bob", password=pbkdf2_sha256.hash("bobbob")).save()
# coding:utf-8 from model import db from model.article import Article from model.comment import Comment from model.user import User from model.board import Board db.connect() db.create_tables([User, Article, Board, Comment], safe=True)