src/mercurialserver/servelog.py
author Paul Crowley <paul@lshift.net>
Mon, 18 Apr 2011 11:44:36 +0100
changeset 300 31c2c6b383fd
parent 296 6feeef02c057
child 307 9009a37ad45b
permissions -rw-r--r--
Refactor ruleset.py so I can be more confident it's correct
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
242
03d8f07230b3 Tidy up file prologues; move credits to CREDITS
Paul Crowley <paul@lshift.net>
parents: 78
diff changeset
     1
"""
03d8f07230b3 Tidy up file prologues; move credits to CREDITS
Paul Crowley <paul@lshift.net>
parents: 78
diff changeset
     2
Hook to log changesets pushed and pulled
03d8f07230b3 Tidy up file prologues; move credits to CREDITS
Paul Crowley <paul@lshift.net>
parents: 78
diff changeset
     3
"""
54
e2ca551c2822 Add push logging
Paul Crowley <paul@lshift.net>
parents:
diff changeset
     4
e2ca551c2822 Add push logging
Paul Crowley <paul@lshift.net>
parents:
diff changeset
     5
from mercurial.i18n import _
e2ca551c2822 Add push logging
Paul Crowley <paul@lshift.net>
parents:
diff changeset
     6
import mercurial.util
e2ca551c2822 Add push logging
Paul Crowley <paul@lshift.net>
parents:
diff changeset
     7
import mercurial.node
e2ca551c2822 Add push logging
Paul Crowley <paul@lshift.net>
parents:
diff changeset
     8
e2ca551c2822 Add push logging
Paul Crowley <paul@lshift.net>
parents:
diff changeset
     9
import os
e2ca551c2822 Add push logging
Paul Crowley <paul@lshift.net>
parents:
diff changeset
    10
import time
262
675474f5be32 New log file format based on JSON/YAML
Paul Crowley <paul@lshift.net>
parents: 242
diff changeset
    11
import fcntl
296
6feeef02c057 Prefer import json over import simplejson
Paul Crowley <paul@lshift.net>
parents: 295
diff changeset
    12
6feeef02c057 Prefer import json over import simplejson
Paul Crowley <paul@lshift.net>
parents: 295
diff changeset
    13
try: 
6feeef02c057 Prefer import json over import simplejson
Paul Crowley <paul@lshift.net>
parents: 295
diff changeset
    14
    import json
6feeef02c057 Prefer import json over import simplejson
Paul Crowley <paul@lshift.net>
parents: 295
diff changeset
    15
except ImportError: 
6feeef02c057 Prefer import json over import simplejson
Paul Crowley <paul@lshift.net>
parents: 295
diff changeset
    16
    import simplejson as json
6feeef02c057 Prefer import json over import simplejson
Paul Crowley <paul@lshift.net>
parents: 295
diff changeset
    17
78
2a3407a14654 Replaced env vars with Python globals
Paul Crowley <paul@lshift.net>
parents: 67
diff changeset
    18
from mercurialserver import ruleset, changes
2a3407a14654 Replaced env vars with Python globals
Paul Crowley <paul@lshift.net>
parents: 67
diff changeset
    19
54
e2ca551c2822 Add push logging
Paul Crowley <paul@lshift.net>
parents:
diff changeset
    20
def hook(ui, repo, hooktype, node=None, source=None, **kwargs):
55
2661b21688d4 Log pulls too
Paul Crowley <paul@lshift.net>
parents: 54
diff changeset
    21
    if hooktype == 'changegroup':
2661b21688d4 Log pulls too
Paul Crowley <paul@lshift.net>
parents: 54
diff changeset
    22
        op = "push"
2661b21688d4 Log pulls too
Paul Crowley <paul@lshift.net>
parents: 54
diff changeset
    23
    elif hooktype == 'outgoing':
2661b21688d4 Log pulls too
Paul Crowley <paul@lshift.net>
parents: 54
diff changeset
    24
        op = "pull"
2661b21688d4 Log pulls too
Paul Crowley <paul@lshift.net>
parents: 54
diff changeset
    25
    else:
59
f96de2d99f00 give logging module a more specific name
Paul Crowley <paul@lshift.net>
parents: 56
diff changeset
    26
        raise mercurial.util.Abort(_('servelog installed as wrong hook type,'
55
2661b21688d4 Log pulls too
Paul Crowley <paul@lshift.net>
parents: 54
diff changeset
    27
            ' must be changegroup or outgoing but is %s') % hooktype)
262
675474f5be32 New log file format based on JSON/YAML
Paul Crowley <paul@lshift.net>
parents: 242
diff changeset
    28
    log = open(repo.join("mercurial-server.log"), "a+")
54
e2ca551c2822 Add push logging
Paul Crowley <paul@lshift.net>
parents:
diff changeset
    29
    try:
262
675474f5be32 New log file format based on JSON/YAML
Paul Crowley <paul@lshift.net>
parents: 242
diff changeset
    30
        fcntl.flock(log.fileno(), fcntl.LOCK_EX)
675474f5be32 New log file format based on JSON/YAML
Paul Crowley <paul@lshift.net>
parents: 242
diff changeset
    31
        log.seek(0, os.SEEK_END)
675474f5be32 New log file format based on JSON/YAML
Paul Crowley <paul@lshift.net>
parents: 242
diff changeset
    32
        # YAML log file format
295
9741d82e5b1e servelog: use simplejson and the % operator for python versions < 2.6
Dale Wijnand <dale.wijnand@gmail.com>
parents: 262
diff changeset
    33
        log.write("- %s\n" % json.dumps(dict(
262
675474f5be32 New log file format based on JSON/YAML
Paul Crowley <paul@lshift.net>
parents: 242
diff changeset
    34
            timestamp=time.strftime("%Y-%m-%d_%H:%M:%S Z", time.gmtime()),
675474f5be32 New log file format based on JSON/YAML
Paul Crowley <paul@lshift.net>
parents: 242
diff changeset
    35
            op=op,
675474f5be32 New log file format based on JSON/YAML
Paul Crowley <paul@lshift.net>
parents: 242
diff changeset
    36
            key=ruleset.rules.get('user'),
675474f5be32 New log file format based on JSON/YAML
Paul Crowley <paul@lshift.net>
parents: 242
diff changeset
    37
            ssh_connection=os.environ['SSH_CONNECTION'],
675474f5be32 New log file format based on JSON/YAML
Paul Crowley <paul@lshift.net>
parents: 242
diff changeset
    38
            nodes=[mercurial.node.hex(ctx.node())
675474f5be32 New log file format based on JSON/YAML
Paul Crowley <paul@lshift.net>
parents: 242
diff changeset
    39
                for ctx in changes.changes(repo, node)],
295
9741d82e5b1e servelog: use simplejson and the % operator for python versions < 2.6
Dale Wijnand <dale.wijnand@gmail.com>
parents: 262
diff changeset
    40
         )))
54
e2ca551c2822 Add push logging
Paul Crowley <paul@lshift.net>
parents:
diff changeset
    41
    finally:
e2ca551c2822 Add push logging
Paul Crowley <paul@lshift.net>
parents:
diff changeset
    42
        log.close()
296
6feeef02c057 Prefer import json over import simplejson
Paul Crowley <paul@lshift.net>
parents: 295
diff changeset
    43