db2

#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
Script para crear una base de datos de un restaurante con dos tablas:
 - categorias de platos (entrada, principal, postres)
 - platos (empanadas, milanesa c/pure, asado, flan c/crema)
"""

import sqlite3

conn = sqlite3.connect("restaurante.db")

cursor = conn.cursor()

# creamos la tabla categoria
query = 'CREATE TABLE categoria(\
        id INTEGER PRIMARY KEY AUTOINCREMENT,\
        nombre TEXT UNIQUE NOT NULL\
        )'
cursor.execute(query)

# creamos la tabla platos
query = 'CREATE TABLE platos(\
        id INTEGER PRIMARY KEY AUTOINCREMENT,\
        nombre TEXT UNIQUE NOT NULL,\
        id_categoria INTEGER NOT NULL,\
        FOREIGN KEY(id_categoria) REFERENCES categoria(id)\
        )'
cursor.execute(query)