def main(nzb_files, options):
if not nzb_files:
parser.print_help()
return
settings = config.read_config(options.config_file)
if not settings:
print '%s has no settings!' % options.config_file
return
if options.debug:
settings['debug'] = options.debug
print settings
if options.pattern:
settings['skip_regex'] = options.pattern
settings['invert'] = True
elif options.par2:
settings['skip_regex'] = ['\.par2']
settings['invert'] = True
nzbs = [nzb for nzb in helper.get_nzb_file(nzb_files)]
for nzb in nzbs:
nzb_name = os.path.split(nzb)[1]
new_dir = helper.get_download_path(settings.get('download_dir'), nzb_name)
settings['download_path'] = new_dir
if not os.path.exists(new_dir):
if settings.get('debug'):
print 'made new dir %s ' % new_dir
os.mkdir(new_dir)
if settings.get('debug'):
print settings['download_path']
download.start(nzb, settings)
def test_get_download_path(self):
self.assertEqual(helper.get_download_path('', 'test.nzb'), 'test')
self.assertEqual(helper.get_download_path('dir/', 'test.nzb'), 'dir/test')
