diff --git a/backend/init_db.py b/backend/init_db.py index 7ee3ac7..84d5475 100644 --- a/backend/init_db.py +++ b/backend/init_db.py @@ -36,4 +36,31 @@ def populate_initial_data(conn): animals = [ ('Luna', 'Dog', 'Labrador Mix', 2, 'Playful and energetic', '/images/luna.jpg', 'Available'), ('Oliver', 'Cat', 'Tabby', 4, 'Independent but affectionate', '/images/oliver.jpg', 'Available'), - + ('Max', 'Dog', 'German Shepherd', 3, 'Loyal and intelligent', '/images/max.jpg', 'Available') + ] + + for animal in animals: + cursor.execute(''' + INSERT INTO animals (name, species, breed, age, personality, image_path, adoption_status) + VALUES (?, ?, ?, ?, ?, ?, ?) + ''', animal) + + conn.commit() + +def initialize_database(db_path): + """Initialize the database by creating tables and populating initial data.""" + conn = sqlite3.connect(db_path) + + # Create the database tables + create_database_tables(conn) + + # Populate the database with initial data + populate_initial_data(conn) + + conn.close() + print("Database initialized successfully with initial data.") + +if __name__ == "__main__": + # Path to the database + db_path = 'animal_shelter.db' + initialize_database(db_path)