Prueba

main.py

#! usr/bin/python
# -*- coding: utf-8 -*-
 
#Agenda con base de datos Sqlite3
#www.pythondiario.com
#Autor: Diego Caraballo
 
#Modulos importados
import sqlite3
import time 
import os
 
#Conexion con Base de Datos Sqlite3
con = sqlite3.connect("agenda.db")
cursor = con.cursor()
#Comprueba si la tabla existe, en caso de no existir la crea
cursor.execute("""CREATE TABLE IF NOT EXISTS amigos (nombre TEXT, direccion TEXT, telefono TEXT, correo TEXT, tipo_amigo TEXT)""")
 
cursor.close()
 
#Declaracion de las funciones
 
def limpiar():
 
	"""Limpia la pantalla"""
 
	if os.name == "posix":
		os.system("clear")
	elif os.name == ("ce", "nt", "dos"):
		os.system("cls")
 
def agregar():
  
	"""Agrega un nuevo contacto a la Agenda"""
 
	print ("Agregar contacto")
	print ("----------------")
	print ("")
 
	con = sqlite3.connect("agenda.db")
	cursor = con.cursor()
 
	nombre = input("Nombre: ")
	direccion = input("Direccion: ")
	telefono = input("Telefono: ")
	correo = input("Correo: ")
	tipo = input("Tipo Amigo: ")

 
	cursor.execute("insert into amigos (nombre, direccion, telefono, correo, tipo_amigo) values ('%s','%s','%s','%s','%s')"%(nombre,direccion,telefono,correo,tipo))
 
	con.commit()
 
	print ("Los datos fueron agregados correctamente")
 
	cursor.close()
	time.sleep(2)
	main()
 
def ver():
  
	"""Devuelve todos los contactos de la agenda"""
 
	print ("Lista de contactos")
	print ("------------------")
	print ("")
 
	con = sqlite3.connect("agenda.db")
	cursor = con.cursor()
 
	cursor.execute("SELECT * FROM amigos")
	resultado = cursor.fetchall()
 
	for i in resultado:
		print ("Nombre: %s Direccion: %s Telefono: %s E-mail: %s Tipo Amigo: %s \n" % (i[0],i[1],i[2],i[3],i[4]))
 
	cursor.close()
 
	print ("")
	input("Presione una tecla para continuar...")
 
	main()
 
def buscar():
  
	"""Buscar un contacto en la agenda"""
 
	print ("Buscar contacto")
	print ("---------------")
	print ("")
 
	con = sqlite3.connect("agenda.db")
	cursor = con.cursor()
 
	buscar = input("Nombre del contacto: ")
 
	cursor.execute ("SELECT * FROM amigos WHERE nombre = '%s'" %(buscar))
 
	x = cursor.fetchall()
 
	print ("")
 
	for i in x:
		print ("Nombre:", i[0])
		print ("Direccion:", i[1])
		print ("Telefono:", i[2])
		print ("Correo:", i[3])
		print ("Tipo Amigo:", i[4])
		print ("")
 
	cursor.close()
 
	print ("")
	input("Presione una tecla para continuar...")
 
	main()
 
def eliminar():
  
	"""Elimina un contacto de la Agenda"""
 
	print ("Eliminar contacto")
	print ("-----------------")
	print ("")
 
	con = sqlite3.connect("agenda.db")
	cursor = con.cursor()
 
	eliminar = input ("Nombre de contacto a eliminar: ")
 
	cursor.execute("DELETE FROM amigos WHERE nombre='%s'"%(eliminar))
 
	con.commit()
 
	cursor.close()
 
	print ("Contacto eliminado correctamente...")
	input()
	main()

# def actualizar():
      
# 	"""Actualizar un contacto a la Agenda"""
 
# 	print ("Actualizar contacto")
# 	print ("----------------")
# 	print ("")
 
#     option = input('Que quieres actualizar: ')

#     if option = '1':
#         contacto = input('Numero de telefono del contacto a editar: ')
#         telefono_actualizar = input('Escribe el nuevo telefono del contacto: ')
#         con = sqlite3.connect("agenda.db")
# 	      cursor = con.cursor()
# 	      cursor.execute("update datos set telefono='%s' where nombre='%s'"%(contacto,telefono_actualizar))
#         con.commit()
#         print ("Los datos fueron actualizados correctamente")
#         cursor.close()
#         time.sleep(2)
#         main()
 
def main():
 
	"""Funcion principal de la Agenda"""
 
	limpiar()
 
	print ("-----------------------------------------")
	print ("----------AGENDA DE AMIGOS-----------------")
	print ("-----------------------------------------")
	print ("""
	[1] Agregar Contacto
	[2] Ver Contactos
	[3] Buscar Contacto
	[4] Eliminar Contacto
	[0] Salir
	""")
 
	opcion = input("Ingresa una opción -> ")
 
	if opcion != "1" and opcion != "2" and opcion != "3" and opcion != "4" and opcion != "5" and opcion != "0":
		print ("Opcion incorrecta")
		input()
		main()
	elif opcion == "1":
		limpiar()
		agregar()
	elif opcion == "2":
		limpiar()
		ver()
	elif opcion == "3":
		limpiar()
		buscar()
	elif opcion == "4":
		limpiar()
		eliminar()
	elif  opcion == "0":
		print ("")
		print ("Cerrando aplicacion")
		print ("")
		print ("")
		# time.sleep(3)
		exit()
  
 
main()