def setUp(self) -> None:
self.app = create_app('testing')
self.app_context = self.app.app_context()
self.app_context.push()
db.create_all()
Role.insert_roles()
self.client = self.app.test_client(use_cookies=True)
def create_db():
"""
Recreates a local database. You probably should not use this on
production.
"""
db.create_all()
db.session.commit()
def setUp(self):
self.app = create_app('testing')
self.app_context = self.app.app_context()
self.app_context.push()
db.create_all()
Role.insert_roles()
self.client = self.app.test_client()
def resetDatabase():
"""Reset database for testing purpose"""
db.drop_all()
db.create_all()
user_password_encrypted = sha256_crypt.hash('user')
admin_password_encrypted = sha256_crypt.hash('admin')
user = Users('user', '*****@*****.**', '081122112211',
user_password_encrypted, False)
admin = Users('admin', '*****@*****.**', '0811221122112',
admin_password_encrypted, True)
trash_category = ListTrashCategory(2, 'dummy_category')
trash_one = {
"trash_category_id": 1,
"admin_id": 2,
"trash_name": "dummy_trash",
"price": 1000,
"photo": "dummy_photo",
"point": 1
}
trash_two = {
"trash_category_id": 1,
"admin_id": 2,
"trash_name": "dummy_trash",
"price": 2000,
"photo": "dummy_photo",
"point": 2
}
trash_instance_one = ListTrash(trash_one)
trash_instance_two = ListTrash(trash_two)
reward = Rewards(2, "reward dummy", 20, "photo", 20, True)
reward1 = Rewards(2, "reward dummy", 20, "photo", 20, True)
reward2 = Rewards(2, "reward dummy", 20, "photo", 20, True)
order = ListOrders({
'user_id': 1,
'adress': "dummy",
'time': datetime.datetime.utcnow(),
'photo': 'url',
'status': 'waiting'
})
db.session.add(user)
db.session.add(admin)
db.session.add(trash_category)
db.session.add(trash_instance_one)
db.session.add(trash_instance_two)
db.session.add(reward)
db.session.add(reward1)
db.session.add(reward2)
db.session.add(order)
db.session.commit()
user_attr = UserAttributes(1, 0, 0, False)
admin_attr = UserAttributes(2, 0, 0, False)
db.session.add(user_attr)
db.session.add(admin_attr)
db.session.commit()
def client():
from apps import create_app, db
flask_app = create_app('testing')
testing_client = flask_app.test_client()
with flask_app.app_context():
db.create_all()
yield testing_client
db.drop_all()
def setUp(self):
"""function to setup app variables"""
config_name = config.TestingConfig
app.config.from_object(config_name)
self.app = app
self.test_client = self.app.test_client()
self.app_context = self.app.app_context()
self.app_context.push()
self.name = 'test user'
self.email = '*****@*****.**'
self.test_username = '******'
self.test_password = '******'
with self.app_context:
db.create_all()
def setUpClass(cls):
# start Firefox
try:
cls.client = webdriver.Firefox()
except:
pass
# skip these tests if the browser could not be started
if cls.client:
# create the application
cls.app = create_app('testing')
cls.app_context = cls.app.app_context()
cls.app_context.push()
# suppress logging to keep unittest output clean
import logging
logger = logging.getLogger('werkzeug')
logger.setLevel("ERROR")
# create the database and populate with some fake data
db.create_all()
Role.insert_roles()
User.generate_fake(10)
Post.generate_fake(10)
# add an administrator user
admin_role = Role.query.filter_by(permissions=0xff).first()
admin = User(email='*****@*****.**',
username='******',
password='******',
role=admin_role,
confirmed=True)
db.session.add(admin)
db.session.commit()
# start the Flask server in a thread
threading.Thread(target=cls.app.run).start()
# give the server a second to ensure it is up
time.sleep(1)
def setUp(self):
db.create_all()
db.session.commit()
tc_sms.send_sms = self.hook_sender
tc_oss.upload_image = self.hook_upload_image
class ArticleComment(db.Model):
__tablename__ = 'article_comment'
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
article_id = db.Column(db.Integer, db.ForeignKey('article.id'))
comment = db.Column(db.TEXT)
addtime = db.Column(db.DATETIME, index=True, default=datetime.now)
def __repr__(self):
return '<ArticleComment %r>' % self.id
if __name__ == "__main__":
db.drop_all()
db.create_all()
flag = 1
if flag == 0:
db.drop_all()
db.create_all()
if flag == 1:
tag0 = AlbumTag(name='风景')
tag1 = AlbumTag(name='动漫')
tag2 = AlbumTag(name='星空')
tag3 = AlbumTag(name='萌宠')
tag4 = AlbumTag(name='静物')
tag5 = AlbumTag(name='汽车')
tag6 = AlbumTag(name='海洋')
tag7 = AlbumTag(name='美女')
tag8 = AlbumTag(name='城市')
tag9 = AlbumTag(name='飞鸟')
def setUp(self):
self.app = app.test_client()
self.app.testing = True
db.drop_all()
db.create_all()
def create_db():
db.create_all()
def run():
db.create_all()
app.config.from_object('config.Config')
app.run(host=app.config['SERVER_IP'], port=app.config['PORT'])
def create_all():
"""Creates database tables"""
db.create_all()
def setUp(self):
self.app = create_app('testing')
self.app_context = self.app.app_context()
self.app_context.push()
db.create_all()
def load_cmd(aUseRandomValues=None):
if aUseRandomValues:
print(' >>> Randomize Values ')
else:
print(' >>> Use Values from input file ')
# Create Tables (if not exist)
db.create_all()
# Truncate
db.session.query(Data).delete()
db.session.commit()
with open('media/transactions_data.csv', newline='') as csvfile:
csvreader = csv.reader(csvfile) # load file
header = next(csvreader) # ignore header (1st line)
'''
Expected format:
HEADER: product_code,product_info,value,currency,type
SAMPLE: Lenovo_Ideapad_3i, Lenovo Ideapad 3i 14.0inch FHD Laptop,9.5,euro,transaction
product_code (string) : Lenovo_Ideapad_3i
product_info (string) : Lenovo Ideapad 3i 14.0inch FHD Laptop
value (integer) : 9
currency (string) : usd, eur
type (string) : transaction (hardoded)
'''
iter = 0 # used for timestamp
for row in csvreader:
iter += 1
if len(row) != 5:
print(' >>> Error parsing line (' + str(iter) + ') -> ' +
' '.join([str(elem) for elem in row]))
continue
item_code = row[0]
item_name = row[1]
if aUseRandomValues:
item_value = randint(5, 100)
else:
item_value = row[2]
item_currency = row[3]
item_type = row[4]
# randomize in the past the transaction date
# The distribuition range ~1 month
item_ts = datetime.utcnow().timestamp() - (2 * iter *
randint(5, 10) * 3600)
item_ts = int(item_ts)
_data = Data(code=item_code,
name=item_name,
value=item_value,
currency=item_currency,
type=item_type,
ts=item_ts)
db.session.add(_data)
db.session.commit()
def initdb():
'''Creates all database tables.'''
db.create_all()
def init_db():
db.create_all()
def create_all():
"""Creates database tables"""
db.create_all()
def main():
db.create_all()
load_admin()
if len(sys.argv) == 2:
load_file_in_db(sys.argv[1])
def setUpClass(cls):
cls.app = createApp()
cls.app_context = cls.app.app_context()
cls.app_context.push()
cls.client = cls.app.test_client()
db.create_all()
def syncdb():
with app.test_request_context():
from apps.users.models import User
db.create_all()
#!flask/bin/python
from migrate.versioning import api
from config import SQLALCHEMY_DATABASE_URI
from config import SQLALCHEMY_MIGRATE_REPO
from apps import db
import os.path
db.create_all()
if not os.path.exists(SQLALCHEMY_MIGRATE_REPO):
api.create(SQLALCHEMY_MIGRATE_REPO, 'database repository')
api.version_control(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_REPO)
else:
api.version_control(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_REPO, api.version(SQLALCHEMY_MIGRATE_REPO))
def syncdb():
with app.test_request_context():
from apps.users.models import User
db.create_all()
