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

import MySQLdb
from random import randint


DB_HOST='localhost'
DB_USER='root'
DB_PASS='pablete123'
DB_NAME='SanaBien'
PATH_FILE='productos.txt' #Ruta del Archivo de datos
FIRST_LINE=2 #Primera Linea del archivo con datos utiles
TABLA="Laboratorio" #Tabla a modificar el Auto Incremento

datos = [DB_HOST, DB_USER, DB_PASS, DB_NAME]
conn = MySQLdb.connect(*datos)
cursor = conn.cursor()
conn.commit()


archivo=open(PATH_FILE, 'r');
i=0;
cursor.execute("ALTER TABLE "+ TABLA+" AUTO_INCREMENT = 1")

a=0;
for linea in archivo.readlines():
    if(a>FIRST_LINE):
        campos=linea.split('\t')
        #campos[1]=campos[1].replace(' ', '')
        #campos[1]=campos[1].replace('-', '')
        campos[1]=campos[1].rstrip('\n')
        campos[1]=campos[1].replace('\r', '')
        print(campos[1])
        precio=randint(100,10000)
        cantidad=randint(1, 1000)

        QUERY="select codigoLaboratorio from Laboratorio where nombre ='"+campos[1]+"'"
        print QUERY
        codigoLaboratorio=cursor.execute(QUERY)
        tupla=cursor.fetchone()
        print tupla
        print len(tupla)
        print tupla[0]
        QUERY="INSERT INTO Medicamento (nombreMedicamento, precio, cantidad, codigoLaboratorio) values ('%s', '%d', '%d', '%d')"% (campos[0], precio, cantidad, tupla[0])
        print QUERY
        cursor.execute(QUERY)
        conn.commit()


    a=a+1

archivo.close()