import contextlib

from os.path import exists

from os import getcwd, chdir, makedirs

import io

import json

import shutil

import sys

from tempfile import mkdtemp

try:

from urllib.request import FancyURLopener

except ImportError:

from urllib import FancyURLopener

from pythonforandroid.logger import (logger, Err_Fore)

IS_PY3 = sys.version_info[0] >= 3

if IS_PY3:

unistr = str

else:

unistr = unicode

class WgetDownloader(FancyURLopener):

version = ('Wget/1.17.1')

urlretrieve = WgetDownloader().retrieve

@contextlib.contextmanager

def current_directory(new_dir):

cur_dir = getcwd()

logger.info(''.join((Err_Fore.CYAN, '-> directory context ', new_dir,

Err_Fore.RESET)))

chdir(new_dir)

yield

logger.info(''.join((Err_Fore.CYAN, '<- directory context ', cur_dir,

Err_Fore.RESET)))

chdir(cur_dir)

@contextlib.contextmanager

def temp_directory():

temp_dir = mkdtemp()

try:

logger.debug(''.join((Err_Fore.CYAN, ' + temp directory used ',

temp_dir, Err_Fore.RESET)))

yield temp_dir

finally:

shutil.rmtree(temp_dir)

logger.debug(''.join((Err_Fore.CYAN, ' - temp directory deleted ',

temp_dir, Err_Fore.RESET)))

def ensure_dir(filename):

if not exists(filename):

makedirs(filename)

class JsonStore(object):

"""Replacement of shelve using json, needed for support python 2 and 3.

"""

def __init__(self, filename):

super(JsonStore, self).__init__()

self.filename = filename

self.data = {}

if exists(filename):

try:

with io.open(filename, encoding='utf-8') as fd:

self.data = json.load(fd)

except ValueError:

print("Unable to read the state.db, content will be replaced.")

def __getitem__(self, key):

return self.data[key]

def __setitem__(self, key, value):

self.data[key] = value

self.sync()

def __delitem__(self, key):

del self.data[key]

self.sync()

def __contains__(self, item):

return item in self.data

def get(self, item, default=None):

return self.data.get(item, default)

def keys(self):

return self.data.keys()

def remove_all(self, prefix):

for key in self.data.keys()[:]:

if not key.startswith(prefix):

continue

del self.data[key]

self.sync()

def sync(self):

# http://stackoverflow.com/questions/12309269/write-json-data-to-file-in-python/14870531#14870531

if IS_PY3:

with open(self.filename, 'w') as fd:

json.dump(self.data, fd, ensure_ascii=False)

else:

with io.open(self.filename, 'w', encoding='utf-8') as fd:

fd.write(unicode(json.dumps(self.data, ensure_ascii=False)))

def which(program, path_env):

'''Locate an executable in the system.'''

import os

def is_exe(fpath):

return os.path.isfile(fpath) and os.access(fpath, os.X_OK)

fpath, fname = os.path.split(program)

if fpath:

if is_exe(program):

return program

else:

for path in path_env.split(os.pathsep):

path = path.strip('"')

exe_file = os.path.join(path, program)

if is_exe(exe_file):

return exe_file

return None

def get_directory(filename):

'''If the filename ends with a recognised file extension, return the

filename without this extension.'''

if filename.endswith('.tar.gz'):

return basename(filename[:-7])

elif filename.endswith('.tgz'):

return basename(filename[:-4])

elif filename.endswith('.tar.bz2'):

return basename(filename[:-8])

elif filename.endswith('.tbz2'):

return basename(filename[:-5])

elif filename.endswith('.zip'):

return basename(filename[:-4])

info('Unknown file extension for {}'.format(filename))

exit(1)