comparison hgext/lfs/wrapper.py @ 35921:47e737d27e01

lfs: factor out a method for extracting the pointer of a single file

This will be useful for filesets, among other things, instead of traversing the
whole context.

35920:2912bed9b0c7

        ctx = repo[r]
        for p in pointersfromctx(ctx).values():
            pointers[p.oid()] = p
    return sorted(pointers.values())

def pointersfromctx(ctx):
    """return a dict {path: pointer} for given single changectx"""
    result = {}
    for f in ctx.files():
        if f not in ctx:
            continue
        fctx = ctx[f]
        if not _islfs(fctx.filelog(), fctx.filenode()):
            continue
        try:
            result[f] = pointer.deserialize(fctx.rawdata())
        except pointer.InvalidPointer as ex:
            raise error.Abort(_('lfs: corrupted pointer (%s@%s): %s\n')
                              % (f, short(ctx.node()), ex))
    return result

def uploadblobs(repo, pointers):
    """upload given pointers from local blobstore"""
    if not pointers:

35921:47e737d27e01

        ctx = repo[r]
        for p in pointersfromctx(ctx).values():
            pointers[p.oid()] = p
    return sorted(pointers.values())

def pointerfromctx(ctx, f):
    """return a pointer for the named file from the given changectx, or None if
    the file isn't LFS."""
    if f not in ctx:
        return None
    fctx = ctx[f]
    if not _islfs(fctx.filelog(), fctx.filenode()):
        return None
    try:
        return pointer.deserialize(fctx.rawdata())
    except pointer.InvalidPointer as ex:
        raise error.Abort(_('lfs: corrupted pointer (%s@%s): %s\n')
                          % (f, short(ctx.node()), ex))

def pointersfromctx(ctx):
    """return a dict {path: pointer} for given single changectx"""
    result = {}
    for f in ctx.files():
        p = pointerfromctx(ctx, f)
        if p:
            result[f] = p
    return result

def uploadblobs(repo, pointers):
    """upload given pointers from local blobstore"""
    if not pointers: