module.py

Go to the documentation of this file.

#
# Collective Mind
#
# See cM LICENSE.txt for licensing details.
# See cM Copyright.txt for copyright details.
#
# Developer(s): (C) Grigori Fursin, started on 2011.09
#

# Should always be here
ini={}
cm_kernel=None

# Local settings
import string
import os
import sys
import tempfile
import StringIO

# ============================================================================
def init(i):
    return {'cm_return':0}

# ============================================================================
def generate(i):

    """
    Generate qr-code

    Input:  {
              cm_string        - string to convert to qr-code
              qr_level         - qr_level (4,5,etc)
              image_size       - picture size
            }

    Output: {
              cm_return        - return code >0 if error
              cm_full_filename - file with image
            }
    """

    if 'cm_string' not in i: return {'cm_return':1, 'cm_error':'"cm_string" is not set'}
    if 'qr_level' not in i: return {'cm_return':1, 'cm_error':'"qr_level" is not set'}
    if 'image_size' not in i: return {'cm_return':1, 'cm_error':'"image_size" is not set'}

    # Import PyQRNative module
    r=cm_kernel.load_code({'cm_path':ini['path'], 'cm_code_name':'PyQRNative'})
    if r['cm_return']>0: return r

    qrm=r['code']

    # Prepare QR code
    qr = qrm.QRCode(int(i['qr_level']), qrm.QRErrorCorrectLevel.L)
    qr.addData(i['cm_string'])
    qr.make()
    im = qr.makeImage()
    im1=im.resize((int(i['image_size']), int(i['image_size'])))

    # Generate tmp file
    r=cm_kernel.gen_cm_tmp_file({})
    if r['cm_return']>0: return r
    tf=r['cm_path']
    uid=r['cm_uid']
  
    # Save image
    im1.save(tf, 'PNG')

    ii={}
    ii['cm_full_filename']=tf

    if i.get('cm_console','')=='txt':
       cm_kernel.print_for_con(tf)
    else:
       ii['cm_web_download']='yes'
       ii['cm_web_set_headers']='yes'

       # Prepare content header
       ii1={'cm_run_module_uoa':ini['cfg']['cm_modules']['cm-web'],
            'cm_action':'prepare_content_header',
            'cm_web_content':'.png',
            'cm_filename':tf}

       r=cm_kernel.access(ii1)
       if r['cm_return']>0: return r

       # Substitute special variables
       wh=[]
       for x in r['cm_web_headers']:
           wh.append(x.replace(cm_kernel.convert_str_to_special('filename'), tf))

       ii['cm_web_headers']=wh

    ii['cm_web_postprocess']='no'
    ii['cm_return']=0

    ii['cm_delete_file_after_view']='yes' # for tmp files!

    return ii