Пример #1
0
def createdb():
	print 'creating db..'
	db.drop_all()
	db.create_all()
	Role.insert_roles()

	print 'adding feed sources..'
	with open('sources.yaml', 'r') as f:
		feed_sources = yaml.load(f)

	# Insert feedprovider
	for site in feed_sources:
		provider_name = unicode(site['name'])
		provider_domain = site['domain']
		provider = FeedProvider(name = provider_name,
								domain = provider_domain,
								favicon = site['favicon'])
		db.session.add(provider)
		db.session.commit()

		# Add feedsource
		for channel_name in site['channels']:
			source_name = unicode(site['name'] + ' - ' + channel_name)
			source_href = site['channels'][channel_name]
			fs = FeedSource(name=source_name, url=source_href, provider=provider)
			db.session.add(fs)

	db.session.commit()

	print 'add feed articles..'
	from app.mod_crawler.fetch import update_db
	update_db()
 def setUp(self):
     app = create_app('testing')
     self.app = app
     self.ctx = app.app_context()
     self.ctx.push()
     self.client = app.test_client()
     db.create_all()
Пример #3
0
 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 create_dev_database():
    """
    Use this when running a development server.  It creates tables without
    using db migration scripts, instead asking SQLAlchemy to create them
    directly from your models.  Good for testing/iterating.  When done,
    write proper db migration scripts and test.
    """
    from app import db
    from app.models import Role, User
    db.drop_all()
    db.create_all()

    admin_role = Role(name='Admin')
    mod_role = Role(name='Moderator')
    user_role = Role(name='User')
    db.session.add(admin_role)
    db.session.add(mod_role)
    db.session.add(user_role)

    user_john = User(username='******', role=admin_role, password='******', email='*****@*****.**', confirmed=True)
    user_susan = User(username='******', role=user_role, password='******', email='*****@*****.**', confirmed=False)
    user_david = User(username='******', role=user_role, password='******', email='*****@*****.**', confirmed=False)

    db.session.add(user_john)
    db.session.add(user_susan)
    db.session.add(user_david)
    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()
     self.client = self.app.test_client()
     self.url = api.url_for(RegisterResource)
Пример #6
0
    def setUp(self):
        app.config['TESTING'] = True
        app.config['SQLALCHEMY_DATABASE_URI']='sqlite:///'+\
                os.path.join(basedir, TEST_DB)

        self.app = app.test_client()
        db.create_all()
Пример #7
0
    def setUp(self):
        """
            Initialize the test cases.
        """
        self.app = create_app(TestConfiguration)
        self.client = self.app.test_client()
        self.app_context = self.app.app_context()
        self.app_context.push()
        self.request_context = self.app.test_request_context('/')
        self.request_context.push()
        db.create_all()

        # Add a few test models.
        role_1 = Role(name='A')
        role_2 = Role(name='B')
        role_3 = Role(name='C')
        role_4 = Role(name='D')
        role_5 = Role(name='E')
        role_6 = Role(name='F')
        role_7 = Role(name='G')
        db.session.add(role_1)
        db.session.add(role_2)
        db.session.add(role_3)
        db.session.add(role_4)
        db.session.add(role_5)
        db.session.add(role_6)
        db.session.add(role_7)
        db.session.commit()
Пример #8
0
	def setUpClass(cls):
		#start Firefox
		try:
			cls.client = webdriver.Firefox()
		except:
			pass

		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 unitteset output clean
			import logging
			logger = logging.getLogger('werkzeug')
			logger.setLevel("ERROR")

			#create database with fake user and post
			db.create_all()

			u = User(username='******', password='******', role=1)
			db.session.add(u)

			p = Post(title='test_post', body='test_post_body', user_id=u.id)
			db.session.add(p)
			db.session.commit()

			#start the Flask server in a thread
        	threading.Thread(target=cls.app.run).start()
Пример #9
0
    def setUpClass(cls):
        try:
            cls.client = webdriver.Firefox()
        except:
            pass

        if cls.client:
            cls.app = create_app('testing')
            cls.app_context = cls.app_context()
            cls.app_context.push()

            import logging
            logger = logging.getLogger('werkzeug')
            logger.setLevel("ERROR")

            db.create_all()
            Role.insert_roles()
            User.generate_fake(10)
            Post.generate_fake(10)

            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()

            threading.Thread(target=cls.app.run).start()
            time.sleep(1)
Пример #10
0
def create_user():
    db.create_all()
    user_datastore.create_user(email='*****@*****.**', password='******')
    user_datastore.create_role(name='user', description='default role')
    user_datastore.create_role(name='server_admin', description='Admin of a Go Server')
    user_datastore.create_role(name='ratings_admin', description='Admin of AGA-Online Ratings')
    db.session.commit()
Пример #11
0
    def setUp(self):
        """
            Initialize the test cases.
        """
        self.app = create_app(TestConfiguration)
        self.app_context = self.app.app_context()
        self.app_context.push()
        self.request_context = self.app.test_request_context('/')
        self.request_context.push()
        db.create_all()

        # Add a few test models.
        self.model_1 = TestModel()
        self.model_2 = TestModel()
        self.model_3 = TestModel()
        self.model_4 = TestModel()
        self.model_5 = TestModel()
        self.model_6 = TestModel()
        self.model_7 = TestModel()
        db.session.add(self.model_1)
        db.session.add(self.model_2)
        db.session.add(self.model_3)
        db.session.add(self.model_4)
        db.session.add(self.model_5)
        db.session.add(self.model_6)
        db.session.add(self.model_7)
        db.session.commit()
Пример #12
0
 def test_notification_list(self):
     db.create_all()
     u1 = User(email='*****@*****.**', username='******', password='******')
     u2 = User(email='*****@*****.**', username='******', password='******')
     t = Talk(title='t', description='d', author=u1)
     c1 = Comment(talk=t, body='c1', author_name='n1',
                  author_email='*****@*****.**', approved=True)
     c2 = Comment(talk=t, body='c2', author_name='n2',
                  author_email='*****@*****.**', approved=True, notify=False)
     c3 = Comment(talk=t, body='c3', author=u2, approved=True)
     c4 = Comment(talk=t, body='c4', author_name='n4',
                  author_email='*****@*****.**', approved=False)
     c5 = Comment(talk=t, body='c5', author=u2, approved=True)
     c6 = Comment(talk=t, body='c6', author_name='n6',
                  author_email='*****@*****.**', approved=True, notify=False)
     db.session.add_all([u1, u2, t, c1, c2, c3, c4, c5])
     db.session.commit()
     email_list = c4.notification_list()
     self.assertTrue(('*****@*****.**', 'n1') in email_list)
     self.assertFalse(('*****@*****.**', 'n2') in email_list)  # notify=False
     self.assertTrue(('*****@*****.**', 'susan') in email_list)
     self.assertFalse(('*****@*****.**', 'n4') in email_list)  # comment author
     self.assertFalse(('*****@*****.**', 'n6') in email_list)
     email_list = c5.notification_list()
     self.assertFalse(('*****@*****.**', 'john') in email_list)
     self.assertTrue(('*****@*****.**', 'n4') in email_list)  # comment author
Пример #13
0
    def setup(self):
        app.config['TESTING'] = True
        app.config['CSRF_ENABLED'] = False
        app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///' + os.path.join(basedir, 'test.db')
        self.app = app.test_client()
        db.create_all()

        def teardown(self):
            db.session.remove()
            db.drop_all()

        def test_avatar(self):
            user = User(nickname='john', email='*****@*****.**')
            avatar = user.avatar(128)
            expected = 'http://www.gravatar.com/avatar/d4c74594d841139328695756648b6bd6'
            assert avatar[0:len(expected)] == expected

        def test_make_unique_nickname(self):
            user = User(nickname='john', email='*****@*****.**')
            db.session.add(user)
            db.session.commit()
            nickname = User.make_unique_nickname('john')

            assert nickname != 'john'
            u = User(nickname = nickname, email='*****@*****.**')
            db.session.add(u)
            db.session.commit()

            nickname2 = User.make_unique_nickname('john')

            assert nickname2 != 'john'
            assert nickname2 != nickname
Пример #14
0
 def setUp(self):
     db.session.close()
     db.drop_all()
     db.create_all()
     self.app = app.test_client()
     if not os.path.exists('users/'):
         os.makedirs('users/')
Пример #15
0
    def setup(self, app):
        self.db_bind = 'ivr'
        mydbplugindir = os.path.join(app.config['BASEDIR'],'app/plugins/ivr/db/ivr.db')
        app.config['SQLALCHEMY_BINDS'].update({self.db_bind : 'sqlite:///%s' % mydbplugindir})

        with app.app_context():
            db.create_all(bind=self.db_bind)
Пример #16
0
def recreate_tables():
    """
    dangerous
    """
    if query_yes_no('This will erase all data in the database. Continue?', default='no'):
        db.drop_all()
        db.create_all()
Пример #17
0
def profile_view(id_num):
    encoded = jwt.encode({'some': 'payload'}, 'secret', algorithm='HS256')
    current_id = session['id_num']
    profile = Myprofile.query.get(id_num)
    data = Mylist.query.filter_by(id_num=current_id).all()
    if request.method == 'GET':
        for each in data:
            if each.id_num == current_id:
                return render_template('profile_view.html', profile=profile, data=data)
        return render_template('profile_view.html', profile=profile)
    else:
        db.create_all()
        url = request.form['image']
        description = request.form['description']
        title = request.form['title']
        id_num = session['id_num']
        new_list = Mylist(id_num=id_num,
                               description=description, url=url, title=title)
        db.session.add(new_list)
        db.session.commit()
        payload = {'some': 'data'}
        jwt_token = jwt.encode({'some': 'payload'}, 'secret', algorithm='HS256')
        
        #send data to database here
        return redirect(url_for('profile_view', profile=profile, id_num=current_id, encoded=encoded))
Пример #18
0
    def setUpClass(cls):
        # 启动Firefox
        try:
            cls.client = webdriver.Firefox()
        except:
            pass

        # 如果无法启动浏览器,则跳过这些测试
        if cls.client:
            # 创建程序
            cls.app = create_app('testing')
            cls.app_context = cls.app.app_context()
            cls.app_context.push()

            # 禁止日志,保持输出简洁
            import logging
            logger = logging.getLogger('werkzeug')
            logger.setLevel("ERROR")

            # 创建数据库
            db.create_all()
            Role.insert_role()

            # 添加管理员
            admin_role = Role.query.filter_by(permission=0xff).first()
            admin = User(email='*****@*****.**',
                         username='******', pasword='cat',
                         role=admin_role, confirmed=True)
            db.session.add(admin)
            db.session.commit()

            # 在一个线程中启动Flask服务器
            threading.Thread(target=cls.app.run).start()
Пример #19
0
    def setUp(self):
        db.create_all()
        db.session.add(Place(name='Testname',
                             address="123 Test st.",
                             city="Englewood",
                             state="Florida",
                             zip_="34224",
                             website="http://fake.com",
                             phone="123-456-7899",
                             owner="Jeff Reiher",
                             yrs_open=1))

        db.session.add(Menu(name="burger",
                            course="dinner",
                            description="test description",
                            price="$1.00",
                            place_id=1))
        
        db.session.add(Users(username="******",
                            email="*****@*****.**",
                            password=bcrypt.generate_password_hash("password"),
                            avatar="picofjeff.jpg"))

        
        db.session.commit()
Пример #20
0
 def setUp(self):
     app.config['TESTING'] = True
     app.config['CSRF_ENABLED'] = False
     app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///' + \
         os.path.join(basedir, 'test.db')
     self.app = app.test_client()
     db.create_all()
Пример #21
0
 def setUp(self):
     """Before each test, set up a blank database"""
     try:
         shutil.rmtree(self.whoosh)
     except OSError as err:
         pass
     db.create_all()
    def setUp(self):
        db.create_all()

        self.create_author_account()
        self.add_story()

        db.session.commit()
 def setUpClass(cls):
     try:
         cls.client = webdriver.Firefox()
     except:
         pass
     if cls.client:
         # create app
         cls.app = create_app('testing')
         cls.app_context = cls.app.app_context()
         cls.app_context.push()
         # create log
         import logging
         logger = logging.getLogger('werkzeug')
         logger.setLevel(logging.ERROR)
         # create database
         db.create_all()
         Role.init_roles()
         User.generate_fake(count=10)
         Post.generate_fake(count=10)
         # add admin
         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()
         # run server in child thread
         Thread(target=cls.app.run).start()
Пример #24
0
 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(use_cookies=True)
Пример #25
0
def db(app, request):
    _db.app = app
    _db.create_all()

    yield _db

    _db.drop_all()
Пример #26
0
def db_create():
	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))
Пример #27
0
 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(use_cookies=True) #测试客户端对象 (启用cookie像浏览器一样可能接收和发半发送cookie)
Пример #28
0
    def setUpClass(cls):
        try:
            cls.client = webdriver.Chrome(executable_path=chrome_driver)
        except:
            pass

        if cls.client:
            # 创建 app
            cls.app = create_app('test')
            cls.app_context = cls.app.app_context()
            cls.app_context.push()

            # 建立数据库
            db.create_all()
            setting = Settings(
                site_admin_email='*****@*****.**',
                site_initiated=True,
            )
            db.session.add(setting)
            db.session.commit()

            user = User(
                name='test_user',
                password='******',
                email='*****@*****.**'
            )
            db.session.add(user)
            db.session.commit()

            cls.host = 'http://localhost:5000'

            import logging
            logger = logging.getLogger('werkzeug')
            logger.setLevel("ERROR")
            threading.Thread(target=cls.app.run).start()
Пример #29
0
    def setUp(self):
        self.app = create_app('testing')

        self.app_context = self.app.app_context()
        self.app_context.push()

        db.create_all()
Пример #30
0
	def setUpClass(cls):
		#launch Firefox
		try:
			cls.client = webdriver.Chrome()
		except:
			pass

		if cls.client:
			cls.app = create_app('testing')
			cls.app_context = cls.app.app_context()
			cls.app_context.push()

			import logging
			logger = logging.getLogger('werkzeug')
			logger.setLevel('ERROR')

			# create database, and fill it up with some faked data
			db.create_all()
			Role.insert_roles()
			User.generate_fake(10)
			Post.generate_fake(10)

			# add administrater
			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()

			# launch 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)
Пример #31
0
 def setUp(self) -> None:
     self.app = create_app(TestConfig)
     self.app_context = self.app.app_context()
     self.app_context.push()
     db.create_all()
Пример #32
0
 def create_app(self):
     db.init_app(app)
     db.create_all()
     app.config['TESTING']
     return app
Пример #33
0
# Create the 2 table: Users and Activities table
class Users(db.Model):
    # Users Table
    # 2 attributes:
    ###  Username (VARCHAR): unique username, primary key <50 characters
    ###  Password (VARCHAR): user's password <50 characters

    __tablename__ = 'Users'
    Username = db.Column(db.VARCHAR(50), primary_key=True)
    Password = db.Column(db.VARCHAR(50), nullable=False)


class Activities(db.Model):
    # Activities Table
    # 4 Attributes:
    ### row_id (INT): Primary Key: unique + autoincrement
    ### Username (VARCHAR): unique username, primary key <50 characters
    ### Activity (VARCHAR): Descriptions of the task < 200 characters
    ### Status (INT): Status of the activity (1: do; 2: doing; 3: done)

    __tablename__ = 'Activities'
    row_id = db.Column(db.Integer, primary_key=True)
    Username = db.Column(db.VARCHAR(50))
    Activity = db.Column(db.VARCHAR(200))
    Status = db.Column(db.Integer)


# Initialize database
db.create_all()
db.session.commit()
Пример #34
0
from app import db, Countries, Cities
db.create_all()  # Creates all table classes defined

UK = Countries(name='United Kingdom')  #Add example to countries table
db.session.add(UK)
db.session.commit()

# Here we reference the country that london belongs to useing 'country', this is what we named the backref variable in db.relationship()
ldn = Cities(name='London', country=UK)
mcr = Cities(name='Manchester',
             country=Countries.query.filter_by(name='United Kingdom').first())

db.session.add(ldn)
db.session.add(mcr)
db.session.commit()
Пример #35
0
 def setUp(self):
     db.create_all()
def deploy():
    db.create_all()
    Role.preset_roles()
Пример #37
0
def start_project():
    """
    start project start-project >>>db.create_all()
    """
    db.create_all()
Пример #38
0
def init_db_user():
    db.create_all()
    user=User('admin','meitianhui')
    db.session.add(user)
    db.session.commit()
Пример #39
0
 def setUp(self):
     self.app = create_app('testing')
     self.app_context = self.app.app_context()
     self.app_context.push()
     db.create_all()
Пример #40
0
def save_todo():
    db.create_all()
    todo_first = Todo(content="111")
    todo_second = Todo(content="222")
    db.session.add_all([todo_first, todo_second])
    db.session.commit()
Пример #41
0
 def setUp(self):
     app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite://'
     db.create_all()
Пример #42
0
 def __init__(self):
     db.create_all()
Пример #43
0
def create_db():
    """Create tables in SQL database from SQLAlchemy models.
    """
    db.engine.execute("CREATE EXTENSION IF NOT EXISTS postgis;")
    db.create_all()
Пример #44
0
 def setUp(self):
     '''在每个测试前运行, 初始化测试环境'''
     db.create_all()
Пример #45
0
 def setUp(self):
     self.app = create_app('testing')  # 创建测试app
     self.app_context = self.app.app_context()
     self.app_context.push()  # 激活上下文
     db.create_all()  # 创建数据库
Пример #46
0
 def setUp(self):
     self.app = app.test_client()
     self.app.application.config['TEST'] = True
     self.app.application.config[
         'SQLALCHEMY_DATABASE_URI'] = 'sqlite:///:memory:'
     db.create_all()
Пример #47
0
# coding:utf-8
#!/usr/bin/env python

from app import create_app, db
from app.models import User, Bullet
from flask_script import Manager, Shell

app = create_app('default')
manager = Manager(app)


def make_shell_context():
    return dict(app=app, db=db, User=User, Bullet=Bullet)


manager.add_command("shell", Shell(make_context=make_shell_context))

if __name__ == '__main__':
    with app.app_context():  # 这两句要加 不然会出现 relation does not exist 报错
        db.create_all()  # 或者提前建好表应该也可以
    manager.run()
 def setUpClass(self):
     app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///'
     self.app = app.test_client()
     db.create_all()
Пример #49
0
def createdb():
    "Create database"
    if prompt_bool("Do you want to create a database?"):
        db.create_all()
Пример #50
0
 def setUp(self):
     app.config['TESTING'] = True
     app.config['CSRF_ENABLED'] = False
     app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///' + os.path.join(basedir, 'test.db')
     db.create_all()
Пример #51
0
def init_db():
    db.drop_all()
    db.create_all()
Пример #52
0
def db_create():
    db.create_all()
Пример #53
0
 def setUp():
     print("Rodando setup...\n")
     with app.app_context():
         db.create_all()