mirror of
https://github.com/coleam00/ai-agents-masterclass.git
synced 2025-11-29 00:23:14 +00:00
50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
import sqlite3
|
|
|
|
"""
|
|
This script is used to create a SQLlite database, add tables
|
|
to it, and insert mock data, all for the OpenAI Swarm demonstration
|
|
with the other Python script in this directory.
|
|
|
|
Simply run this script with the command:
|
|
|
|
python load_sql_data.py
|
|
|
|
And then you will have a database loaded and ready to use
|
|
with the agent swarm!
|
|
"""
|
|
|
|
def execute_sql_script(cursor, script_file):
|
|
# Opens the .sql file given as script_file
|
|
with open(script_file, 'r') as sql_file:
|
|
sql_script = sql_file.read()
|
|
|
|
# Gets all the sql commands and executes them one at a time
|
|
statements = sql_script.split(';')
|
|
for statement in statements:
|
|
if statement.strip():
|
|
cursor.execute(statement)
|
|
|
|
def main():
|
|
# Connect to the database
|
|
conn = sqlite3.connect('rss-feed-database.db')
|
|
cursor = conn.cursor()
|
|
|
|
# Execute SQL script to create tables for the AI RSS Feed system
|
|
execute_sql_script(cursor, 'ai-news-complete-tables.sql')
|
|
conn.commit()
|
|
|
|
# Execute SQL script to insert mock data for the AI RSS Feed system
|
|
execute_sql_script(cursor, 'ai-news-complete-mock-data.sql')
|
|
conn.commit()
|
|
|
|
# Query table to make sure things are looking good
|
|
cursor.execute("SELECT * FROM rss_feeds")
|
|
feeds = cursor.fetchall()
|
|
for feed in feeds:
|
|
print(feed)
|
|
|
|
# Close the connection
|
|
conn.close()
|
|
|
|
if __name__ == "__main__":
|
|
main() |