ArgParse
This commit is contained in:
parent
dd3fa2f9bc
commit
d3ef2d508a
5 changed files with 72 additions and 17 deletions
2
.gitignore
vendored
2
.gitignore
vendored
|
|
@ -37,7 +37,7 @@ MANIFEST
|
|||
pip-log.txt
|
||||
pip-delete-this-directory.txt
|
||||
|
||||
# Unit test / coverage reports
|
||||
# Unit examples / coverage reports
|
||||
htmlcov/
|
||||
.tox/
|
||||
.nox/
|
||||
|
|
|
|||
0
README.md
Normal file
0
README.md
Normal file
71
csv_to_sql_file.py
Normal file
71
csv_to_sql_file.py
Normal file
|
|
@ -0,0 +1,71 @@
|
|||
import logging, argparse, sys, os
|
||||
import csv
|
||||
import re
|
||||
|
||||
|
||||
# python csv_to_sql_file.py -file file.csv -table sql_table -output output.sql -headers -auto_type -date_format
|
||||
# varchar integer float date bool
|
||||
|
||||
DEFAULT_DATE_FORMAT = "DD/MM/YYYY"
|
||||
|
||||
|
||||
def csv_to_file(csv_file, table, output_file, headers, auto_type, date_format):
|
||||
print(csv_file, table, output_file, headers, auto_type, date_format)
|
||||
|
||||
|
||||
def main():
|
||||
args = set_argparse()
|
||||
|
||||
CSV_FILE = args.file
|
||||
TABLE = args.table
|
||||
OUTPUT_FILE = args.output
|
||||
HEADERS = args.headers
|
||||
AUTO_TYPE = args.auto_type
|
||||
DATE_FORMAT = args.date_format
|
||||
|
||||
logging.info('CSV file : %s', CSV_FILE)
|
||||
logging.info('Table : %s', TABLE)
|
||||
logging.info('Output file : %s', OUTPUT_FILE)
|
||||
logging.info('Headers : %s', HEADERS)
|
||||
logging.info('Auto type : %s', AUTO_TYPE)
|
||||
logging.info('Date Format : %s', DATE_FORMAT)
|
||||
|
||||
csv_to_file(CSV_FILE, TABLE, OUTPUT_FILE, HEADERS, AUTO_TYPE, DATE_FORMAT)
|
||||
exit(0)
|
||||
|
||||
|
||||
def set_argparse():
|
||||
parser = argparse.ArgumentParser(description='')
|
||||
parser.action_groups.pop()
|
||||
required = parser.add_argument_group('required arguments')
|
||||
optional = parser.add_argument_group('optional arguments')
|
||||
required.add_argument('-f', '--file', help=' (default: None)', default=None, required=True)
|
||||
|
||||
optional.add_argument('-t', '--table', help=' (default: filename)', default=None)
|
||||
optional.add_argument('-o', '--output', help=' (default: output.sql)', default='output.sql')
|
||||
optional.add_argument('-h', '--headers', help=' (default: True)', default=True)
|
||||
optional.add_argument('-at', '--auto_type', help=' (default: True)', default=True)
|
||||
optional.add_argument('-df', '--date_format', help=f' (default: {DEFAULT_DATE_FORMAT})',
|
||||
default=DEFAULT_DATE_FORMAT)
|
||||
|
||||
optional.add_argument('-i', '--info', help='Info mode (default: True)', default=True, action='store_false')
|
||||
optional.add_argument('-d', '--debug', help='Debug mode (default: False)', default=False, action='store_true')
|
||||
|
||||
try:
|
||||
args = parser.parse_args()
|
||||
except argparse.ArgumentError as error:
|
||||
logging.error('Catching an argumentError {}'.format(error))
|
||||
sys.exit('Catching an argumentError {}'.format(error))
|
||||
|
||||
if args.debug:
|
||||
logging.basicConfig(stream=sys.stderr, level=logging.DEBUG)
|
||||
elif args.info:
|
||||
logging.basicConfig(stream=sys.stderr, level=logging.INFO)
|
||||
else:
|
||||
logging.basicConfig(stream=sys.stderr, level=logging.ERROR)
|
||||
|
||||
return args
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
16
main.py
16
main.py
|
|
@ -1,16 +0,0 @@
|
|||
# This is a sample Python script.
|
||||
|
||||
# Press Maj+F10 to execute it or replace it with your code.
|
||||
# Press Double Shift to search everywhere for classes, files, tool windows, actions, and settings.
|
||||
|
||||
|
||||
def print_hi(name):
|
||||
# Use a breakpoint in the code line below to debug your script.
|
||||
print(f'Hi, {name}') # Press Ctrl+F8 to toggle the breakpoint.
|
||||
|
||||
|
||||
# Press the green button in the gutter to run the script.
|
||||
if __name__ == '__main__':
|
||||
print_hi('PyCharm')
|
||||
|
||||
# See PyCharm help at https://www.jetbrains.com/help/pycharm/
|
||||
0
test_csv_to_sql_file.py
Normal file
0
test_csv_to_sql_file.py
Normal file
Loading…
Add table
Add a link
Reference in a new issue