-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathgenerador_tablas.py
47 lines (33 loc) · 959 Bytes
/
generador_tablas.py
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
import sys
import os
# Ojo: nombre de script es primer argumento
if len(sys.argv) == 2:
filename = str(sys.argv[1])
else:
print "El argumente debe ser el nombre del archivo"
sys.exit()
if os.path.isfile('./' + filename):
extension = filename[-4:len(filename)]
name = filename[0:len(filename)-4]
csv = open(filename, 'r')
else:
print "Nombre de archivo no existe"
sys.exit()
if extension == '.tsv':
cols = csv.readline().split('\t')
elif extension == '.csv':
cols = csv.readline().split(',')
out = open(name + '.txt', 'w+')
print >>out, 'CREATE TABLE ' + name + ' ('
for c in cols[0:len(cols)-2]:
# Si hay espacios ponemos corchetes alrededor
if c.find(' ') != -1:
c = '[' + c + ']'
print >>out, c + " STRING,"
## La ultima columna
lastcol = cols[len(cols)-2]
if lastcol.find(' ') != -1:
lastcol = '[' + lastcol + ']'
print >>out, lastcol + " STRING) ;"
csv.close()
out.close()