Random Table Creation, Data Insertion, and Querying

  • Share this:

Code introduction


This code snippet creates a randomly named table, inserts some random data, and queries the data from this table.


Technology Stack : sqlite3

Code Type : Database operation

Code Difficulty : Intermediate


                
                    
import sqlite3
import random

def generate_random_table_name():
    letters = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
    return ''.join(random.choice(letters) for i in range(5))

def create_random_table(db_connection):
    cursor = db_connection.cursor()
    table_name = generate_random_table_name()
    columns = ['id INTEGER PRIMARY KEY', 'name TEXT', f'{generate_random_table_name()} TEXT']
    create_table_query = f"CREATE TABLE IF NOT EXISTS {table_name} ({', '.join(columns)})"
    cursor.execute(create_table_query)
    db_connection.commit()

def insert_random_data(db_connection):
    cursor = db_connection.cursor()
    table_name = generate_random_table_name()
    insert_query = f"INSERT INTO {table_name} (name, {table_name}) VALUES (?, ?)"
    data = ('John Doe', 'Some Random Text')
    cursor.execute(insert_query, data)
    db_connection.commit()

def query_random_data(db_connection):
    cursor = db_connection.cursor()
    table_name = generate_random_table_name()
    query = f"SELECT * FROM {table_name}"
    cursor.execute(query)
    return cursor.fetchall()                
              
Tags: