randomPython/cohetes.py at master · programingfrik/randomPython · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
#!/usr/bin/python3

#
# Esto es un script para elaborar frases cohetes usando los
# maravillosos conocimientos obtenidos de esta página:
#
#   http://paginaspersonales.deusto.es/airibar/Personal/Textos_web/Textos_lenguaje.html
#

import random
import sys
import re

ffrases = "cohete1.txt"
fdiscurso = "cohete2.txt"
titulo = re.compile("^\\[\\w+\\]$")

def cargarColumnas(fichero):
    global titulo
    columnas = []
    temp = []
    with open(fichero, "r", errors="replace") as f:
        linea = f.readline()
        while linea != "":
            linea = linea.replace("\n", "").strip()
            if titulo.match(linea):
                if len(temp) > 0:
                    columnas.append(temp)
                    temp = []
            elif linea != "":
                temp.append(linea)
            linea = f.readline()
        if len(temp) > 0:
            columnas.append(temp)
    return columnas

def armarFrase(columnas):
    frase = ""
    for columna in columnas:
        if len(columna) > 0:
            frase += " " + random.choice(columna)
    return frase

def generarFrase():
    global ffrases
    columnas = cargarColumnas(ffrases)
    frase = armarFrase(columnas)
    print(frase)

def generarDiscurso(cant):
    global fdiscurso
    discurso = ""
    columnas = cargarColumnas(fdiscurso)
    for i in range(cant):
        discurso += " " + armarFrase(columnas)
    print(discurso)

def main():
    if ((len(sys.argv) == 2)
        and (sys.argv[1].upper() == "FRASE")):
        generarFrase()
    elif ((len(sys.argv) == 3)
          and (sys.argv[1].upper() == "DISCURSO")
          and (sys.argv[2].isdigit())):
        generarDiscurso(int(sys.argv[2]))
    else:
        print("Error parámetros equivocados!\n\nLo correcto "
              + "es:\n cohetes.py frase\n cohetes.py discurso N")

if __name__ == "__main__":
    main()