obterdatos.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import csv
import logging
from random import randrange
from config import *

def descapitalice(concello):
    return(f"{concello[0].lower()}{concello[1:]}")

def dato():
    lista = []
    with open(DATAFILE, "r", newline="\n") as ficheiro:
        reader = csv.reader(ficheiro, delimiter=";")
        lista = list(reader)
        try:
            datos_concello = lista[randrange(len(lista))]
            lista.remove(datos_concello)
        except ValueError:
            logging.warning("Non quedan concellos por publicar")
            return

    with open(DATAFILE, "w", newline="\n") as ficheiro:
        writer = csv.writer(ficheiro, delimiter=";")
        for linha in lista:
            writer.writerow(linha)

    with open(USEDFILE,"a", newline="\n") as ficheiro:
        writer = csv.writer(ficheiro,delimiter=";")
        writer.writerow(datos_concello)

    if datos_concello[1].startswith("A ") or datos_concello[1].startswith("As ") or datos_concello[1].startswith("O ") or datos_concello[1].startswith("Os "):
        concello_dc = descapitalice(datos_concello[1])
        concello = f"d{concello_dc}"
    else:
        concello = f"de {datos_concello[1]}"

    vacas = datos_concello[2]
    habitantes = datos_concello[3]
    vacas_x_habitante = (str(round(float(vacas) / float(habitantes),4))).replace(".",",")

    if int(vacas) == int(habitantes):
        return f"No Concello {concello} hai un total de {vacas} vacas e {habitantes} habitantes. Tantas vacas como habitantes"
    elif int(vacas) == 0:
        return f"No Concello {concello} non hai ningunha vaca. Ten un total de {habitantes} habitantes."
    elif int(vacas) == 1:
        return f"No Concello {concello} hai un total de {vacas} vaca e {habitantes} habitantes. {vacas_x_habitante} vacas por habitante."
    else:
        return f"No Concello {concello} hai un total de {vacas} vacas e {habitantes} habitantes. {vacas_x_habitante} vacas por habitante."


if __name__ == '__main__':
    print(dato())