Modernize the code a bit:

use re module
   use .split() string method

Doesn't use 'for line in sys.stdin'; that ends up changing its interactive
behaviour.
This commit is contained in:
Andrew M. Kuchling 2003-04-24 17:17:56 +00:00
parent 946c53ed7f
commit b7878d09e5
1 changed files with 6 additions and 5 deletions

View File

@ -23,15 +23,16 @@
# - Outputs the sorted fields with exactly one space between them
# - Handles blank input lines correctly
import regex
import re
import string
import sys
def main():
prog = regex.compile('^\(.*\)=\([-+]?[0-9]+\)')
prog = re.compile('^(.*)=([-+]?[0-9]+)')
def makekey(item, prog=prog):
if prog.match(item) >= 0:
var, num = prog.group(1, 2)
match = prog.match(item)
if match:
var, num = match.group(1, 2)
return string.atoi(num), var
else:
# Bad input -- pretend it's a var with value 0
@ -40,7 +41,7 @@ def main():
line = sys.stdin.readline()
if not line:
break
items = string.split(line)
items = line.split()
items = map(makekey, items)
items.sort()
for num, var in items: