Skip to content

Commit

Permalink
config
Browse files Browse the repository at this point in the history
  • Loading branch information
ocworld committed Apr 7, 2019
1 parent 85d4e9b commit 6e8a4de
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 3 deletions.
10 changes: 8 additions & 2 deletions nart/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,9 +25,15 @@ def main():

args = parser.parse_args()

config = Config.from_file(args.file, encoding='utf-8') if args.file else None
config = Config.from_file(args.file, encoding='utf-8') if args.file else Config()
config.use_stdoutwriter = config.use_stdoutwriter if config.use_stdoutwriter else False
config.use_csvwriter = config.use_csvwriter if config.use_csvwriter else False
config.start_on_time = config.start_on_time if config.start_on_time else False
config.interval = config.interval if config.interval else 1
config.timeunit = config.timeunit if config.timeunit else Timeunit.MIN

config.use_stdoutwriter = True if args.verbose else config.use_stdoutwriter
config.use_csvwriter = True if args.verbose else config.use_csvwriter
config.use_csvwriter = True if args.csvout else config.use_csvwriter
config.start_on_time = True if args.startontime else config.start_on_time
config.interval = args.interval if args.interval else config.interval
config.timeunit = Timeunit.from_str(args.timeunit) if args.timeunit else config.timeunit
Expand Down
3 changes: 2 additions & 1 deletion nart/model/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,8 @@


class Config:
def __init__(self, timeunit, interval, start_on_time, use_stdoutwriter, use_csvwriter, csvwriter_outpath):
def __init__(self, timeunit=None, interval=None, start_on_time=None, use_stdoutwriter=None, use_csvwriter=None,
csvwriter_outpath=None):
self.timeunit = timeunit
self.interval = interval
self.start_on_time = start_on_time
Expand Down

0 comments on commit 6e8a4de

Please sign in to comment.