Nivel 35 - Python con MySQL - Reto 7 - Consultas interactivas

[@santysalgado4387]

#CÓDIGO FUNCIONANDO INICIO
import mysql.connector
from mysql.connector import Error
import os
def conectarse_bd(usuario, contrasenia):
'''Conectarse a una base de datos'''
conexion = None
try:
conexion = mysql.connector.connect(
host = "localhost",
user = usuario,
passwd = contrasenia,
database = "world")
if conexion:
print("Conexión realizada correctamente.")
except Error as e:
print("Ha ocurrido el siguiente error:")
print(e)
return conexion
def desconectarse(conexion):
'''Cerrar una conexoión a un servidor MySQL'''
if conexion:
conexion.close()
print("Conexión finalizada correctamente.")
def consulta(conexion):
'''Crea consulta y devuelve el resultado'''
resultado = None
try:
consulta = "SELECT ID, Name FROM city LIMIT 10"
cursor = conexion.cursor(buffered=True)
cursor.execute(consulta)
resultado = cursor.fetchall()
except Error as e:
print("Ha ocurrido un error")
print(e)
finally:
cursor.close()
return resultado
#Flujo del programa
usuario = os.environ.get("usuario_mysql")
contrasenia = os.environ.get("contrasenia_mysql")
conexion = conectarse_bd(usuario, contrasenia)
if conexion:
resultado = consulta(conexion)
desconectarse(conexion)
if resultado is not None:
for fila in resultado:
print(fila)
#CÓDIGO FUNCIONANDO FINAL