def setUp(self): self.app = create_app('testing') self.app_context = self.app.app_context() self.app_context.push() database.create_all() Role.insert_roles() self.client = self.app.test_client(use_cookies=True) self.bob = User(username='******', password='******', email='*****@*****.**', confirmed=True) self.arthur = User(username='******', password='******', email='*****@*****.**', confirmed=True) self.clair = User(username='******', password='******', email='*****@*****.**', confirmed=True) self.chat_bob_arthur = Chat() self.chat_bob_arthur.add_users([self.bob, self.arthur]) database.session.add_all( [self.bob, self.arthur, self.clair, self.chat_bob_arthur]) database.session.commit()
def __init__(self): database.create_all() if not os.path.exists(SQLALCHEMY_MIGRATE_CONT): api.create(SQLALCHEMY_MIGRATE_CONT, "database container") api.version_control(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_CONT) else: api.version_control(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_CONT, api.version(SQLALCHEMY_MIGRATE_CONT))
def create_new_db(): # Drop old database, then make a new one. print('Clearing current database...') database.reflect() database.drop_all() print('Creating new database...') database.create_all() database.session.commit() print("Database created.") print("Loading data from JSON...")
def client(): file = tempfile.mkstemp() db, app.config["SQLALCHEMY_DATABASE_URI"] = file[ 0], "sqlite:///" + file[1] + ".db" app.config["TESTING"] = True with app.test_client() as client: with app.app_context(): database.init_app(app) database.create_all() database.session.commit() yield client os.close(db)
def setUp(self): self.app = create_app('testing') self.app_context = self.app.app_context() self.app_context.push() database.create_all() Role.insert_roles() self.bob = User(username='******', password='******', email='*****@*****.**', confirmed=True) self.arthur = User(username='******', password='******', email='*****@*****.**', confirmed=True) self.clair = User(username='******', password='******', email='*****@*****.**', confirmed=True) self.morgana = User(username='******', password='******', email='*****@*****.**', confirmed=True) self.ophelia = User(username='******', password='******', email='*****@*****.**', confirmed=True) self.artorias = User(username='******', password='******', email='*****@*****.**', confirmed=True) self.chat_bob_arthur = Chat() self.chat_bob_artorias = Chat() self.chat_bob_clair = Chat() self.chat_morgana_arthur = Chat() self.chat_morgana_bob = Chat(name='chat_morgana_bob_') self.chat_bob_arthur.add_users([self.bob, self.arthur]) self.chat_bob_artorias.add_users([self.bob, self.artorias]) self.chat_bob_clair.add_users([self.bob, self.clair]) self.chat_morgana_arthur.add_users([self.arthur, self.morgana]) self.chat_morgana_bob.add_users([self.bob, self.morgana]) database.session.add_all([ self.bob, self.arthur, self.ophelia, self.artorias, self.clair, self.morgana, self.chat_bob_arthur, self.chat_bob_artorias, self.chat_bob_clair, self.chat_morgana_arthur, self.chat_morgana_bob ]) database.session.commit()
def app(request): app = create_app(TestConfig) app.app_context().push() # def teardown(): # app.app_context().pop() # request.addfinalizer(teardown) # return app with app.app_context(): # alternative pattern to app.app_context().push() # all commands indented under 'with' are run in the app context database.create_all() yield app # Note that we changed return for yield, see below for why database.session.remove( ) # looks like db.session.close() would work as well database.drop_all()
def setUp(self): self.app = create_app('testing') self.app_context = self.app.app_context() self.app_context.push() database.create_all() Role.insert_roles() self.bob = User(username='******', password='******', email='*****@*****.**', confirmed=True) self.arthur = User(username='******', password='******', email='*****@*****.**', confirmed=True) self.chat_bob_arthur = Chat() self.chat_bob_arthur.add_users([self.bob, self.arthur]) self.message1 = Message(text='hi there1', sender=self.arthur, recipient=self.bob, was_read=True, chat=self.chat_bob_arthur) self.message2 = Message(text='hi there2', sender=self.arthur, recipient=self.bob, chat=self.chat_bob_arthur) self.message3 = Message(text='hi there3', sender=self.arthur, recipient=self.bob, chat=self.chat_bob_arthur) self.message4 = Message(text='hi there4', sender=self.bob, recipient=self.arthur, chat=self.chat_bob_arthur) database.session.add_all([ self.bob, self.arthur, self.chat_bob_arthur, self.message1, self.message2, self.message3, self.message4 ]) database.session.commit()
def authenticated_client(): file = tempfile.mkstemp() db, app.config["SQLALCHEMY_DATABASE_URI"] = file[ 0], "sqlite:///" + file[1] + ".db" app.config["TESTING"] = True with app.test_client() as client: with app.app_context(): database.init_app(app) database.create_all() database.session.commit() login.init_app(app) test_user = User(username="******", email="*****@*****.**") test_user.set_password("admin") database.session.add(test_user) database.session.commit() test_todo_1 = Todo(task="Cooking", owner=test_user) test_todo_2 = Todo(task="Playing", owner=test_user) database.session.add(test_todo_1) database.session.add(test_todo_2) database.session.commit() yield client os.close(db)
def client(): file = tempfile.mkstemp() db, app.config["SQLALCHEMY_DATABASE_URI"] = file[ 0], "sqlite:///" + file[1] + ".db" app.config["TESTING"] = True with app.test_client() as client: with app.app_context(): database.init_app(app) database.create_all() database.session.commit() login.init_app(app) test_user = User(username="******", email="*****@*****.**") test_user.set_password("admin") database.session.add(test_user) database.session.commit() @login.user_loader def load_user(user_id): return User.query.get(int(user_id)) yield client os.close(db)
def reset(): database.drop_all() database.create_all() importData()
def initdb(): create_all()
from datetime import datetime from app import database as db class ChatroomMessages(db.Model): id = db.Column(db.Integer, primary_key=True, autoincrement=True) chatroomID = db.Column(db.Integer) message = db.Column(db.String(10050), nullable=False) sentUserID = db.Column(db.Integer, nullable=False) timestamp = db.Column(db.DateTime, nullable=False) def addChatroomMessage(cchatroomID, cmessage, csentUserID, ctimestamp): cr_message = ChatroomMessages(chatroomID=cchatroomID, message=cmessage, sentUserID=csentUserID, timestamp=ctimestamp) db.session.add(cr_message) db.session.commit() return True def getChatroomMessages(cid): return ChatroomMessages.query.filter_by(chatroomID=cid) db.create_all() #Call this before doing any database stuff
def setUp(self): database.session.close() database.drop_all() database.create_all()
from datetime import datetime from app import database as db class UserBadges(db.Model): id = db.Column(db.Integer, primary_key=True, autoincrement=True) uid = db.Column(db.Integer, nullable=False) badgesID = db.Column(db.Integer, nullable=False) def addUserBadges(buid, bbadgesID): user_badge = UserBadges(uid=buid, badgesID=bbadgesID) db.session.add(user_badge) db.session.commit() return True db.create_all()
def setUp(self): app.config["TESTING"]=True app.config["CSRF_ENABLED"]=False app.config["SQLALCHEMY_DATABASE_URI"]="postgresql://*****:*****@localhost:5432/database" self.app=app.test_client() database.create_all()
def setUp(self): db.create_all(app=app) create_user('first', 'last', 'email', 'uname', 'pw') self.user_id = get_user_with_username('uname').id create_project('name', 'desc', self.user_id) self.project = Project.query.filter_by(name='name').first()
import app app = app.create_app() from app import database with app.app_context(): database.create_all()
def setUp(self): db.create_all(app=app)
def setUp(self): self.app = create_app("test") self.app_context = self.app.app_context() self.app_context.push() database.create_all() self.client = self.app.test_client(use_cookies=True)
class ItemForm(Form): name = TextField( 'Item', validators=[DataRequired()] ) category = TextField( 'Category', validators=[DataRequired()] ) user = SelectField( 'Owner', validators=[DataRequired()] ) description = TextAreaField( 'Description', validators=[DataRequired()] ) submit = SubmitField() delete = SubmitField() class UserForm(Form): name = TextField( 'Name', validators=[DataRequired()] ) email = TextField( 'Email', validators=[DataRequired(), Length(min=6, max=40)] ) blueprint.backend = SQLAlchemyBackend(OAuth, database.session, cache=cache, user=current_user) database.create_all()
def create_table(): from app import database database.create_all() return redirect(url_for('basic.home'))
def create_table(): from app import database database.create_all() return redirect(url_for("basic.home"))