Added script to generate notification.json for the emergencynotification mechanism

sitescripts/management/bin/generateHgAuth.py

 # coding: utf-8
 
 # This file is part of the Adblock Plus web scripts,
 # Copyright (C) 2006-2013 Eyeo GmbH
 #
 # Adblock Plus is free software: you can redistribute it and/or modify
 # it under the terms of the GNU General Public License version 3 as
 # published by the Free Software Foundation.
 #
 # Adblock Plus 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 Adblock Plus.  If not, see <http://www.gnu.org/licenses/>.
 
 import os, re, sys, subprocess, tarfile
 from StringIO import StringIO
 from sitescripts.utils import get_config, setupStderr
 
 def generateData(authRepo):
   command = ['hg', '-R', authRepo, 'archive', '-r', 'default', '-t', 'tar', '-p', '.', '-']
   data = subprocess.check_output(command)
 
   users = {}
   repos = []
   tarFile = tarfile.open(mode='r:', fileobj=StringIO(data))
-  fileInfo = tarFile.next()
-  while fileInfo:
+  for fileInfo in tarFile:
     name = fileInfo.name
     if name.startswith('./'):
       name = name[2:]
 
     if fileInfo.type == tarfile.REGTYPE and name.startswith('users/'):
       name = os.path.basename(name).lower()
       options = []
       match = re.search(r'^(.*)\[(.*)\]$', name)
       if match:
         name = match.group(1)
@@ skipping 14 matching lines @@
         elif option == 'trusted':
           user['trusted'] = True
         else:
           print >>sys.stderr, 'Unknown user option: %s' % option
       user['key'] = re.sub(r'\s', '', tarFile.extractfile(fileInfo).read())
       users[name] = user
     elif fileInfo.type == tarfile.REGTYPE and name.startswith('repos/'):
       repos.append(fileInfo)
     elif fileInfo.type == tarfile.REGTYPE and not name.startswith('.'):
       print >>sys.stderr, 'Unrecognized file in the repository: %s' % name
-    fileInfo = tarFile.next()
 
   for fileInfo in repos:
     name = os.path.basename(fileInfo.name).lower()
     repoUsers = tarFile.extractfile(fileInfo).readlines()
     for user in repoUsers:
       user = user.strip()
       if user == '' or user.startswith('#'):
         continue
       if user in users:
         users[user]['repos'].append(name)
       else:
         print >>sys.stderr, 'Unknown user listed for repository %s: %s' % (name, user)
 
   for user in users.itervalues():
     if user['disabled']:
       continue
     yield 'no-pty,environment="HGUSER=%s",environment="HGREPOS=%s" %s %s\n' % (
       user['name'] if not user['trusted'] else '',
       ' '.join(user['repos']),
       'ssh-rsa' if user['keytype'] == 'rsa' else 'ssh-dss',
       user['key']
     )
+  tarFile.close()
 
 def hook(ui, repo, **kwargs):
   setupStderr()
 
-  result = generateData(get_config().get('hg', 'auth_repository'))
+  result = generateData(repo)
 
   file = open(get_config().get('hg', 'auth_file'), 'wb')
   for s in result:
     file.write(s)
   file.close()
 
 if __name__ == '__main__':
-  hook()
+  hook(None, get_config().get('hg', 'auth_repository'))