• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• MultiPlanarUNet-master
  • resources
  • setup.py
  • HISTORY.rst
  • README.md
  • requirements.txt
  • LICENSE.txt
  • MANIFEST.in
  • mpunet
    • evaluate
      • loss_functions.py
      • metrics.py
      • __init__.py
    • callbacks
      • funcs.py
      • validation.py
      • __init__.py
      • callbacks.py
      • mcp_clean.py
    • database
      • db_conn.py
      • default_inserts.py
      • default_tables.py
      • __init__.py
    • utils
      • decorators.py
      • conv_arithmetics.py
      • system.py
      • filters
        • __init__.py
        • filter_activity.py
      • plotting.py
      • __init__.py
      • utils.py
      • fusion
        • fusion_training.py
        • fuse_and_predict.py
        • __init__.py
    • augmentation
      • augmenters.py
      • elastic_deformation.py
      • __init__.py
    • interpolation
      • sample_grid.py
      • view_interpolator.py
      • regular_grid_interpolator.py
      • linalg.py
      • __init__.py
    • models
      • multitask_unet2d.py
      • fusion_model.py
      • unet.py
      • model_init.py
      • __init__.py
      • unet3D.py
    • hyperparameters
      • hparams.py
      • __init__.py
    • __init__.py
    • sequences
      • isotrophic_live_view_sequence_3d.py
      • isotrophic_live_view_sequence.py
      • multi_task_sequence.py
      • base_sequence.py
      • simple_sequences.py
      • patch_sequence_3d.py
      • sliding_patch_sequence_3d.py
      • __init__.py
      • isotrophic_live_view_sequence_2d.py
    • tests
      • integration
        • test_image_pair_with_valid_image.py
        • __init__.py
      • test_all.py
      • __init__.py
      • system
        • __init__.py
        • test_system.py
      • unit
        • __init__.py
      • installation
        • test_imports.py
        • __init__.py
        • test_setup.py
    • preprocessing
      • input_prep.py
      • data_preparation_funcs.py
      • scaling.py
      • __init__.py
    • bin
      • branch.py
      • mp.py
      • version
        • __init__.py
        • version_controller.py
      • train_fusion.py
      • summary.py
      • toy_data.py
      • trim_channels.py
      • predict.py
      • run_tests.py
      • cv_split.py
      • train.py
      • __init__.py
      • predict_3D.py
      • defaults
        • MultiPlanar
          • train_hparams.yaml
        • 3D
          • train_hparams.yaml
        • __init__.py
        • MultiTask
          • task_2.yaml
          • task_1.yaml
          • train_hparams.yaml
      • cv_experiment.py
      • init_project.py
    • train
      • __init__.py
      • utils.py
      • trainer.py
    • image
      • image_queue.py
      • image_pair_loader.py
      • auditor.py
      • __init__.py
      • image_pair.py
    • errors
      • deprecated_warnings.py
      • image_errors.py
      • __init__.py
      • implementation_change_errors.py
    • logging
      • default_logger.py
      • logger.py
      • log_results.py
      • __init__.py

import subprocess
import os
import pkgutil


class VersionController(object):
    def __init__(self, logger=None, package='mpunet'):
        from mpunet.logging.default_logger import ScreenLogger
        self.package_name = package
        self.package_loader = pkgutil.get_loader(package)
        self.logger = logger or ScreenLogger()
        self.git_path = os.path.split(os.path.split(self.package_loader.path)[0])[0]
        self._mem_path = None

    def log_version(self, logger=None):
        logger = logger or self.logger
        logger("{} version: {} ({}, {})".format(self.package_name,
                                                self.version,
                                                self.branch,
                                                self.current_commit))

    def __enter__(self):
        self._mem_path = os.getcwd()
        os.chdir(self.git_path)

    def __exit__(self, exc_type, exc_val, exc_tb):
        os.chdir(self._mem_path)
        self._mem_path = None

    def check_git(self):
        return bool(self.git_query("git status")) and \
               os.path.exists(self.git_path + "/.git")

    def git_query(self, string):
        with self:
            try:
                p = subprocess.Popen(string.split(),
                                     stdout=subprocess.PIPE,
                                     stderr=subprocess.PIPE)
                out, err = p.communicate()
            except FileNotFoundError:
                return None
            if not out:
                return None
            out = out.decode("utf-8").strip(" \n")
        return out

    @property
    def remote_url(self):
        return self.git_query("git config --get remote.origin.url")

    @property
    def version(self):
        module = self.package_loader.load_module()
        return module.__version__

    @property
    def current_commit(self):
        return self.git_query("git rev-parse --short HEAD")

    @property
    def latest_commit_in_branch(self, branch=None):
        branch = branch or self.branch
        url = self.remote_url
        commit = self.git_query("git ls-remote {} refs/heads/{}".format(
            url, branch
        ))
        if commit is None:
            raise OSError("Could not determine latest commit, did not find git.")
        return commit[:7]

    @property
    def branch(self):
        return self.git_query("git symbolic-ref --short HEAD")

    def set_commit(self, commit_id):
        self.git_query("git reset --hard {}".format(str(commit_id)[:7]))

    def set_branch(self, branch):
        self.git_query("git checkout {}".format(branch))

    def set_version(self, version):
        version = str(version).lower().strip(" v")
        self.set_branch("v{}".format(version))
        self.set_commit(self.latest_commit_in_branch)