contrib/fuzz/dirstate_corpus.py
author Pulkit Goyal <7895pulkit@gmail.com>
Tue, 21 Jul 2020 13:58:58 +0530
changeset 45321 dc457177dbc1
parent 43854 ba84a1ae4ae5
child 48966 6000f5b25c9b
permissions -rw-r--r--
localrepo: only use 'bookmarksinstore' requirement if we have 'store' This adds check that whether we have the 'store' requirement or not. If we don't have that, we skip adding the 'bookmarksinstore' requirement and warn user about it. Differential Revision: https://phab.mercurial-scm.org/D8771

from __future__ import absolute_import, print_function

import argparse
import os
import zipfile

ap = argparse.ArgumentParser()
ap.add_argument("out", metavar="some.zip", type=str, nargs=1)
args = ap.parse_args()

reporoot = os.path.normpath(os.path.join(os.path.dirname(__file__), '..', '..'))
dirstate = os.path.join(reporoot, '.hg', 'dirstate')

with zipfile.ZipFile(args.out[0], "w", zipfile.ZIP_STORED) as zf:
    if os.path.exists(dirstate):
        with open(dirstate, 'rb') as f:
            zf.writestr("dirstate", f.read())