announce.py

#!/usr/bin/env python3

import json
import argparse
import codecs
import os
import socket
import subprocess
import re
import netifaces as netif
from cpuinfo import cpuinfo

# Force encoding to UTF-8
import locale                                  # Ensures that subsequent open()s
locale.getpreferredencoding = lambda _=None: 'UTF-8'  # are UTF-8 encoded.

import sys
#sys.stdin = open('/dev/stdin', 'r')
#sys.stdout = open('/dev/stdout', 'w')
#sys.stderr = open('/dev/stderr', 'w')

# Utility functions for the announce.d files
def toUTF8(line):
  return line.decode("utf-8")

def call(cmdnargs):
  output = subprocess.check_output(cmdnargs)
  lines = output.splitlines()
  lines = [toUTF8(line) for line in lines]
  return lines

# Local used functions
def setValue(node,path,value):
  ''' Sets a value inside a complex data dictionary.
      The path Array must have at least one element.
  '''
  key = path[0]
  if len(path) == 1:
    node[key] = value;
  elif key in node:
    setValue(node[key],path[1:],value)
  else:
    node[path[0]] = {}
    setValue(node[key],path[1:],value)

def gateway(batadv_dev):
  output = subprocess.check_output(["batctl","-m",batadv_dev,"gwl","-n"])
  output_utf8 = output.decode("utf-8")
  lines = output_utf8.splitlines()
  gw = None

  for line in lines:
    gw_line = re.match(r"^=> +([0-9a-f:]+) ", line)
    if gw_line:
      gw = gw_line.group(1)

  return gw

def clients(batadv_dev):
  output = subprocess.check_output(["batctl","-m",batadv_dev,"tl","-n"])
  output_utf8 = output.decode("utf-8")
  lines = output_utf8.splitlines()

  count = 0

  for line in lines:
    client_line = re.match(r"^\s\*\s[0-9a-f:]+\s+-\d\s\[[W\.]+\]", line)
    if client_line:
      count += 1

  return count

def addresses(bridge_dev):
  ip_addrs = netif.ifaddresses(bridge_dev)
  ip_list = []

  try:
    for ip6 in netif.ifaddresses(bridge_dev)[netif.AF_INET6]:
      raw6 = ip6['addr'].split('%')
      ip_list.append(raw6[0])
  except:
    pass

  return ip_list

def mac_mesh(fastd_dev,meshmode=False):
  interface = netif.ifaddresses(fastd_dev)
  mesh = []
  mac = None

  try:
    mac = interface[netif.AF_LINK]
    mesh.append(mac[0]['addr'])
  except:
    KeyError

  if meshmode:
    return mesh
  else:
    return mac[0]['addr']

def cpu_info():
  info = cpuinfo.get_cpu_info()
  return info['brand_raw']

parser = argparse.ArgumentParser()

parser.add_argument('-d', '--directory', action='store',
                  help='structure directory',required=True)

parser.add_argument('-b', '--batman', action='store',
                  help='batman-adv device',default='bat0')

parser.add_argument('-f', '--fastd', action='store',
                  help='batman-adv device',default='mesh-vpn')

parser.add_argument('-i', '--interface', action='store',
                  help='freifunk bridge',default='br0')

parser.add_argument('-s', '--sitecode', action='store',
                  help='freifunk site code',default='ffgotham')

args = parser.parse_args()

options = vars(args)

directory = options['directory']
batadv_dev = options['batman']
fastd_dev = options['fastd']
bridge_dev = options['interface']
sitecode = options['sitecode']

data = {}

for dirname, dirnames, filenames in os.walk(directory):
  for filename in filenames:
    if filename[0] != '.':
      relPath = os.path.relpath(dirname + os.sep + filename,directory);
      fh = open(dirname + os.sep + filename,'r', errors='replace')
      source = fh.read()
      fh.close()
      value = eval(source)
      setValue(data,relPath.rsplit(os.sep),value)
print(json.dumps(data))