396861feb1
At Ksplice we used /usr/bin/python because we shipped dependencies as Debian / Red Hat packages, which would be installed against the system Python. We were also very careful to use only Python 2.3 features so that even old system Python would still work. None of that is true at Humbug. We expect users to install dependencies themselves, so it's more likely that the Python in $PATH is correct. On OS X in particular, it's common to have five broken Python installs and there's no expectation that /usr/bin/python is the right one. The files which aren't marked executable are not interesting to run as scripts, so we just remove the line there. (In general it's common to have libraries that can also be executed, to run test cases or whatever, but that's not the case here.) (imported from commit 437d4aee2c6e66601ad3334eefd50749cce2eca6)
67 lines
2.1 KiB
Python
Executable file
67 lines
2.1 KiB
Python
Executable file
#!/usr/bin/env python
|
|
import sys
|
|
from os import path
|
|
import logging
|
|
import time
|
|
import simplejson
|
|
import subprocess
|
|
import unicodedata
|
|
|
|
sys.path.append(path.join(path.dirname(__file__), '..', 'api'))
|
|
import humbug
|
|
|
|
humbug_client = humbug.Client()
|
|
|
|
def fetch_public_streams():
|
|
public_streams = set()
|
|
|
|
try:
|
|
res = humbug_client.get_public_streams()
|
|
if res.get("result") == "success":
|
|
streams = res["streams"]
|
|
else:
|
|
logging.error("Error getting public streams:\n%s" % (res,))
|
|
return None
|
|
except Exception:
|
|
logging.exception("Error getting public streams:")
|
|
return None
|
|
|
|
for stream in streams:
|
|
# Zephyr class names are canonicalized by first applying NFKC
|
|
# normalization and then lower-casing server-side
|
|
canonical_cls = unicodedata.normalize("NFKC", stream).lower().encode("utf-8")
|
|
if canonical_cls in ['security', 'login', 'network', 'ops', 'user_locate',
|
|
'mit',
|
|
'hm_ctl', 'hm_stat', 'zephyr_admin', 'zephyr_ctl']:
|
|
# These zephyr classes cannot be subscribed to by us, due
|
|
# to MIT's Zephyr access control settings
|
|
continue
|
|
|
|
public_streams.add(canonical_cls)
|
|
|
|
return public_streams
|
|
|
|
if __name__ == "__main__":
|
|
log_file = "/home/humbug/sync_public_streams.log"
|
|
logger = logging.getLogger(__name__)
|
|
log_format = "%(asctime)s: %(message)s"
|
|
logging.basicConfig(format=log_format)
|
|
formatter = logging.Formatter(log_format)
|
|
logger.setLevel(logging.DEBUG)
|
|
file_handler = logging.FileHandler(log_file)
|
|
file_handler.setFormatter(formatter)
|
|
logger.addHandler(file_handler)
|
|
|
|
while True:
|
|
time.sleep(15)
|
|
public_streams = fetch_public_streams()
|
|
if public_streams is None:
|
|
continue
|
|
|
|
f = file("/home/humbug/public_streams.tmp", "w")
|
|
f.write(simplejson.dumps(list(public_streams)) + "\n")
|
|
f.close()
|
|
|
|
subprocess.call(["mv", "/home/humbug/public_streams.tmp", "/home/humbug/public_streams"])
|
|
|