mercurial/lock.py

Fix up test-bad-pull output

-----BEGIN PGP SIGNED MESSAGE-----
Hash: SHA1

Fix up test-bad-pull output

manifest hash: 4ed6670e4f20a0ad9e5b483ccc3003e8bd5c094c
-----BEGIN PGP SIGNATURE-----
Version: GnuPG v1.4.0 (GNU/Linux)

iD8DBQFCwxwdywK+sNU5EO8RAm4RAJ9MuISeQhLqfxo4/w3WLjk18y6gXwCeMwCf
4nvEMnJCPMyk4MSMfzPnCnQ=
=wAOy
-----END PGP SIGNATURE-----

# lock.py - simple locking scheme for mercurial
#
# Copyright 2005 Matt Mackall <[e-mail]>
#
# This software may be used and distributed according to the terms
# of the GNU General Public License, incorporated herein by reference.

import os, time
import util

class LockHeld(Exception):
    pass

class lock:
    def __init__(self, file, wait = 1):
        self.f = file
        self.held = 0
        self.wait = wait
        self.lock()

    def __del__(self):
        self.release()

    def lock(self):
        while 1:
            try:
                self.trylock()
                return 1
            except LockHeld, inst:
                if self.wait:
                    time.sleep(1)
                    continue
                raise inst

    def trylock(self):
        pid = os.getpid()
        try:
            util.makelock(str(pid), self.f)
            self.held = 1
        except:
            raise LockHeld(util.readlock(self.f))

    def release(self):
        if self.held:
            self.held = 0
            try:
                os.unlink(self.f)
            except: pass