FilesExpand file tree

 master

/

util.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

144 lines (115 loc) · 4.04 KB

 master

/

util.py

Top

File metadata and controls

• Code
• Blame

144 lines (115 loc) · 4.04 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

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

class ChromeDownloader(FancyURLopener):

version = (

'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 '

'(KHTML, like Gecko) Chrome/28.0.1500.71 Safari/537.36')

urlretrieve = ChromeDownloader().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)