Skip to content

Commit

Permalink
Add updated version of the main wrapper function using argparse
Browse files Browse the repository at this point in the history
  • Loading branch information
witsch committed Sep 24, 2011
1 parent 4574012 commit 697bbcf
Show file tree
Hide file tree
Showing 3 changed files with 51 additions and 7 deletions.
4 changes: 4 additions & 0 deletions config.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
from PythonTidy import VERSION

version = VERSION
summary = 'Cleans up, regularizes, and reformats the text of Python scripts.'
35 changes: 35 additions & 0 deletions runner.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,35 @@
from argparse import ArgumentParser, FileType
from sys import stdin, stdout, exit
from config import version, summary
from PythonTidy import tidy_up
from PythonTidyWrapper import Config


parser = ArgumentParser(description=summary)
parser.add_argument('input', nargs='?', type=FileType('r'), default=stdin,
help='specify input file instead of `stdin`')
parser.add_argument('output', nargs='?', type=FileType('w'), default=stdout,
help='specify output file instead of `stdout`')
parser.add_argument('-v', '--version', action='version',
version='%%(prog)s %s' % version)
parser.add_argument('-d', '--dump', action='store_true',
help='dump default configuration parameters')
parser.add_argument('-c', '--config', default=None,
help='read configuration parameters from file')


def main():
args = parser.parse_args()
if args.dump:
config = Config()
config.from_pythontidy_namespace()
config.write(file=stdout)
exit()
if args.config:
config = Config(file=args.config)
config.to_pythontidy_namespace()
tidy_up(args.input, args.output)


if __name__ == "__main__":
main()
19 changes: 12 additions & 7 deletions setup.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,14 @@
from setuptools import setup
from PythonTidy import VERSION as version
from config import version, summary


install_requires = ['setuptools']
try:
import argparse
argparse # make pyflakes happy
except ImportError:
install_requires.append('argparse')

summary = 'Cleans up, regularizes, and reformats the text of Python scripts.'

setup(name='PythonTidy',
version=version,
Expand All @@ -21,12 +28,10 @@
author_email='[email protected]',
url='http://pypi.python.org/pypi/PythonTidy',
license='GPL version 2',
py_modules=['PythonTidy'],
py_modules=['PythonTidy', 'PythonTidyWrapper', 'config', 'runner'],
platforms='Any',
install_requires=[
'setuptools',
],
install_requires=install_requires,
entry_points={
'console_scripts': ['pythontidy = PythonTidy:main'],
'console_scripts': ['pythontidy = runner:main'],
},
)

0 comments on commit 697bbcf

Please sign in to comment.