def test_extract_attr():
code = '''
import sys
sys.stdin.write('hello, world!\\n')
'''
module_node = parse_source(code)
module_accessors = list(get_module_accessors(module_node))
#
assert 'sys.stdin' in module_accessors
assert 'sys.stdin.write' in module_accessors
def test_extract_attr():
code = '''
import sys
sys.stdin.write('hello, world!\\n')
'''
module_node = parse_source(code)
module_accessors = list(get_module_accessors(module_node))
#
assert 'sys.stdin' in module_accessors
assert 'sys.stdin.write' in module_accessors
def main(argv):
try:
filename = argv[1]
except IndexError:
print 'usage: %s [filename]' % argv[0]
return 1
#
with open(filename) as f:
module_node = parse_source(f.read())
import_names = list(get_import_names(module_node))
accessors = list(get_module_accessors(module_node))
print import_names
print accessors
def test_extract_attr_with_from_and_as():
code = '''
from sys import stdin as mystdin
mystdin.write('hello, world!\\n')
from os.path import dirname as mydirname
print mydirname(__file__)
'''
module_node = parse_source(code)
module_accessors = list(get_module_accessors(module_node))
#
assert 'sys.stdin.write' in module_accessors
assert 'os.path.dirname' in module_accessors
def test_extract_attr_with_from_and_as():
code = '''
from sys import stdin as mystdin
mystdin.write('hello, world!\\n')
from os.path import dirname as mydirname
print mydirname(__file__)
'''
module_node = parse_source(code)
module_accessors = list(get_module_accessors(module_node))
#
assert 'sys.stdin.write' in module_accessors
assert 'os.path.dirname' in module_accessors
def main(argv):
try:
filename = argv[1]
except IndexError:
print 'usage: %s [filename]' % argv[0]
return 1
#
with open(filename) as f:
module_node = parse_source(f.read())
import_names = list(get_import_names(module_node))
accessors = list(get_module_accessors(module_node))
print import_names
print accessors
