third-party/pre-commit
1
0
Fork 0
mirror of https://github.com/pre-commit/pre-commit.git synced 2026-04-14 17:41:45 +04:00
Code Issues Projects Releases Packages Wiki Activity Actions

make an exe stub for windows

Browse source
This commit is contained in:
Anthony Sottile 2020-09-28 18:37:10 -07:00
parent bc198b89ca
commit fbd529204b
4 changed files with 35 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

24
testing/zipapp/make
View file

@ -2,6 +2,8 @@
import argparse
import base64
import hashlib
import importlib.resources
import io
import os.path
import shutil
import subprocess
@ -30,11 +32,25 @@ def _check_no_shared_objects(wheeldir: str) -> None:
raise AssertionError(zip_filename, filename)
def _add_shim(dest: str) -> None:
shim = os.path.join(HERE, 'python')
shutil.copy(shim, dest)
bio = io.BytesIO()
with zipfile.ZipFile(bio, 'w') as zipf:
zipf.write(shim, arcname='__main__.py')
with open(os.path.join(dest, 'python.exe'), 'wb') as f:
f.write(importlib.resources.read_binary('distlib', 't32.exe'))
f.write(b'#!py.exe -3\n')
f.write(bio.getvalue())
def _write_cache_key(version: str, wheeldir: str, dest: str) -> None:
cache_hash = hashlib.sha256(f'{version}\n'.encode())
for filename in sorted(os.listdir(wheeldir)):
cache_hash.update(f'{filename}\n'.encode())
with open(os.path.join(HERE, 'fakepython'), 'rb') as f:
with open(os.path.join(HERE, 'python'), 'rb') as f:
cache_hash.update(f.read())
with open(os.path.join(dest, 'CACHE_KEY'), 'wb') as f:
f.write(base64.urlsafe_b64encode(cache_hash.digest()).rstrip(b'='))
@ -62,11 +78,13 @@ def main() -> int:
_msg('validating wheels...')
_check_no_shared_objects(wheeldir)
_msg('adding fakepython / __main__.py...')
shutil.copy(os.path.join(HERE, 'fakepython'), tmpdir)
_msg('adding __main__.py...')
mainfile = os.path.join(tmpdir, '__main__.py')
shutil.copy(os.path.join(HERE, 'entry'), mainfile)
_msg('adding shim...')
_add_shim(tmpdir)
_msg('copying file_lock.py...')
file_lock_py = os.path.join(HERE, '../../pre_commit/file_lock.py')
file_lock_py_dest = os.path.join(tmpdir, 'pre_commit/file_lock.py')