def create_tables(): from models import Image, Imageuri, ImageFragment, Metadata, Annotation Image.create_table() Imageuri.create_table() ImageFragment.create_table() Metadata.create_table() Annotation.create_table()
def create_db(): logging.info("Creating SQLite database %s..." % DB_FILE) # Define database db = SqliteDatabase(DB_FILE, threadlocals=True) # Connect to database db.connect() # Create tables Category.create_table() Feed.create_table() Post.create_table() Image.create_table() # Create default Unsorted category Category.create(name='Unsorted', comment='Uncategorised feeds', order=1) logging.info("Database created.")
import asyncio import peewee import telepot from InstagramAPI import InstagramAPI from decouple import config from telepot.aio.loop import MessageLoop from telepot.aio.delegate import pave_event_space, per_chat_id, create_open, include_callback_query_chat_id from models import Image from post_generator.post_generator_chat_handler import GeneratePostChatHandler try: Image.create_table() except peewee.OperationalError: print('Table already exists') INSTAGRAM_USERNAME = config('INSTAGRAM_USERNAME') INSTAGRAM_PASSWORD = config('INSTAGRAM_PASSWORD') def main(): instagram_bot = InstagramAPI(INSTAGRAM_USERNAME, INSTAGRAM_PASSWORD) instagram_bot.login() bot = telepot.aio.DelegatorBot(config('TOKEN'), [ include_callback_query_chat_id(pave_event_space())( per_chat_id(types=['private']), create_open, GeneratePostChatHandler, timeout=10), ]) loop = asyncio.get_event_loop()