common.py 9.43 KB
Newer Older
1 2
# -*- coding: utf-8 -*-

3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
# some functions coming from gajim sources (thanks)

## Copyright (C) 2003-2008 Yann Leboulanger <asterix AT lagaule.org>
## Copyright (C) 2005-2006 Dimitur Kirov <dkirov AT gmail.com>
##                         Nikos Kouremenos <kourem AT gmail.com>
## Copyright (C) 2006 Alex Mauer <hawke AT hawkesnest.net>
## Copyright (C) 2006-2007 Travis Shirk <travis AT pobox.com>
## Copyright (C) 2006-2008 Jean-Marie Traissard <jim AT lapin.org>
## Copyright (C) 2007 Lukas Petrovicky <lukas AT petrovicky.net>
##                    James Newton <redshodan AT gmail.com>
##                    Julien Pivotto <roidelapluie AT gmail.com>
## Copyright (C) 2007-2008 Stephan Erb <steve-e AT h3c.de>
## Copyright (C) 2008 Brendan Taylor <whateley AT gmail.com>
##                    Jonathan Schleifer <js-gajim AT webkeks.org>
##

19 20 21 22 23 24 25 26 27 28 29 30 31 32
# Copyright 2010, Florent Le Coz <louizatakk@fedoraproject.org>

# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation version 3 of the License.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

33 34 35
"""
various useful functions
"""
36

37
from datetime import datetime, timedelta
38 39 40 41
import base64
import os
import mimetypes
import hashlib
42
import subprocess
43 44
import curses
import sys
45 46
import select
import errno
47
import time
48
import traceback
49

50 51
import xmpp

52 53 54 55 56 57
ROOM_STATE_NONE = 11
ROOM_STATE_CURRENT = 10
ROOM_STATE_PRIVATE = 15
ROOM_STATE_MESSAGE = 12
ROOM_STATE_HL = 13

58 59 60 61 62 63 64
def debug(string):
    """
    Print a string in a file.
    Useful since debuging cannot be displayed on screen because it's
    a CLI software
    """
    fdes = open("debug", 'a')
65 66 67 68 69 70 71
    try:
        fdes.write(string)
    except:
        try:
            fdes.write(string.encode('utf-8'))
        except:
            fdes.write(string.encode('utf-8'))
72
    fdes.close()
73

74
def get_base64_from_file(path):
75 76 77
    """
    Convert the content of a file to base64
    """
78 79 80 81 82
    if not os.path.isfile(path):
        return (None, None, "File does not exist")
    size = os.path.getsize(path)
    if size > 16384:
        return (None, None,"File is too big")
83 84
    fdes = open(path, 'rb')
    data = fdes.read()
85 86 87 88
    encoded = base64.encodestring(data)
    sha1 = hashlib.sha1(data).hexdigest()
    mime_type = mimetypes.guess_type(path)[0]
    return (encoded, mime_type, sha1)
89 90

def get_output_of_command(command):
91 92 93
    """
    Runs a command and returns its output
    """
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
    try:
        child_stdin, child_stdout = os.popen2(command)
    except ValueError:
        return None

    output = child_stdout.readlines()
    child_stdout.close()
    child_stdin.close()

    return output

def is_in_path(command, return_abs_path=False):
    """
    Return True if 'command' is found in one of the directories in the user's
    path. If 'return_abs_path' is True, return the absolute path of the first
    found command instead. Return False otherwise and on errors
    """
    for directory in os.getenv('PATH').split(os.pathsep):
        try:
            if command in os.listdir(directory):
                if return_abs_path:
                    return os.path.join(directory, command)
                else:
                    return True
        except OSError:
            # If the user has non directories in his path
            pass
    return False

123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173
def get_stripped_jid(jid):
    """
    Return the stripped JID (bare representation)
    nick@server/resource -> nick@server
    """
    if isinstance(jid, basestring):
        jid = xmpp.JID(jid)
    return jid.getStripped()

def is_jid(jid):
    """
    Return True if this is a valid JID
    """
    if xmpp.JID(jid).getNode() != '':
        return True
    return False

def jid_get_node(jid):
    """
    nick@server/resource -> nick
    """
    if isinstance(jid, basestring):
        jid = xmpp.JID(jid)
    return jid.getNode()

def jid_get_domain(jid):
    """
    nick@server/resource -> server
    """
    if isinstance(jid, basestring):
        jid = xmpp.JID(jid)
    return jid.getDomain()

def jid_get_resource(jid):
    """
    nick@server/resource -> resource
    """
    if isinstance(jid, basestring):
        jid = xmpp.JID(jid)
    return jid.getResource()

def is_jid_the_same(a, b):
    """
    Compare two bare jids
    """
    if isinstance(a, basestring):
        a = xmpp.JID(a)
    if isinstance(b, basestring):
        b = xmpp.JID(b)
    return a.bareMatch(b)

174
DISTRO_INFO = {
175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194
        'Arch Linux': '/etc/arch-release',
        'Aurox Linux': '/etc/aurox-release',
        'Conectiva Linux': '/etc/conectiva-release',
        'CRUX': '/usr/bin/crux',
        'Debian GNU/Linux': '/etc/debian_version',
        'Fedora Linux': '/etc/fedora-release',
        'Gentoo Linux': '/etc/gentoo-release',
        'Linux from Scratch': '/etc/lfs-release',
        'Mandrake Linux': '/etc/mandrake-release',
        'Slackware Linux': '/etc/slackware-version',
        'Solaris/Sparc': '/etc/release',
        'Source Mage': '/etc/sourcemage_version',
        'SUSE Linux': '/etc/SuSE-release',
        'Sun JDS': '/etc/sun-release',
        'PLD Linux': '/etc/pld-release',
        'Yellow Dog Linux': '/etc/yellowdog-release',
        # many distros use the /etc/redhat-release for compatibility
        # so Redhat is the last
        'Redhat Linux': '/etc/redhat-release'
}
195 196 197 198 199 200 201 202 203 204 205 206 207 208

def temp_failure_retry(func, *args, **kwargs):
    """
    workaround for a temporary and specific failure
    """
    while True:
        try:
            return func(*args, **kwargs)
        except (os.error, IOError, select.error), ex:
            if ex.errno == errno.EINTR:
                continue
            else:
                raise

209
def get_os_info():
210 211 212 213 214
    """
    Returns a detailed and well formated string containing
    informations about the operating system
    """
    if os.name == 'posix':
215 216 217 218 219
        executable = 'lsb_release'
        params = ' --description --codename --release --short'
        full_path_to_executable = is_in_path(executable, return_abs_path = True)
        if full_path_to_executable:
            command = executable + params
220 221 222 223 224 225
            process = subprocess.Popen([command], shell=True,
                                       stdin=subprocess.PIPE,
                                       stdout=subprocess.PIPE,
                                       close_fds=True)
            process.wait()
            output = temp_failure_retry(process.stdout.readline).strip()
226 227 228 229 230
            # some distros put n/a in places, so remove those
            output = output.replace('n/a', '').replace('N/A', '')
            return output

        # lsb_release executable not available, so parse files
231 232
        for distro_name in DISTRO_INFO:
            path_to_file = DISTRO_INFO[distro_name]
233 234 235 236 237 238
            if os.path.exists(path_to_file):
                if os.access(path_to_file, os.X_OK):
                    # the file is executable (f.e. CRUX)
                    # yes, then run it and get the first line of output.
                    text = get_output_of_command(path_to_file)[0]
                else:
239 240 241
                    fdes = open(path_to_file)
                    text = fdes.readline().strip() # get only first line
                    fdes.close()
242 243 244 245 246 247 248 249 250 251 252
                    if path_to_file.endswith('version'):
                        # sourcemage_version and slackware-version files
                        # have all the info we need (name and version of distro)
                        if not os.path.basename(path_to_file).startswith(
                        'sourcemage') or not\
                        os.path.basename(path_to_file).startswith('slackware'):
                            text = distro_name + ' ' + text
                    elif path_to_file.endswith('aurox-release') or \
                    path_to_file.endswith('arch-release'):
                        # file doesn't have version
                        text = distro_name
253 254
                    elif path_to_file.endswith('lfs-release'):
                        # file just has version
255 256 257 258 259 260 261 262 263 264 265
                        text = distro_name + ' ' + text
                os_info = text.replace('\n', '')
                return os_info

        # our last chance, ask uname and strip it
        uname_output = get_output_of_command('uname -sr')
        if uname_output is not None:
            os_info = uname_output[0] # only first line
            return os_info
    os_info = 'N/A'
    return os_info
266 267 268 269 270 271 272 273 274 275 276 277 278 279

def datetime_tuple(timestamp):
    """
    Convert timestamp using strptime and the format: %Y%m%dT%H:%M:%S

    Because of various datetime formats are used the following exceptions
    are handled:
            - Optional milliseconds appened to the string are removed
            - Optional Z (that means UTC) appened to the string are removed
            - XEP-082 datetime strings have all '-' cahrs removed to meet
              the above format.
    """
    timestamp = timestamp.split('.')[0]
    timestamp = timestamp.replace('-', '')
280 281 282 283
    try:
        ret = datetime.strptime(timestamp, '%Y%m%dT%H:%M:%SZ')
    except ValueError:      # Support the deprecated format, XEP 0091 :(
        ret = datetime.strptime(timestamp, '%Y%m%dT%H:%M:%S')
284 285 286 287
    # convert UTC to local time, with DST etc.
    dst = timedelta(seconds=time.altzone)
    ret -= dst
    return ret