comparison mercurial/hg.py @ 35619:c751b9fdbc40

share: use context manager or utility function to write file

35618:c780e0649e41

    Extensions can wrap this function and write additional entries to
    destrepo/.hg/shared to indicate additional pieces of data to be shared.
    """
    default = defaultpath or sourcerepo.ui.config('paths', 'default')
    if default:
        fp = destrepo.vfs("hgrc", "w", text=True)
        fp.write("[paths]\n")
        fp.write("default = %s\n" % default)
        fp.close()

    with destrepo.wlock():
        if bookmarks:
            fp = destrepo.vfs('shared', 'w')
            fp.write(sharedbookmarks + '\n')
            fp.close()

def _postshareupdate(repo, update, checkout=None):
    """Maybe perform a working directory update after a shared repo is created.

    ``update`` can be a boolean or a revision to update to.

35619:c751b9fdbc40

    Extensions can wrap this function and write additional entries to
    destrepo/.hg/shared to indicate additional pieces of data to be shared.
    """
    default = defaultpath or sourcerepo.ui.config('paths', 'default')
    if default:
        with destrepo.vfs("hgrc", "w", text=True) as fp:
            fp.write("[paths]\n")
            fp.write("default = %s\n" % default)

    with destrepo.wlock():
        if bookmarks:
            destrepo.vfs.write('shared', sharedbookmarks + '\n')

def _postshareupdate(repo, update, checkout=None):
    """Maybe perform a working directory update after a shared repo is created.

    ``update`` can be a boolean or a revision to update to.