#!/usr/bin/env python

#coding: utf-8

‘‘‘

+-----------------------------------------------------------------------+

|Author: Cheng Wenfeng <277546922@qq.com>                               |

+-----------------------------------------------------------------------+

‘‘‘

import sys

import base64

import getopt

from Crypto.Cipher import AES

if len(sys.argv) < 2:

        print ‘Syntax: %s handle[encode|decode] str \n \

        eg: %s  encode  277546922@qq.com‘ % (sys.argv[0],sys.argv[0])

        sys.exit()

opts,args=getopt.getopt(sys.argv[1:], "h", [‘help‘])

for opt,arg in  opts:

        if opt == ‘--help‘ or opt == ‘-h‘      :

           print ‘Syntax: %s handle[encode|decode] str \n \

        eg: %s  encode  277546922@qq.com‘ % (sys.argv[0],sys.argv[0])

        sys.exit()

Formatkey = ‘\0‘

FormatStr = lambda s: s+(16 - len(s)%16)*Formatkey

key=‘qq277546922qqqqq‘

data=http://www.mamicode.com/str(sys.argv[2])

def encode(key,data):

        keyIV=key

        obj = AES.new(key, AES.MODE_CBC,keyIV)

        ciphertext = base64.b64encode(obj.encrypt(FormatStr(data)))

        print ciphertext

def decode(key,data):

        keyIV=key

        obj2 = AES.new(key, AES.MODE_CBC,keyIV)

        try:

                newmesg = obj2.decrypt(base64.b64decode(data))

                print newmesg.replace(‘\0‘,‘‘)

        except  TypeError:

                print ‘Decode Fail‘

if sys.argv[1]==‘encode‘:

        encode(key,data)

elif sys.argv[1]==‘decode‘:

        decode(key,data)

else :

        print ‘handle error‘

        sys.exit()