def to_attachment()

in pygenie/adapter/genie_2.py [0:0]

• 11 lines of code
• 5 McCabe index (conditional complexity)

def to_attachment(att):
    if is_str(att) and os.path.isfile(att):
        with open(att, 'r') as content_file:
            content = content_file.read()
        return dict(name=os.path.basename(att), data=base64.b64encode(content))
    elif isinstance(att, dict):
        try:
            return dict(name=att['name'], data=base64.b64encode(att['data']))
        except KeyError:
            raise GenieAttachmentError("in-line attachment is missing required keys ('name', 'data') ({})".format(att))
    raise GenieAttachmentError("cannot handle attachment '{}'".format(att))