Skip to content
Snippets Groups Projects
table_check.py 912 B
Newer Older
  • Learn to ignore specific revisions
  • import sqlite3
    
    def table_check(): #This function is used on start to make a new Database if not already exists.
        create_table_data = """
            CREATE TABLE WEB_URL(
                LONG_URL TEXT NOT NULL,
                SHORT_URL TEXT NOT NULL,
                USERNAME TEXT
            );
            """
        create_table_analytics = """
            CREATE TABLE ANALYTICS(
                SHORT_URL TEXT NOT NULL,
                CALLS INT DEFAULT 1
            )
        """
        with sqlite3.connect('db/urls.db') as conn:
            cursor = conn.cursor()
            try: #Try making the database structure, if fails Database was already created.
                cursor.execute(create_table_data)
            except sqlite3.OperationalError:
                pass
            try: #Try making the database structure, if fails Database was already created.
                cursor.execute(create_table_analytics)
            except sqlite3.OperationalError:
                pass