winzer/elwig-misc
3
1
Fork 0
Code Pull Requests Activity

Fix csv change

Browse Source
This commit is contained in:
Lorenz Stechauner
2023-05-05 00:03:25 +02:00
parent 3b7a28cc23
commit df3066ebfe
2 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
wgmaster/export.py
View File

@ -139,9 +139,10 @@ def main() -> None:
cols = [t[0] for t in cur.description] cols = [t[0] for t in cur.description]
with utils.csv_open(f'{args.output}/{t_name}.csv') as f: with utils.csv_open(f'{args.output}/{t_name}.csv') as f:
f.header(cols) f.header(*cols)
for row in cur: for row in cur:
f.row((utils.convert_value(val, table=t_name, column=col) for col, val in zip(cols, row)), raw=True) values = (utils.convert_value(val, table=t_name, column=col) for col, val in zip(cols, row))
f.row(*values, raw=True)
print(f'Exported {t_name} successfully!', flush=True) print(f'Exported {t_name} successfully!', flush=True)
finally: finally:

4
wgmaster/utils.py
View File

@ -56,8 +56,8 @@ class CsvFile:
def __init__(self, file: TextIO): def __init__(self, file: TextIO):
self.file = file self.file = file
self.writer = csv.writer(self.file, doublequote=False, quoting=csv.QUOTE_NONE, escapechar=None) self.writer = csv.writer(self.file, doublequote=False, quoting=csv.QUOTE_NONE, escapechar='\\', quotechar=None)
self.reader = csv.reader(self.file, doublequote=False, quoting=csv.QUOTE_NONE, escapechar=None) self.reader = csv.reader(self.file, doublequote=False, quoting=csv.QUOTE_NONE, escapechar='\\', quotechar=None)
def __enter__(self) -> CsvFile: def __enter__(self) -> CsvFile:
return self return self