From d40a78ef73c793cc1d23118a6a1c6a478d930dfd Mon Sep 17 00:00:00 2001 From: Zach DeCook Date: Wed, 10 Feb 2021 07:33:59 -0500 Subject: zachwalk: also read from stdin --- zachwalk.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'zachwalk.py') diff --git a/zachwalk.py b/zachwalk.py index a7d1a8d..aee0458 100755 --- a/zachwalk.py +++ b/zachwalk.py @@ -4,6 +4,7 @@ import sys import socket import ssl +import fileinput from dateutil.parser import parse DEFAULT = '1970-01-01' @@ -40,9 +41,8 @@ def gnd(fp): pass return DEFAULT -def main(argv): - with open(argv[1]) as f: - for line in f: +def main(): + for line in fileinput.input(): #stdin or file from argv if line[0:2] == '=>': # plz don't use multiple spaces. url = line.split(' ')[1] @@ -54,4 +54,4 @@ def getdesc(line): return ' '.join(line.split(' ')[2:]) if __name__ == '__main__': - main(sys.argv) + main() -- cgit 1.4.1