comparison mercurial/wireproto.py @ 20923:d771641b7051

wireproto: use decorator for the ubundle command

20922:588e24f3eea3

            raise
        sopener.mustaudit = oldaudit

    return streamres(streamer(repo, entries, total_bytes))

def unbundle(repo, proto, heads):
    their_heads = decodelist(heads)

    def check_heads():
        heads = repo.heads()

        return pushres(r)

    finally:
        fp.close()
        os.unlink(tempname)

commands.update({
    'unbundle': (unbundle, 'heads'),
})

20923:d771641b7051

            raise
        sopener.mustaudit = oldaudit

    return streamres(streamer(repo, entries, total_bytes))

@wireprotocommand('unbundle', 'heads')
def unbundle(repo, proto, heads):
    their_heads = decodelist(heads)

    def check_heads():
        heads = repo.heads()

        return pushres(r)

    finally:
        fp.close()
        os.unlink(tempname)