__tablename__ = 'websiteoffline'
__bind_key__ = 'clearnet'
__table_args__ = {"schema": "public", 'extend_existing': True}
id = db.Column(db.Integer, primary_key=True,
autoincrement=True, unique=True)
webstatus = db.Column(db.INTEGER)
class Admin_Flagged(db.Model):
__tablename__ = 'flagged'
__bind_key__ = 'clearnet'
__table_args__ = {"schema": "public", 'extend_existing': True}
id = db.Column(db.Integer, primary_key=True,
autoincrement=True, unique=True)
user_id = db.Column(db.INTEGER)
vendorname = db.Column(db.TEXT)
howmany = db.Column(db.INTEGER)
typeitem = db.Column(db.INTEGER)
listingid = db.Column(db.INTEGER)
listingtitle = db.Column(db.INTEGER)
flagged_user_id_1 = db.Column(db.INTEGER)
flagged_user_id_2 = db.Column(db.INTEGER)
flagged_user_id_3 = db.Column(db.INTEGER)
flagged_user_id_4 = db.Column(db.INTEGER)
flagged_user_id_5 = - db.Column(db.INTEGER)
db.configure_mappers()
db.create_all()
db.session.commit()
# How to create a MasterDetailView
#class DetailView(ModelView):
# datamodel = SQLAInterface(DetailTable, db.session)
#class MasterView(MasterDetailView):
# datamodel = SQLAInterface(MasterTable, db.session)
# related_views = [DetailView]
# How to create a MultipleView
#class MultipleViewsExp(MultipleView):
# views = [GroupModelView, ContactModelView]
#View Registration
db.configure_mappers() #very important! for searchability
db.create_all()
fill_gender()
appbuilder.add_view(LawyerChartView(),
'Lawyer Age Chart',
icon='fa-dashboard',
category='Reports')
appbuilder.add_view(LawyerTimeChartView(),
'Lawyer Time Chart',
icon='fa-dashboard',
category='Reports')
appbuilder.add_view(PlaintiffChartView(),
'Plaintiff Age Chart',
icon='fa-dashboard',
def resetdb(fast=False):
"""Erase db and/or create a new one with an admin account."""
from pycountry import countries
from app.auth import models as auth_models
from app.seeds import models as seeds_models
from app.shop import models as shop_models
from app.shop.models import Country, State
resp = input(
'WARNNG: This will erase existing database and create a new one! '
'Proceed anyway? y/N: '
)
if 'y' in resp.lower():
print('Erasing existing database if present...')
db.session.rollback()
db.session.remove()
if db.engine.dialect.name == 'postgresql':
db.engine.execute('drop schema if exists public cascade')
db.engine.execute('create schema public')
db.drop_all()
print('Configuring mappers...')
db.configure_mappers()
print('Creating new database...')
db.create_all()
db.session.commit()
admin = User()
db.session.add(admin)
print('Populating countries table...')
db.session.add_all(
sorted(
Country.generate_from_alpha3s(c.alpha3 for c in countries),
key=lambda x: x.name
)
)
db.session.flush()
print('Setting safe to ship countries...')
stsfile = Path(
app.config['JSON_FOLDER'],
'safe_to_ship_countries.json'
)
try:
with stsfile.open('r', encoding='utf-8') as ifile:
sts = json.loads(ifile.read())
for c in sts:
if isinstance(c, str):
alpha3 = c
thresh = None
else:
alpha3 = c[0]
thresh = c[1]
country = Country.get(alpha3=alpha3)
if thresh:
country.at_own_risk_threshold = thresh
country.safe_to_ship = True
db.session.flush()
except FileNotFoundError:
db.session.rollback()
raise FileNotFoundError(
'Could not find file "{}". This file should be a JSON list '
'containing alpha3 country codes for countries we can safely '
'ship to, including ones that become at own risk above a '
'certain cost total, which should be 2 value lists formatted '
'["<alpha3", <int or decimal cost above which is at own '
'risk>], e.g.: [... , "JPN", "NLD", ["NOR", 50], "PRI", '
'"ESP", ...]'.format(stsfile.absolute())
)
print('Setting noship countries...')
ncfile = Path(app.config['JSON_FOLDER'], 'noship_countries.json')
try:
with ncfile.open('r', encoding='utf-8') as ifile:
a3s = json.loads(ifile.read())
for alpha3 in a3s:
country = Country.get(alpha3=alpha3)
country.noship = True
db.session.flush()
except FileNotFoundError:
db.session.rollback()
raise FileNotFoundError(
'Could not find file "{}"! This file should be a JSON list '
'containing alpha3 country codes for countries we cannot '
'ship to. e.g.: ["BGD", "BRA", "CHN", ... ]'
.format(ncfile.absolute())
)
print('Populating States/Provinces/etc...')
try:
sfile = Path(app.config['JSON_FOLDER'], 'states.json')
with sfile.open('r', encoding='utf-8') as ifile:
d = json.loads(ifile.read())
db.session.add_all(
State.generate_from_dict(d)
)
db.session.flush()
except FileNotFoundError:
db.session.rollback()
raise FileNotFoundError(
'Could not find file "{}"! If it does not exist, it should '
'be created and contain a JSON object formatted: { "<country '
'alpha3 code>": { "<state abbreviation>": "<state name>", '
'... }, ... } e.g. {"USA": {"AL": "Alabama", "AK": '
'"Alaska", ... }, "CAN": {"AB": "Alberta", "BC": '
'"British Columbia", ... }, ... }'.format(sfile.absolute())
)
print('Setting California sales tax...')
rfile = Path(app.config['JSON_FOLDER'], 'rates.json')
try:
with rfile.open('r', encoding='utf-8') as ifile:
rates = json.loads(ifile.read())
ca = State.get(
country=Country.get(alpha3='USA'), abbreviation='CA'
)
ca.tax = Decimal(str(rates['sales tax']['USA']['CA']))
db.session.flush()
except FileNotFoundError:
raise FileNotFoundError(
'Could not find file "{}"! It should contain a JSON object '
'including: { "sales tax": {"USA": {"CA":<tax rate>i, ... }, '
'... }, ... }'.format(rfile.absolute())
)
print('Creating first administrator account...')
if fast:
admin.name = 'admin'
admin.email = 'admin@localhost'
admin.set_password('sgsadmin') # Very secure!
else:
admin.name = input('Enter name for admin account: ')
admin.email = input('Enter email address for admin account: ')
while True:
pw = getpass('Enter new password: '******'Confirm new password: '******'Passwords do not match! Please try again.')
else:
break
admin.set_password(pw)
admin.grant_permission(Permission.MANAGE_SEEDS)
admin.grant_permission(Permission.MANAGE_USERS)
admin.confirmed = True
print('Admin account "{}" created!'.format(admin.name))
db.session.commit()
print('Database was successfully created!')
else:
print('Aborted.')
from app import db
from models import *
#Configure the mappers BEFORE create_all()
db.configure_mappers()
db.create_all()
db.session.add(User('admin', 'admin', 'admin', 'admin', 'admin', 'admin', 'admin'))
db.session.add(User('aaa', 'aaa', 'a', 'a', 'agate', '1', 'first'))
db.session.add(User('bbb', 'bbb', 'b', 'b', 'bbb', '2', 'first'))
db.session.add(User('ccc', 'ccc', 'c', 'c', 'agate', '2', 'first'))
db.session.add(filestable(name=u'gravity', filetype='mp4', size='14444', mediatype='movie', ownerid='1', ownerhostel='agate', views=0))
db.session.commit()
def resetdb():
"""Runs the unit tests without test coverage."""
db.session.close()
db.drop_all()
db.configure_mappers()
db.create_all()
