import glob
import plistlib
import os
from bplist import BPlistReader
import cPickle
import gzip

def read_file(filename):
    f = open(filename, "rb")
    data = f.read()
    f.close()
    return data

def write_file(filename,data):
    f = open(filename, "wb")
    f.write(data)
    f.close()

def makedirs(dirs):
    try:
        os.makedirs(dirs)
    except:
        pass

def getHomePath(foldername, filename):
    home = os.path.expanduser('~')
    folderpath = os.path.join(home, foldername)
    if not os.path.exists(folderpath):
        makedirs(folderpath)
    return os.path.join(folderpath, filename)

def readHomeFile(foldername, filename):
    path = getHomePath(foldername, filename)
    if not os.path.exists(path):
        return None
    return read_file(path)

#return path to HOME+foldername+filename
def writeHomeFile(foldername, filename, data):
    filepath = getHomePath(foldername, filename)
    write_file(filepath, data)
    return filepath
    
def readPlist(filename):
    f = open(filename,"rb")
    d = f.read(16)
    f.close()
    if d.startswith("bplist"):
        return BPlistReader.plistWithFile(filename)
    else:
        return plistlib.readPlist(filename)

def parsePlist(s):
    if s.startswith("bplist"):
        return BPlistReader.plistWithString(s)
    else:
        return plistlib.readPlistFromString(s)

#http://stackoverflow.com/questions/1094841/reusable-library-to-get-human-readable-version-of-file-size
def sizeof_fmt(num):
    for x in ['bytes','KB','MB','GB','TB']:
        if num < 1024.0:
            return "%d%s" % (num, x)
        num /= 1024.0
        
#http://www.5dollarwhitebox.org/drupal/node/84
def convert_bytes(bytes):
    bytes = float(bytes)
    if bytes >= 1099511627776:
        terabytes = bytes / 1099511627776
        size = '%.2fT' % terabytes
    elif bytes >= 1073741824:
        gigabytes = bytes / 1073741824
        size = '%.2fG' % gigabytes
    elif bytes >= 1048576:
        megabytes = bytes / 1048576
        size = '%.2fM' % megabytes
    elif bytes >= 1024:
        kilobytes = bytes / 1024
        size = '%.2fK' % kilobytes
    else:
        size = '%.2fb' % bytes
    return size

def xor_strings(a,b):
    r=""
    for i in xrange(len(a)):
        r+= chr(ord(a[i])^ord(b[i]))
    return r

hex = lambda data: " ".join("%02X" % ord(i) for i in data)
ascii = lambda data: "".join(c if 31 < ord(c) < 127 else "." for c in data)

def hexdump(d):
    for i in xrange(0,len(d),16):
        data = d[i:i+16]
        print "%08X | %s | %s" % (i, hex(data).ljust(47), ascii(data))

def search_plist(directory, matchDict):
    for p in map(os.path.normpath, glob.glob(directory + "/*.plist")):
        try:
            d = plistlib.readPlist(p)
            ok = True
            for k,v in matchDict.items():
                if d.get(k) != v:
                    ok = False
                    break
            if ok:
                print "Using plist file %s" % p
                return d
        except:
            continue

def save_pickle(filename,data):
    f = gzip.open(filename,"wb")
    cPickle.dump(data, f, cPickle.HIGHEST_PROTOCOL)
    f.close()
    
def load_pickle(filename):
    f = gzip.open(filename,"rb")
    data = cPickle.load(f)
    f.close()
    return data