|
1 |
| -import git_sim.git_sim as gs |
2 |
| -import os, sys |
3 | 1 | import argparse
|
| 2 | +import datetime |
| 3 | +import os |
4 | 4 | import pathlib
|
5 |
| -import time, datetime |
| 5 | +import subprocess |
| 6 | +import sys |
| 7 | +import time |
| 8 | +from argparse import Namespace |
| 9 | +from typing import Type |
| 10 | + |
6 | 11 | import cv2
|
7 | 12 | import git
|
8 |
| -import subprocess |
9 |
| -from manim import config, WHITE |
| 13 | +from manim import WHITE, config |
10 | 14 | from manim.utils.file_ops import open_file as open_media_file
|
11 | 15 |
|
| 16 | +from git_sim.git_sim_add import GitSimAdd |
| 17 | +from git_sim.git_sim_base_command import GitSimBaseCommand |
| 18 | +from git_sim.git_sim_branch import GitSimBranch |
| 19 | +from git_sim.git_sim_cherrypick import GitSimCherryPick |
| 20 | +from git_sim.git_sim_commit import GitSimCommit |
| 21 | +from git_sim.git_sim_log import GitSimLog |
| 22 | +from git_sim.git_sim_merge import GitSimMerge |
| 23 | +from git_sim.git_sim_rebase import GitSimRebase |
| 24 | +from git_sim.git_sim_reset import GitSimReset |
| 25 | +from git_sim.git_sim_restore import GitSimRestore |
| 26 | +from git_sim.git_sim_revert import GitSimRevert |
| 27 | +from git_sim.git_sim_stash import GitSimStash |
| 28 | +from git_sim.git_sim_status import GitSimStatus |
| 29 | +from git_sim.git_sim_tag import GitSimTag |
| 30 | + |
| 31 | + |
| 32 | +def get_scene_for_command(args: Namespace) -> Type[GitSimBaseCommand]: |
| 33 | + |
| 34 | + if args.subcommand == "log": |
| 35 | + return GitSimLog |
| 36 | + elif args.subcommand == "status": |
| 37 | + return GitSimStatus |
| 38 | + elif args.subcommand == "add": |
| 39 | + return GitSimAdd |
| 40 | + elif args.subcommand == "restore": |
| 41 | + return GitSimRestore |
| 42 | + elif args.subcommand == "commit": |
| 43 | + return GitSimCommit |
| 44 | + elif args.subcommand == "stash": |
| 45 | + return GitSimStash |
| 46 | + elif args.subcommand == "branch": |
| 47 | + return GitSimBranch |
| 48 | + elif args.subcommand == "tag": |
| 49 | + return GitSimTag |
| 50 | + elif args.subcommand == "reset": |
| 51 | + return GitSimReset |
| 52 | + elif args.subcommand == "revert": |
| 53 | + return GitSimRevert |
| 54 | + elif args.subcommand == "merge": |
| 55 | + return GitSimMerge |
| 56 | + elif args.subcommand == "rebase": |
| 57 | + return GitSimRebase |
| 58 | + elif args.subcommand == "cherry-pick": |
| 59 | + return GitSimCherryPick |
| 60 | + |
| 61 | + raise NotImplementedError(f"command '{args.subcommand}' is not yet implemented.") |
| 62 | + |
12 | 63 |
|
13 | 64 | def main():
|
14 | 65 | parser = argparse.ArgumentParser(
|
@@ -257,7 +308,8 @@ def main():
|
257 | 308 | if args.light_mode:
|
258 | 309 | config.background_color = WHITE
|
259 | 310 |
|
260 |
| - scene = gs.GitSim(args) |
| 311 | + scene_class = get_scene_for_command(args=args) |
| 312 | + scene = scene_class(args=args) |
261 | 313 | scene.render()
|
262 | 314 |
|
263 | 315 | if args.video_format == "webm":
|
|
0 commit comments