|
1 | 1 | import os
|
2 | 2 | from tkinter import filedialog, messagebox
|
| 3 | +from copiar import copiar_origem_destino |
3 | 4 |
|
4 | 5 |
|
5 |
| -continuar = messagebox.askyesno(title='Adicionar linha', |
| 6 | +def deve_continuar(): |
| 7 | + return messagebox.askyesno(title='Adicionar linha', |
6 | 8 | message='Gostaria de selecionar um par origem e destino?')
|
7 | 9 |
|
8 |
| -with open(os.path.join('input', 'diretorios.csv'), 'w') as dircsv: |
9 |
| - # cabeçalho do CSV |
10 |
| - dircsv.write('origem,destino\n') |
11 |
| - while continuar: |
12 |
| - origem = filedialog.askdirectory() |
13 |
| - destino = filedialog.askdirectory() |
14 |
| - dircsv.write(f'{origem},{destino}\n') |
15 |
| - continuar = messagebox.askyesno(title='Adicionar linha', |
16 |
| - message='Gostaria de selecionar mais um par origem e destino?') |
| 10 | +def construir_origem_destino(nomearquivo, modo='w'): |
| 11 | + with open(nomearquivo, modo) as dircsv: |
| 12 | + if modo == 'w': |
| 13 | + # cabeçalho do CSV |
| 14 | + dircsv.write('origem,destino\n') |
| 15 | + while deve_continuar(): |
| 16 | + origem = filedialog.askdirectory() |
| 17 | + destino = filedialog.askdirectory() |
| 18 | + dircsv.write(f'{origem},{destino}\n') |
| 19 | + |
| 20 | +if __name__ == "__main__": |
| 21 | + arquivo = os.path.join('input', 'diretorios.csv') |
| 22 | + if os.path.exists(arquivo): |
| 23 | + construir_origem_destino(arquivo, 'a') |
| 24 | + else: |
| 25 | + construir_origem_destino(arquivo) |
| 26 | + copiar_origem_destino(arquivo) |
17 | 27 |
|
0 commit comments