Subversion Repositories SmartDukaan

from Crypto.Hash import MD5

from Crypto.Cipher import DES

import string

_password = 'dtr18Feb2015'

_salt = '\xA9\x9B\xC8\x32\x56\x35\xE3\x03'

_iterations = 19

ALPHABET = string.ascii_uppercase + string.ascii_lowercase + \

           string.digits + '-_'

ALPHABET_REVERSE = dict((c, i) for (i, c) in enumerate(ALPHABET))

BASE = len(ALPHABET)

SIGN_CHARACTER = '$'

def encryptMessage(plaintext_to_encrypt1):

    hasher = MD5.new()

    hasher.update(_password)

    hasher.update(_salt)

    result = hasher.digest()

    for i in range(1, _iterations):

        hasher = MD5.new()

        hasher.update(result)

        result = hasher.digest()

    padding = 8 - len(plaintext_to_encrypt1) % 8

    plaintext_to_encrypt1 += chr(padding) * padding

    encoder = DES.new(result[:8], DES.MODE_CBC, result[8:16])

    encrypted = encoder.encrypt(plaintext_to_encrypt1)

    check1 =  encrypted.encode('base64')

    return check1

def decryptMessage(encryptedData):

    padding = 8 - len(encryptedData) % 8

    encryptedData += chr(padding) * padding

    hasher = MD5.new()

    hasher.update(_password)

    hasher.update(_salt)

    result = hasher.digest()

    for i in range(1, _iterations):

        hasher = MD5.new()

        hasher.update(result)

        result = hasher.digest()

    encoder2 = DES.new(result[:8], DES.MODE_CBC, result[8:16])

    decodeString =  encoder2.decrypt(encryptedData.decode('base64'))

    return ''.join(e for e in decodeString if e.isalnum() or e is '/' or e is ':' or e is ',' or e is '.')

def num_encode(n):

    if n < 0:

        return SIGN_CHARACTER + num_encode(-n)

    s = []

    while True:

        n, r = divmod(n, BASE)

        s.append(ALPHABET[r])

        if n == 0: break

    return ''.join(reversed(s))

def num_decode(s):

    if s[0] == SIGN_CHARACTER:

        return -num_decode(s[1:])

    n = 0

    for c in s:

        n = n * BASE + ALPHABET_REVERSE[c]

    return n