conf.py

   1 #!/usr/bin/env python
   2
   3 # Author: Martin Matusiak <numerodix@gmail.com>
   4 # Licensed under the GNU Public License, version 3.
   5
   6 import os
   7 import sys
   8
   9 from sdict import sdict, sdictlist
  10 import regex
  11 import system
  12
  13
  14 class Config:
  15     def __init__(self):
  16
  17         self.opts = sdict()
  18
  19         self.interfaces = sdictlist(sort_key="priority")
  20         self.interface_media = ["wired", "wireless"]
  21         self.interface_fields = ["interface", "priority"]
  22
  23         self.networks = sdictlist(sort_key="priority")
  24         self.network_fields = ["essid", "key", "priority", "vpn_exec"]
  25
  26         self.scan_count = 0
  27         self.live_networks = sdictlist(sort_key="priority")
  28
  29         # Constants
  30         self.max_priortiy = 255
  31         self.min_priortiy = 0
  32
  33         # Runtime info
  34         self.program_name = "dumbwifi"
  35         self.configfile = "/etc/%s.conf" % self.program_name
  36         self.pidfile = "/var/run/%s.pids" % self.program_name
  37
  38         # Logging
  39         self.logfile = "/var/log/%s" % self.program_name
  40         self.logfile_size = 10*(1024**2)
  41
  42         # Streams
  43         self.had_ip = False
  44         self.have_ip = False
  45
  46         # External executables
  47         self.tools = [
  48             "dhclient",
  49             "dhcpcd",
  50             "ifconfig",
  51             "iwconfig",
  52             "iwlist",
  53             "mii-tool",
  54             "pgrep",
  55             "ps",
  56             "tail",
  57             ]
  58
  59         for cmd_name in self.tools:
  60             var_name = cmd_name.replace("-", "_")
  61             value = system.os_which(cmd_name)
  62             self.__dict__[var_name] = value
  63
  64     def init_config(self):
  65         # reset program_name to match the executable that is running
  66         # without disturbing the already instantiated vars using program_name
  67         self.program_name = os.path.basename(sys.argv[0])
  68
  69         lines = system.getfile(self.configfile)
  70         if not lines:
  71             raise Exception, "Failed to read configfile %s" % self.configfile
  72         lines = regex.replaceall("\s*;.*", "", lines).strip() # strip comments
  73
  74         self.interfaces.clear()
  75         self.networks.clear()
  76
  77         for medium in self.interface_media:
  78             ifaces = \
  79                 self.read_config_block(lines, medium, self.interface_fields)
  80             for iface in ifaces:
  81                 iface.medium = medium
  82             self.interfaces.extend(ifaces)
  83
  84         nets = self.read_config_block(lines, "wireless_network", self.network_fields)
  85         self.networks.extend(nets)
  86
  87     def read_config_block(self, s, block_name, fields):
  88         blocks = regex.findall("\[%s\]\s+((?:.+=.+\s*)+)" % (block_name), s)
  89         entries = sdictlist()
  90         for block in blocks:
  91             entry = sdict()
  92             for field in fields:
  93                 value = regex.find1("\s*%s\s*=(.*)" % field, block)
  94                 if value: entry[field] = value.strip()
  95     #       if len(entry) < fieldcount:
  96     #           msg = "In configuration file %s:\n"
  97     #           msg += "Failed to read config for '%s' block: "
  98     #           msg += "got %d field(s), expected %d"
  99     #           err_fg(msg % (configfile, block_name, len(entry), fieldcount))
 100     #           sys.exit(1)
 101             entries.append(entry)
 102         return entries
 103
 104
 105
 106 config = Config()