def create_db(): FLASK_CONFIG = os.getenv("FLASK_CONFIG", "None") app = create_app(FLASK_CONFIG) app_context = app.app_context() app_context.push() from project.serate.models import Serata from project.corsi.models import Corso from project.tags.models import Tag from project.ruoli.models import Ruolo try: ruolo_list = Ruolo.query.all() except Exception as message: print("No db data") print("Creazione struttura senza utenti, post e commenti") db.create_all() print("Start creating roles") Ruolo.insert_roles() print("Start creating tags") Tag.insert_test_tags() print("Start creating corsi") Corso.insert_test_corsi() print("Start creating serate") Serata.insert_test_serate() db.session.remove() app_context.pop()
def setUp(self): self.app = create_app('testing') self.app_context = self.app.app_context() self.app_context.push() db.create_all() Tag.insert_test_tags() Corso.insert_test_corsi() Serata.insert_test_serate() self.client = self.app.test_client()
def create_test_db(): print("Start creating test db") FLASK_CONFIG = os.getenv("FLASK_CONFIG", "None") app = create_app(FLASK_CONFIG) app_context = app.app_context() app_context.push() from project.serate.models import Serata from project.corsi.models import Corso from project.tags.models import Tag from project.ruoli.models import Ruolo from project.utenti.models import Utente from project.blog.models import Post from project.commenti.models import Comment from random import randint from faker import Faker try: user_list = Utente.query.all() course_list = Corso.query.all() post_list = Post.query.all() comment_list = Comment.query.all() post_list = Post.query.all() ruolo_list = Ruolo.query.all() print("DB Tables already exists") except Exception as message: print(f"No db data exist, inserting them:") def users(count=100): fake = Faker("it_IT") i = 0 while i < count: u = Utente( email=fake.email(), username=fake.user_name(), password="******", confirmed=True, name=fake.name(), location=fake.city(), about_me=fake.text(), member_since=fake.past_date(), ) db.session.add(u) try: db.session.commit() i += 1 except IntegrityError: db.session.rollback() def posts(count=100): fake = Faker("it_IT") user_count = Utente.query.count() for i in range(count): u = Utente.query.offset(randint(0, user_count - 1)).first() p = Post(body=fake.text(), timestamp=fake.past_date(), author=u) db.session.add(p) db.session.commit() def comments(count=100): fake = Faker("it_IT") user_count = Utente.query.count() post_count = Post.query.count() for i in range(count): u = Utente.query.offset(randint(0, user_count - 1)).first() p = Post.query.offset(randint(0, post_count - 1)).first() c = Comment(body=fake.text(), timestamp=fake.past_date(), post=p, author=u) db.session.add(c) db.session.commit() print("Creating structure") db.create_all() db.session.commit() print("Creating roles") Ruolo.insert_roles() print("Creating fake users") users(2) print("Creating test users") Utente.insert_test_users() print("Creating tags") Tag.insert_test_tags() print("Creating corsi") Corso.insert_test_corsi() print("Creating serate") Serata.insert_test_serate() print("Creating posts fake") posts(3) print("Creating commenti fake") comments(3) print("\nDB Dummy data inserted succesfully") db.session.remove() app_context.pop()
print("Creating structure") db.create_all() print("Creating roles") Ruolo.insert_roles() print("Creating fake users") users(20) print("Creating test users") Utente.insert_test_users() print("Creating tags") Tag.insert_test_tags() print("Creating corsi") Corso.insert_test_corsi() print("Creating serate") Serata.insert_test_serate() print("Creating posts fake") posts() print("Creating commenti fake") comments() ###DEBUGS print("\n#### DATA DEBUG ####\n") # Read a course courses = Corso.query.all()