mirror of
https://github.com/davidhalter/django-stubs.git
synced 2025-12-11 22:41:55 +08:00
40bf7a562d
* Add missing type for setup_databases `django.test.utils` was seemingly missing the type for `setup_databases`. This change resolves my issue locally. The type was copied directly from `django.test.runner`. * Fix typecheck_tests runner Co-authored-by: Xavier Francisco <xavier.n.francisco@gmail.com>
32 lines
991 B
Python
32 lines
991 B
Python
import shutil
|
|
from typing import Optional
|
|
|
|
from git import RemoteProgress, Repo
|
|
|
|
from scripts.paths import DJANGO_SOURCE_DIRECTORY
|
|
|
|
|
|
class ProgressPrinter(RemoteProgress):
|
|
def line_dropped(self, line: str) -> None:
|
|
print(line)
|
|
|
|
def update(self, op_code, cur_count, max_count=None, message=""):
|
|
print(self._cur_line)
|
|
|
|
|
|
def checkout_django_branch(django_version: str, commit_sha: Optional[str]) -> Repo:
|
|
branch = f"stable/{django_version}.x"
|
|
if DJANGO_SOURCE_DIRECTORY.exists():
|
|
shutil.rmtree(DJANGO_SOURCE_DIRECTORY)
|
|
DJANGO_SOURCE_DIRECTORY.mkdir(exist_ok=True, parents=False)
|
|
repo = Repo.clone_from(
|
|
"https://github.com/django/django.git",
|
|
DJANGO_SOURCE_DIRECTORY,
|
|
progress=ProgressPrinter(),
|
|
branch=branch,
|
|
depth=100,
|
|
)
|
|
if commit_sha and repo.head.commit.hexsha != commit_sha:
|
|
repo.remote("origin").fetch(branch, progress=ProgressPrinter(), depth=100)
|
|
repo.git.checkout(branch)
|