remove carriage return
This commit is contained in:
210
materials.py
210
materials.py
@@ -1,105 +1,105 @@
|
|||||||
#!/usr/bin/env python
|
#!/usr/bin/env python
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
# -*- mode: python-mode; python-indent-offset: 2; tab-width: 2 -*-
|
# -*- mode: python-mode; python-indent-offset: 2; tab-width: 2 -*-
|
||||||
"""
|
"""
|
||||||
Created on Mon Dec 21 19:39:44 2015
|
Created on Mon Dec 21 19:39:44 2015
|
||||||
|
|
||||||
@author: Daniel
|
@author: Daniel
|
||||||
"""
|
"""
|
||||||
import sys, os
|
import sys, os
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
|
||||||
def list():
|
def list():
|
||||||
from os import listdir
|
from os import listdir
|
||||||
|
|
||||||
for data in listdir("./data"):
|
for data in listdir("./data"):
|
||||||
file_name = data.split('.')[0]
|
file_name = data.split('.')[0]
|
||||||
if file_name != "INFO":
|
if file_name != "INFO":
|
||||||
print(file_name)
|
print(file_name)
|
||||||
|
|
||||||
|
|
||||||
def read(input, search_keys=None):
|
def read(input, search_keys=None):
|
||||||
import json
|
import json
|
||||||
from pprint import pprint
|
from pprint import pprint
|
||||||
|
|
||||||
def find_values(id, json_repr):
|
def find_values(id, json_repr):
|
||||||
results = []
|
results = []
|
||||||
|
|
||||||
def _decode_dict(a_dict):
|
def _decode_dict(a_dict):
|
||||||
try: results.append(a_dict[id])
|
try: results.append(a_dict[id])
|
||||||
except KeyError: pass
|
except KeyError: pass
|
||||||
return a_dict
|
return a_dict
|
||||||
|
|
||||||
json.loads(json_repr, object_hook=_decode_dict) # return value ignored
|
json.loads(json_repr, object_hook=_decode_dict) # return value ignored
|
||||||
return results
|
return results
|
||||||
|
|
||||||
try:
|
try:
|
||||||
with open(os.path.join('data', input + '.json')) as data_file:
|
with open(os.path.join('data', input + '.json')) as data_file:
|
||||||
if search_keys is not None:
|
if search_keys is not None:
|
||||||
data = data_file.read()
|
data = data_file.read()
|
||||||
for search_key in search_keys:
|
for search_key in search_keys:
|
||||||
found_values = find_values(search_key, data)
|
found_values = find_values(search_key, data)
|
||||||
#print(found_values)
|
#print(found_values)
|
||||||
for value in found_values:
|
for value in found_values:
|
||||||
#print('%-5s' % value[0] + ' ' + value[1])
|
#print('%-5s' % value[0] + ' ' + value[1])
|
||||||
print(search_key + ' = ' + '%s' % value[0] + ' ' + value[1])
|
print(search_key + ' = ' + '%s' % value[0] + ' ' + value[1])
|
||||||
|
|
||||||
#def getLength(element):
|
#def getLength(element):
|
||||||
# try:
|
# try:
|
||||||
# element.__iter__
|
# element.__iter__
|
||||||
# return sum([getLength(i) for i in element])
|
# return sum([getLength(i) for i in element])
|
||||||
# except:
|
# except:
|
||||||
# return 1
|
# return 1
|
||||||
#print(len(data))
|
#print(len(data))
|
||||||
#print(getLength(json.loads(data)))
|
#print(getLength(json.loads(data)))
|
||||||
|
|
||||||
else:
|
else:
|
||||||
data = json.load(data_file)
|
data = json.load(data_file)
|
||||||
pprint(data)
|
pprint(data)
|
||||||
|
|
||||||
except (OSError, IOError) as err:
|
except (OSError, IOError) as err:
|
||||||
print(str(err))
|
print(str(err))
|
||||||
sys.exit(2)
|
sys.exit(2)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
parser = argparse.ArgumentParser(description='Material database.')
|
parser = argparse.ArgumentParser(description='Material database.')
|
||||||
|
|
||||||
subparsers = parser.add_subparsers(help='commands', dest='command')
|
subparsers = parser.add_subparsers(help='commands', dest='command')
|
||||||
|
|
||||||
# list materials
|
# list materials
|
||||||
list_parser = subparsers.add_parser('list',
|
list_parser = subparsers.add_parser('list',
|
||||||
description='List materials.',
|
description='List materials.',
|
||||||
help='list all available materials')
|
help='list all available materials')
|
||||||
|
|
||||||
# material
|
# material
|
||||||
material_parser = subparsers.add_parser('material',
|
material_parser = subparsers.add_parser('material',
|
||||||
help='the material to get information from')
|
help='the material to get information from')
|
||||||
material_parser.add_argument('materialname', action='store',
|
material_parser.add_argument('materialname', action='store',
|
||||||
help='list available material information')
|
help='list available material information')
|
||||||
material_parser.add_argument('-r', '--rho', dest='const_collection',
|
material_parser.add_argument('-r', '--rho', dest='const_collection',
|
||||||
action='append_const', const="rho",
|
action='append_const', const="rho",
|
||||||
help='get density from material')
|
help='get density from material')
|
||||||
material_parser.add_argument('-T', '--Tm', dest='const_collection',
|
material_parser.add_argument('-T', '--Tm', dest='const_collection',
|
||||||
action='append_const', const="Tm",
|
action='append_const', const="Tm",
|
||||||
help='get melting point temperature')
|
help='get melting point temperature')
|
||||||
|
|
||||||
parser.add_argument('-v', '--version', action='version',
|
parser.add_argument('-v', '--version', action='version',
|
||||||
version='%(prog)s 1.0')
|
version='%(prog)s 1.0')
|
||||||
|
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
print(args)
|
print(args)
|
||||||
if args.command == 'list':
|
if args.command == 'list':
|
||||||
list()
|
list()
|
||||||
elif args.command == 'material':
|
elif args.command == 'material':
|
||||||
if args.const_collection:
|
if args.const_collection:
|
||||||
read(args.materialname, search_keys=args.const_collection)
|
read(args.materialname, search_keys=args.const_collection)
|
||||||
else:
|
else:
|
||||||
read(args.materialname)
|
read(args.materialname)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user