plot_legend.py

#!/usr/bin/env python3
import sys
import os
import argparse
import logging
from src.file_handler import file_handler
import src.graph_tool_functions as gt
from src.version import repo_version
__version__ = repo_version()



DESCRIPTION = '''
Create legends to accompany networks.
'''



## Pass arguments.
def main():
	## Pass command line arguments. 
	parser = argparse.ArgumentParser(formatter_class=argparse.RawDescriptionHelpFormatter, description=DESCRIPTION)
	parser.add_argument('-i', '--input_table', metavar='colors.tsv', 
		required=False, default=sys.stdin, type=lambda x: file_handler(x, 'r'), 
		help='Input [compressed] file listing ID and hex colors to use for the legend (default: stdin)'
	)
	parser.add_argument('-o', '--output_plot', metavar='legend.pdf', 
		required=True, type=lambda x: file_handler(x, 'ws'),
		help='Output plot file'
	)
	parser.add_argument('-s', '--size_factor', metavar=0.5,
		required=False, default=0.5, type=float,
		help='Factor to scale legend size by'
	)
	parser.add_argument('-d', '--delim',
		required=False, default='\t', type=str,
		help='Delimiter of columns in input_table file (default: \\t)'
	)
	parser.add_argument('--version', action='version',
		version=f"{__version__}",
		help='Print repo version'
	)
	parser.add_argument('--debug', 
		required=False, action='store_true', 
		help='Print DEBUG info (default: %(default)s)'
	)
	args = parser.parse_args()
	
	## Set up basic debugger
	if args.debug:
		logging.getLogger().setLevel(logging.DEBUG)
	
	logging.debug('%s', args) ## DEBUG
	logging.info('Version: %s', __version__) ##INFO
	
	## Workflow
	with args.input_table as in_color_file, args.output_plot as out_plot_file:
		gt.plot_legend(in_color_file, out_plot_file, args.size_factor, args.delim)



if __name__ == '__main__':
	log_format = "%(asctime)s [%(levelname)s] "+os.path.basename(__file__)+": %(message)s"
	logging.basicConfig(format=log_format, stream=sys.stderr, level=logging.INFO)
	main()