import os

from argparse import ArgumentParser

import barcode

from barcode.version import version

from barcode.writer import ImageWriter, SVGWriter

# Optional PyQt4 GUI

try:

from PyQt4 import QtCore

except ImportError:

QtCore = None # lint:ok

# No GUI available yet

QtCore = None

IMG_FORMATS = ('BMP', 'GIF', 'JPEG', 'MSP', 'PCX', 'PNG', 'TIFF', 'XBM')

def open_gui(args, parser=None):

pass

def list_types(args, parser=None):

print('\npyBarcode available barcode formats:')

print(', '.join(barcode.PROVIDED_BARCODES))

print('\n')

print('Available image formats')

print('Standard: svg')

if ImageWriter is not None:

print('Pillow:', ', '.join(IMG_FORMATS))

else:

print('Pillow: disabled')

print('\n')

def create_barcode(args, parser):

args.type = args.type.upper()

if args.type != 'SVG' and args.type not in IMG_FORMATS:

parser.error(

'Unknown type {type}. Try list action for available '

'types.'.format(type=args.type)

)

args.barcode = args.barcode.lower()

if args.barcode not in barcode.PROVIDED_BARCODES:

parser.error(

'Unknown barcode {bc}. Try list action for available '

'barcodes.'.format(bc=args.barcode)

)

if args.type != 'SVG':

opts = {'format': args.type}

writer = ImageWriter()

else:

opts = {'compress': args.compress}

writer = SVGWriter()

out = os.path.normpath(os.path.abspath(args.output))

name = barcode.generate(

args.barcode, args.code, writer, out, opts, args.text

)

print('New barcode saved as {0}.'.format(name))

def main():

msg = []

if ImageWriter is None:

msg.append(

'Image output disabled (Pillow not found), --type option disabled.'

)

else:

msg.append(

'Image output enabled, use --type option to give image '

'format (png, jpeg, ...).'

)

if QtCore is None:

msg.append('PyQt not found, gui action disabled.')

else:

msg.append('PyQt found. Use gui action to get a simple GUI.')

parser = ArgumentParser(

description='Create standard barcodes via cli.', epilog=' '.join(msg)

)

parser.add_argument(

'-v', '--version', action='version', version='%(prog)s ' + version

)

subparsers = parser.add_subparsers(title='Actions')

create_parser = subparsers.add_parser(

'create', help='Create a barcode '

'with the given options.'

)

create_parser.add_argument('code', help='Code to render as barcode.')

create_parser.add_argument(

'output',

help='Filename for output '

'without extension, e. g. mybarcode.'

)

create_parser.add_argument(

'-c',

'--compress',

action='store_true',

help='Compress output, only recognized if type is svg.'

)

create_parser.add_argument(

'-b', '--barcode', help='Barcode to use '

'[default: %(default)s].'

)

create_parser.add_argument(

'--text', help='Text to show under the '

'barcode.'

)

if ImageWriter is not None:

create_parser.add_argument(

'-t', '--type', help='Type of output '

'[default: %(default)s].'

)

list_parser = subparsers.add_parser(

'list', help='List available '

'image and code types.'

)

list_parser.set_defaults(func=list_types)

if QtCore is not None:

gui_parser = subparsers.add_parser(

'gui', help='Opens a simple '

'PyQt GUI to create barcodes.'

)

gui_parser.set_defaults(func=open_gui)

create_parser.set_defaults(

type='svg',

compress=False,

func=create_barcode,

barcode='code39',

text=None

)

args = parser.parse_args()

try:

func = args.func

except AttributeError:

parser.error("You need to tell me what to do.")

else:

func(args, parser)

if __name__ == '__main__':

main()