]> kaliko git repositories - mpd-sima.git/blob - sima/utils/utils.py
Fixed code smell
[mpd-sima.git] / sima / utils / utils.py
1 # -*- coding: utf-8 -*-
2 #
3 # Copyright (c) 2010, 2011, 2013, 2014, 2015 kaliko <kaliko@azylum.org>
4 #
5 #  This file is part of sima
6 #
7 #  sima is free software: you can redistribute it and/or modify
8 #  it under the terms of the GNU General Public License as published by
9 #  the Free Software Foundation, either version 3 of the License, or
10 #  (at your option) any later version.
11 #
12 #  sima is distributed in the hope that it will be useful,
13 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
14 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 #  GNU General Public License for more details.
16 #
17 #  You should have received a copy of the GNU General Public License
18 #  along with sima.  If not, see <http://www.gnu.org/licenses/>.
19 #
20 #
21 """generic tools and utilities for sima
22 """
23 # pylint: disable=C0111
24
25 import logging
26 import traceback
27 import sys
28
29 from argparse import ArgumentError, Action
30 from base64 import b64decode as push
31 from codecs import getencoder
32 from datetime import datetime
33 from os import environ, access, getcwd, W_OK, R_OK
34 from os.path import dirname, isabs, join, normpath, exists, isdir, isfile
35 from time import sleep
36
37
38 def getws(dic):
39     """
40     Decode Obfuscated api key.
41     Only preventing API keys harvesting over the network
42     https://developer.echonest.com/forums/thread/105
43     """
44     aka = push(bytes(dic.get('apikey') + '=', 'utf-8'))
45     aka = getencoder('rot-13')(str((aka), 'utf-8'))[0]
46     dic.update({'apikey':aka})
47
48 def get_mpd_environ():
49     """
50     Retrieve MPD env. var.
51     """
52     passwd = host = None
53     mpd_host_env = environ.get('MPD_HOST')
54     if mpd_host_env:
55         # If password is set:
56         # mpd_host_env = ['pass', 'host'] because MPD_HOST=pass@host
57         mpd_host_env = mpd_host_env.split('@')
58         mpd_host_env.reverse()
59         host = mpd_host_env[0]
60         if len(mpd_host_env) > 1 and mpd_host_env[1]:
61             passwd = mpd_host_env[1]
62     return (host, environ.get('MPD_PORT', None), passwd)
63
64 def normalize_path(path):
65     """Get absolute path
66     """
67     if not isabs(path):
68         return normpath(join(getcwd(), path))
69     return path
70
71 def exception_log():
72     """Log unknown exceptions"""
73     log = logging.getLogger(__name__)
74     log.error('Unhandled Exception!!!')
75     log.error(''.join(traceback.format_exc()))
76     log.info('Please report the previous message'
77              ' along with some log entries right before the crash.')
78     log.info('thanks for your help :)')
79     log.info('Quiting now!')
80     sys.exit(1)
81
82
83 class SigHup(Exception):
84     """SIGHUP raises this Exception"""
85     pass
86
87 # ArgParse Callbacks
88 class Obsolete(Action):
89     # pylint: disable=R0903
90     """Deal with obsolete arguments
91     """
92     def __call__(self, parser, namespace, values, option_string=None):
93         raise ArgumentError(self, 'obsolete argument')
94
95 class FileAction(Action):
96     """Generic class to inherit from for ArgParse action on file/dir
97     """
98     # pylint: disable=R0903
99     def __call__(self, parser, namespace, values, option_string=None):
100         self._file = normalize_path(values)
101         self._dir = dirname(self._file)
102         self.parser = parser
103         self.checks()
104         setattr(namespace, self.dest, self._file)
105
106     def checks(self):
107         """control method
108         """
109         pass
110
111 class Wfile(FileAction):
112     # pylint: disable=R0903
113     """Is file writable
114     """
115     def checks(self):
116         if isdir(self._file):
117             self.parser.error('need a file not a directory: {}'.format(self._file))
118         if not exists(self._dir):
119             #raise ArgumentError(self, '"{0}" does not exist'.format(self._dir))
120             self.parser.error('directory does not exist: {0}'.format(self._dir))
121         if not exists(self._file):
122             # Is parent directory writable then
123             if not access(self._dir, W_OK):
124                 self.parser.error('no write access to "{0}"'.format(self._dir))
125         else:
126             if not access(self._file, W_OK):
127                 self.parser.error('no write access to "{0}"'.format(self._file))
128
129 class Rfile(FileAction):
130     # pylint: disable=R0903
131     """Is file readable
132     """
133     def checks(self):
134         if not exists(self._file):
135             self.parser.error('file does not exist: {0}'.format(self._file))
136         if not isfile(self._file):
137             self.parser.error('not a file: {0}'.format(self._file))
138         if not access(self._file, R_OK):
139             self.parser.error('no read access to "{0}"'.format(self._file))
140
141 class Wdir(FileAction):
142     # pylint: disable=R0903
143     """Is directory writable
144     """
145     def checks(self):
146         if not exists(self._file):
147             self.parser.error('directory does not exist: {0}'.format(self._file))
148         if not isdir(self._file):
149             self.parser.error('not a directory: {0}'.format(self._file))
150         if not access(self._file, W_OK):
151             self.parser.error('no write access to "{0}"'.format(self._file))
152
153 class Throttle:
154     """throttle decorator"""
155     def __init__(self, wait):
156         self.wait = wait
157         self.last_called = datetime.now()
158
159     def __call__(self, func):
160         def wrapper(*args, **kwargs):
161             while self.last_called + self.wait > datetime.now():
162                 sleep(0.1)
163             result = func(*args, **kwargs)
164             self.last_called = datetime.now()
165             return result
166         return wrapper
167
168 # http client exceptions (for webservices)
169
170 class WSError(Exception):
171     pass
172
173 class WSNotFound(WSError):
174     pass
175
176 class WSTimeout(WSError):
177     pass
178
179 class WSHTTPError(WSError):
180     pass
181
182
183 # VIM MODLINE
184 # vim: ai ts=4 sw=4 sts=4 expandtab