hgext/infinitepush/store.py

44061:cbc5755df6bf

# based on bundleheads extension by Gregory Szorc <[e-mail]>

from __future__ import absolute_import

import abc
import hashlib
import os
import subprocess
import tempfile

from mercurial.pycompat import open
from mercurial import (
    node,
    pycompat,
)
from mercurial.utils import procutil

NamedTemporaryFile = tempfile.NamedTemporaryFile


class BundleWriteException(Exception):

    def _filepath(self, filename):
        return os.path.join(self._dirpath(filename), filename)

    def write(self, data):
        filename = node.hex(hashlib.sha1(data).digest())
        dirpath = self._dirpath(filename)

        if not os.path.exists(dirpath):
            os.makedirs(dirpath)

44062:2d49482d0dd4

# based on bundleheads extension by Gregory Szorc <[e-mail]>

from __future__ import absolute_import

import abc
import os
import subprocess
import tempfile

from mercurial.pycompat import open
from mercurial import (
    node,
    pycompat,
)
from mercurial.utils import (
    hashutil,
    procutil,
)

NamedTemporaryFile = tempfile.NamedTemporaryFile


class BundleWriteException(Exception):

    def _filepath(self, filename):
        return os.path.join(self._dirpath(filename), filename)

    def write(self, data):
        filename = node.hex(hashutil.sha1(data).digest())
        dirpath = self._dirpath(filename)

        if not os.path.exists(dirpath):
            os.makedirs(dirpath)