uvozi.py

import pandas as pd
from pandas import DataFrame

from Database import Repo
from model import *
from typing import Dict
from re import sub
import dataclasses



# Vse kar delamo z bazo se nahaja v razredu Repo.
repo = Repo()



def uvozi_recepte(pot):
    """
    Uvozimo csv z recepti v v bazo.
    Pri tem predpostavljamo, da imamo tabele že ustvarjene. 
    Vsako vrstico posebaj še obdelamo, da v bazi dobimo željeno strukturo. Uporabimo
    razred Repo za klic funkcij za uvoz v bazo.
    """

    df = pd.read_csv(pot)

    
    for row in df.itertuples():
        # row je seznam posamezne vrstice, ki se začne z indexom 1
        if row[4] != 'mins' or row[4] != 'hr' or row[4] != 'hrs':
            print(row)
        elif row[4] == 'hr':
            row[3] = 60 * int(row[3])
        else:
            continue

        if row[6] != 'mins' or row[6] != 'hr' or row[6] != 'hrs':
            print(row)
        elif row[6] == 'hr':
            row[5] = 60 * int(row[5])
        else:
            continue
        
        repo.dodaj_recept(
            Recepti(
                ime=row[2],
                st_porcij=row[7],
                cas_priprave=row[3],
                cas_kuhanja=row[5] 
            )
        )

def uvozi_kategorije(pot):
    df = pd.read_csv(pot)

    for row in df.itertuples():

        repo.dodaj_gen_brez_serial(
            Kategorije(
            id_recepta = row[1],
            kategorija = row[2]
            )
        )

def uvozi_kulinarike(pot):
    df = pd.read_csv(pot)

    for row in df.itertuples():

        repo.dodaj_gen_brez_serial(
            Kulinarike(
            id_recepta = row[1],
            kulinarika = row[2]
            )
        )

def uvozi_oznake(pot):
    df = pd.read_csv(pot)

    for row in df.itertuples():

        repo.dodaj_gen_brez_serial(
            Oznake(
            id_recepta = row[1],
            oznaka = row[2]
            )
        )

def uvozi_sestavine_receptov(pot):
    df = pd.read_csv(pot, error_bad_lines=False)

    for row in df.itertuples():

        repo.dodaj_sestavino(

            SestavineReceptov(
            id_recepta = row[1],
            kolicina = row[2],
            enota = row[3],
            sestavina = row[4],
            )
        )

def uvozi_postopke(pot):
    df = pd.read_csv(pot, error_bad_lines=False)

    for row in df.itertuples():
        repo.dodaj_postopek(
            Postopki(
            id_recepta = row[1],
            st_koraka = row[2],
            postopek = row[3],
            )
        )

def uvozi_nutrientske_vrednosti(pot):
    df = pd.read_csv(pot, error_bad_lines=False)

    for row in df.itertuples():
        repo.dodaj_gen_brez_serial(
            NutrientskeVrednosti(
            id_recepta = row[1],
            kalorije = row[8],
            proteini = row[11],
            ogljikovi_hidrati = row[9],
            mascobe = row[10]
            )
        )

def uvozi_sestavine(pot):
    df = pd.read_csv(pot, error_bad_lines=False)

    for row in df.itertuples():
        repo.dodaj_gen(
            Sestavine(
            ime = row[1],
            kalorije = row[2],
            proteini = row[5],
            ogljikovi_hidrati = row[6],
            mascobe = row[8]
            )
        )


def uvozi_csv(pot, ime):
    """
    Uvozimo csv v bazo brez večjih posegov v podatke.
    Ustvarimo pandasov DataFrame ter nato z generično metodo ustvarimo ter
    napolnimo tabelo v postgresql bazi.
    """
    df = pd.read_csv(pot, sep=";",skiprows=[0], encoding="Windows-1250")

    # Zamenjamo pomišljaje z prazno vrednostjo
    df.replace(to_replace="-", value="", inplace=True)

    # Naredimo tabelo z dodatnim serial primary key stolpcem
    repo.df_to_sql_create(df, ime, add_serial=True, use_camel_case=True)

    # uvozimo podatke v to isto tabelo
    repo.df_to_sql_insert(df, ime, use_camel_case=True)




# Primeri uporabe. Zakomentiraj določene vrstice, če jih ne želiš izvajat!
    

##pot = "obdelani-podatki/recepti.csv"
##pot = "obdelani-podatki/kategorije.csv"
##pot = "obdelani-podatki/sestavine-receptov.csv"
##pot = "obdelani-podatki/postopki.csv"
##pot = "obdelani-podatki/kulinarike.csv"
##pot = "obdelani-podatki/oznake.csv"
##pot = "obdelani-podatki/sestavine.csv"

# Uvozi csv s cenami izdelkov v ločene (in povezane) entitete
# Tabele morajo biti prej ustvarjene, da zadeva deluje

##uvozi_recepte(pot)
##uvozi_kategorije(pot)
##uvozi_sestavine_receptov(pot)
##uvozi_postopke(pot)
##uvozi_kulinarike(pot)
##uvozi_oznake(pot)
##uvozi_nutrientske_vrednosti(pot)
##uvozi_sestavine(pot)


# Uvozi csv s cenami, le da tokar uvozi le eno tabelo, ki jo
# predhodno še ustvari, če ne obstaja.

#uvozi_csv(pot, "NovaTabela")



## S pomočjo generične metode dobimo seznam izdelkov in kategorij
## Privzete nastavi
#
## Dobimo prvih 100 izdelkov
#recepti = repo.dobi_gen(Recept, skip=0, take=100)
#
#t = repo.dobi_gen(ReceptPosSes)
#
## Dobimo prvih 10 kategorij
#kategorije = repo.dobi_gen(Kategorija)
#
## Dodamo novo kategorijo

#nova_kategorija = Kategorije(
#    kategorija="Nova kategorija"
#    )
#
#repo.dodaj_gen(nova_kategorija)
#
## vrednost nova_kategorija.id je sedaj določen na podlagi
## serial vrednosti iz baze in jo lahko uporabimo naprej.
#
#
## Dodamo nov recept v to kategorijo
#novi_recept = Recept(
#    ime = 'Novi recept',
#    kategorija=nova_kategorija.id
#)
#repo.dodaj_gen(novi_recept)
#
#
## Dobimo recept z idjem 832
#recept = repo.dobi_gen_id(Recept, 832)