File tree Expand file tree Collapse file tree 1 file changed +10
-8
lines changed Expand file tree Collapse file tree 1 file changed +10
-8
lines changed Original file line number Diff line number Diff line change @@ -31,15 +31,16 @@ def _validate(url):
31
31
raise argparse .ArgumentTypeError (str (e ))
32
32
return _url
33
33
34
- def validate_url ( url ):
35
- if os .path .exists (url ):
36
- with open ( url , 'r' ) as opener :
37
- content = opener . read ()
38
- else :
39
- content = url
34
+ def _try_read ( path ):
35
+ if not os .path .exists (path ):
36
+ return None
37
+
38
+ with open ( path , 'r' ) as o :
39
+ return o . read ()
40
40
41
+ def validate_url (url ):
41
42
urls = []
42
- for _url in content .splitlines ():
43
+ for _url in url .splitlines ():
43
44
if not _url :
44
45
continue
45
46
@@ -245,7 +246,8 @@ def validate(parser, args):
245
246
elif not os .path .exists (file_path ):
246
247
parser .error (f"File \" { file_path } \" is not exist" )
247
248
else :
248
- file_path = urls
249
+ file_content = _try_read (urls )
250
+ file_path = file_content if file_content is not None else urls
249
251
try :
250
252
args .URL = validate_url (file_path )
251
253
except argparse .ArgumentTypeError as e :
You can’t perform that action at this time.
0 commit comments