summaryrefslogtreecommitdiffstats
path: root/util.py
blob: 1e932cb324f71a1bd55b71db24f7b05d4c6786c9 (plain)
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
import os
import sys

class LogType:
    Non = 0
    Err = 1
    Warn = 2
    Msg = 3
    @staticmethod
    def from_int(lloglevel):
        iloglevel = int(lloglevel)
        if iloglevel == 0:
            return LogType.Non
        if iloglevel == 1:
            return LogType.Err
        if iloglevel == 2:
            return LogType.Warn
        if iloglevel == 3:
            return LogType.Msg

loglevel = LogType.Non

def raw_input_with_default(default, prompt):
    inp = raw_input("%s (default=%s): " % (prompt, default))
    if inp == "":
        return default
    return inp

def log(logtype, message, data = None):
    global loglevel
    if loglevel >= logtype:
        print message
        if not data is None:
            print "data:\n\tdata"

homedir = os.getenv("HOME")
if homedir is None:
    homedir = os.path.dirname(sys.argv[0])
confdir = os.path.join(homedir, ".4grab")