def init_database(test_client): # Create the database and the database table # Insert user data db.create_all() db.session.add_all(branch_helper.create_branch_array()) db.session.commit() user1 = User('Dipin', '*****@*****.**', 'abc', '1', 1) user2 = Professor("Professor 1", "*****@*****.**", "prof1", 2) # user_other_branch = User('Ram', '*****@*****.**', 'abc', '1', 2) db.session.add(user1) db.session.add(user2) # Commit the changes for the users db.session.commit() course = Course(details="Course on Data structures", prof_id=user2.id, name="Data structures", course_code='203', can_apply=True) db.session.add(course) db.session.commit() quiz = Quiz(course_id=course.id, name="Quiz1", start_time=datetime(2015, 6, 5, 8, 10, 10, 10), end_time=datetime(2015, 6, 5, 8, 10, 12, 10)) db.session.add(quiz) db.session.commit() # Adding two Question in Quiz1 question1 = Question(quiz_id=quiz.id, question="Odd one out", ans='4', marks=2, is_multicorrect=True, is_partial=True) question2 = Question(quiz_id=quiz.id, question="Cities in Maharastra", ans='2,3,4', marks=4, is_multicorrect=True, is_partial=True) db.session.add(question1) db.session.add(question2) db.session.commit() # Options for 1st question option1 = Option(question_id=question1.id, option='django', is_right=False) option2 = Option(question_id=question1.id, option='flask', is_right=False) option3 = Option(question_id=question1.id, option='ruby on rails', is_right=False) option4 = Option(question_id=question1.id, option='expressjs', is_right=True) db.session.add(option1) db.session.add(option2) db.session.add(option3) db.session.add(option4) db.session.commit() # Options for 2nd question option1 = Option(question_id=question2.id, option='Indore', is_right=False) option2 = Option(question_id=question2.id, option='Nasik', is_right=True) option3 = Option(question_id=question2.id, option='Mumbai', is_right=True) option4 = Option(question_id=question2.id, option='Bombay', is_right=True) db.session.add(option1) db.session.add(option2) db.session.add(option3) db.session.add(option4) db.session.commit() # Request for enrollment in course req = Request(user_id=user1.id, course_id=course.id, title="Request to Access Course", details="Please allow me!!") db.session.add(req) db.session.commit() yield # this is where the testing happens! db.drop_all()
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///' + \ os.path.join(basedir, '..', '..', 'data.sqlite') app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False if os.getenv('testing') == 'true': print("TESTING MODE") app.config['WTF_CSRF_ENABLED'] = False app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///' + \ os.path.join(basedir, '..', '..', 'data_test.sqlite') db = SQLAlchemy(app) login_manager = LoginManager() login_manager.init_app(app) login_manager.login_view = 'professors.login' ALLOWED_EXT = ['pdf', 'jpg', 'jpeg', 'png'] import cms.models import cms.branch_helper as branch_helper db.create_all() db.session.add_all(branch_helper.create_branch_array()) db.session.commit() from cms.core.views import core from cms.error_pages.handlers import error_pages from cms.professor.views import professors from cms.course.views import CourseBluerint app.register_blueprint(core) app.register_blueprint(error_pages) app.register_blueprint(professors) app.register_blueprint(CourseBluerint) print("route path is, ", core.root_path)
import os from flask_login import LoginManager from sqlalchemy.orm import session app = Flask(__name__) app.config['SECRET_KEY'] = 'mysecret' basedir = os.path.abspath(os.path.dirname(__file__)) print(basedir) app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///' + \ os.path.join(basedir, '..', '..', 'data.sqlite') app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False db = SQLAlchemy(app) Migrate(app, db, render_as_batch=True) login_manager = LoginManager() login_manager.init_app(app) login_manager.login_view = 'professors.login' import cms.models db.create_all() from cms.branch_helper import create_branch_array db.session.add_all(create_branch_array()) db.session.commit() db.session.add( cms.models.User('Dipin', '*****@*****.**', 'dipin', '2', 1)) db.session.add( cms.models.Professor('Professor One', '*****@*****.**', 'prof1', '1')) db.session.commit()