Refactor as package with subpackages

This makes it cleaner for introducing checks, fixes, tests, docs,
and tests in the future. Currently can be run like this:

  python -m csv_metadata_quality

CSV input and output paths are still hard coded.

See: https://dev.to/codemouse92/dead-simple-python-project-structure-and-imports-38c6
This commit is contained in:
Alan Orth 2019-07-26 22:11:10 +03:00
parent ef5b8f7244
commit 232d28e13e
Signed by: alanorth
GPG Key ID: 0FB860CC9C45B1B9
4 changed files with 27 additions and 18 deletions

View File

View File

@ -0,0 +1,4 @@
from csv_metadata_quality import app
if __name__ == '__main__':
app.run()

View File

@ -0,0 +1,17 @@
import csv_metadata_quality.fix as fix
import pandas as pd
def run():
# Read all fields as strings so dates don't get converted from 1998 to 1998.0
#df = pd.read_csv('/home/aorth/Downloads/2019-07-26-Bioversity-Migration.csv', dtype=str)
#df = pd.read_csv('/tmp/quality.csv', dtype=str)
df = pd.read_csv('/tmp/omg.csv', dtype=str)
# Fix whitespace in all columns
for column in df.columns.values.tolist():
print(f'DEBUG: {column}')
df[column] = df[column].apply(fix.whitespace)
# Write
df.to_csv('/tmp/omg.fixed.csv', index=False)

View File

@ -1,14 +1,16 @@
#!/usr/bin/env python3
import pandas as pd
import re
def fix_whitespace(field):
def alan():
print('Alan')
def whitespace(field):
"""Fix whitespace issues.
Return string with leading, trailing, and consecutive whitespace trimmed.
"""
import re
# Skip fields with missing values
if pd.isna(field):
@ -38,17 +40,3 @@ def fix_whitespace(field):
return new_field
# Read all fields as strings so dates don't get converted from 1998 to 1998.0
#df = pd.read_csv('/home/aorth/Downloads/2019-07-26-Bioversity-Migration.csv', dtype=str)
#df = pd.read_csv('/tmp/quality.csv', dtype=str)
df = pd.read_csv('/tmp/omg.csv', dtype=str)
# Fix whitespace in all columns
for column in df.columns.values.tolist():
print(f'DEBUG: {column}')
df[column] = df[column].apply(fix_whitespace)
# Write
df.to_csv('/tmp/omg.fixed.csv', index=False)