# -*- coding: utf-8 -*-
#
# Copyright © 2012 Pierre Raybaut
# Licensed under the terms of the MIT License
# (see winpython/__init__.py for details)

"""
WinPython Package Manager

Created on Fri Aug 03 14:32:26 2012
"""
# pypy3 to patch from 'python' to 'pypy3': 379 493 497 627 692 696 743 767 785
from __future__ import print_function

import os
import os.path as osp
from pathlib import Path
import shutil
import re
import sys
import subprocess

# Local imports
from winpython import utils
from winpython.config import DATA_PATH
from winpython.py3compat import configparser as cp

# from former wppm separate script launcher
import textwrap
from argparse import ArgumentParser, HelpFormatter, RawTextHelpFormatter
from winpython import py3compat

from winpython import piptree 

#  import information reader
# importlib_metadata before Python 3.8
try:
    from importlib import metadata as metadata  # Python-3.8
    metadata = metadata.metadata
except:
    try:
        from importlib_metadata import metadata   # <Python-3.8
    except:
        metadata = None  # nothing available
    
# Workaround for installing PyVISA on Windows from source:
os.environ['HOME'] = os.environ['USERPROFILE']

# pep503 defines normalized package names: www.python.org/dev/peps/pep-0503
def normalize(name):
    """ return normalized (unique) name of a package"""
    return re.sub(r"[-_.]+", "-", name).lower()

def get_official_description(name):
    """Extract package Summary description from pypi.org"""
    from winpython import utils
    this = normalize(name)
    this_len = len(this)
    pip_ask = ['pip', 'search', this, '--retries', '0']
    if len(this)<2:  # don't ask stupid things
        return ''
    try:
        #  .run work when .popen fails when no internet
        pip_res = (utils.exec_run_cmd(pip_ask)+'\n').splitlines()
        pip_filter = [l for l in pip_res if this + " (" ==
                      normalize(l[:this_len])+l[this_len:this_len+2]]
        pip_desc = (pip_filter[0][len(this)+1:]).split(" - ", 1)[1] 
        return pip_desc.replace("://", " ")
    except:
        return ''

def get_package_metadata(database, name, gotoWWW=False, update=False):
    """Extract infos (description, url) from the local database"""
    # Note: we could use the PyPI database but this has been written on
    # machine which is not connected to the internet
    # we store only  normalized names now (PEP 503)
    db = cp.ConfigParser()
    db.readfp(open(osp.join(DATA_PATH, database)))
    my_metadata = dict(
        description='',
        url='https://pypi.org/project/' + name,
    )
    for key in my_metadata:
        # wheel replace '-' per '_' in key
        for name2 in (name, normalize(name)):
            try:
                my_metadata[key] = db.get(name2, key)
                break
            except (cp.NoSectionError, cp.NoOptionError):
                pass
    db_desc = my_metadata.get('description')

    if my_metadata.get('description') == '' and metadata:  
        # nothing in package.ini, we look in our installed packages
        try:
            my_metadata['description']=(
                    metadata(name)['Summary']+'\n').splitlines()[0]
        except:
            pass

    if  my_metadata['description'] == '' and gotoWWW:
        # still nothing, try look on pypi
        the_official = get_official_description(name)
        if the_official != '':
            my_metadata['description'] = the_official

    if update == True and db_desc == '' and my_metadata['description'] != '':
        # we add new findings in our packgages.ini list, if it's required
        try:
            db[normalize(name)] = {}
            db[normalize(name)]['description'] = my_metadata['description']
            with open(osp.join(DATA_PATH, database), 'w') as configfile:
                db.write(configfile)
        except:
            pass
    return my_metadata

    
class BasePackage(object):
    def __init__(self, fname):
        self.fname = fname
        self.name = None
        self.version = None
        self.architecture = None
        self.pyversion = None
        self.description = None
        self.url = None

    def __str__(self):
        text = "%s %s" % (self.name, self.version)
        pytext = ""
        if self.pyversion is not None:
            pytext = " for Python %s" % self.pyversion
        if self.architecture is not None:
            if not pytext:
                pytext = " for Python"
            pytext += " %dbits" % self.architecture
        text += "%s\n%s\nWebsite: %s\n[%s]" % (
            pytext,
            self.description,
            self.url,
            osp.basename(self.fname),
        )
        return text

    def is_compatible_with(self, distribution):
        """Return True if package is compatible with distribution in terms of
        architecture and Python version (if applyable)"""
        iscomp = True
        if self.architecture is not None:
            # Source distributions (not yet supported though)
            iscomp = (
                iscomp
                and self.architecture
                == distribution.architecture
            )
        if self.pyversion is not None:
            # Non-pure Python package
            iscomp = (
                iscomp
                and self.pyversion == distribution.version
            )
        return iscomp

    def extract_optional_infos(self, update=False):
        """Extract package optional infos (description, url)
        from the package database"""
        metadata = get_package_metadata(
            'packages.ini', self.name, True, update=update
        )
        for key, value in list(metadata.items()):
            setattr(self, key, value)


class Package(BasePackage):
    def __init__(self, fname, update=False):
        BasePackage.__init__(self, fname)
        self.files = []
        self.extract_infos()
        self.extract_optional_infos(update=update)

    def extract_infos(self):
        """Extract package infos (name, version, architecture)
        from filename (installer basename)"""
        bname = osp.basename(self.fname)
        if bname.endswith(('32.whl', '64.whl')):
            # {name}[-{bloat}]-{version}-{python tag}-{abi tag}-{platform tag}.whl
            # ['sounddevice','0.3.5','py2.py3.cp34.cp35','none','win32']
            # PyQt5-5.7.1-5.7.1-cp34.cp35.cp36-none-win_amd64.whl
            bname2 = bname[:-4].split("-")
            self.name = bname2[0]
            self.version = '-'.join(list(bname2[1:-3]))
            self.pywheel, abi, arch = bname2[-3:]
            self.pyversion = (
                None
            )  # Let's ignore this  self.pywheel
            # wheel arch is 'win32' or 'win_amd64'
            self.architecture = (
                32 if arch == 'win32' else 64
            )
            return
        elif bname.endswith(('.zip', '.tar.gz', '.whl')):
            # distutils sdist
            infos = utils.get_source_package_infos(bname)
            if infos is not None:
                self.name, self.version = infos
                return
        raise NotImplementedError(
            "Not supported package type %s" % bname
        )


class WininstPackage(BasePackage):
    def __init__(self, fname, distribution):
        BasePackage.__init__(self, fname)
        self.logname = None
        self.distribution = distribution
        self.architecture = distribution.architecture
        self.pyversion = distribution.version
        self.extract_infos()
        self.extract_optional_infos()

    def extract_infos(self):
        """Extract package infos (name, version, architecture)"""
        match = re.match(
            r'Remove([a-zA-Z0-9\-\_\.]*)\.exe', self.fname
        )
        if match is None:
            return
        self.name = match.groups()[0]
        self.logname = '%s-wininst.log' % self.name
        fd = open(
            osp.join(
                self.distribution.target, self.logname
            ),
            'U',
        )
        searchtxt = 'DisplayName='
        for line in fd.readlines():
            pos = line.find(searchtxt)
            if pos != -1:
                break
        else:
            return
        fd.close()
        match = re.match(
            r'Python %s %s-([0-9\.]*)'
            % (self.pyversion, self.name),
            line[pos + len(searchtxt) :],
        )
        if match is None:
            return
        self.version = match.groups()[0]

    def uninstall(self):
        """Uninstall package"""
        subprocess.call(
            [self.fname, '-u', self.logname],
            cwd=self.distribution.target,
        )


class Distribution(object):
    def __init__(
        self, target=None, verbose=False, indent=False
    ):
        self.target = target
        self.verbose = verbose
        self.indent = indent

        # if no target path given, take the current python interpreter one
        if self.target is None:
            self.target = os.path.dirname(sys.executable)
        self.to_be_removed = (
            []
        )  # list of directories to be removed later

        self.version, self.architecture = utils.get_python_infos(
            target
        )
        # name of the exe (python.exe or pypy3;exe)
        self.short_exe = osp.basename(utils.get_python_executable(self.target))

    def clean_up(self):
        """Remove directories which couldn't be removed when building"""
        for path in self.to_be_removed:
            try:
                shutil.rmtree(path, onerror=utils.onerror)
            except WindowsError:
                print(
                    "Directory %s could not be removed"
                    % path,
                    file=sys.stderr,
                )

    def remove_directory(self, path):
        """Try to remove directory -- on WindowsError, remove it later"""
        try:
            shutil.rmtree(path)
        except WindowsError:
            self.to_be_removed.append(path)

    def copy_files(
        self,
        package,
        targetdir,
        srcdir,
        dstdir,
        create_bat_files=False,
    ):
        """Add copy task"""
        srcdir = osp.join(targetdir, srcdir)
        if not osp.isdir(srcdir):
            return
        offset = len(srcdir) + len(os.pathsep)
        for dirpath, dirnames, filenames in os.walk(srcdir):
            for dname in dirnames:
                t_dname = osp.join(dirpath, dname)[offset:]
                src = osp.join(srcdir, t_dname)
                dst = osp.join(dstdir, t_dname)
                if self.verbose:
                    print("mkdir: %s" % dst)
                full_dst = osp.join(self.target, dst)
                if not osp.exists(full_dst):
                    os.mkdir(full_dst)
                package.files.append(dst)
            for fname in filenames:
                t_fname = osp.join(dirpath, fname)[offset:]
                src = osp.join(srcdir, t_fname)
                if dirpath.endswith('_system32'):
                    # Files that should be copied in %WINDIR%\system32
                    dst = fname
                else:
                    dst = osp.join(dstdir, t_fname)
                if self.verbose:
                    print("file:  %s" % dst)
                full_dst = osp.join(self.target, dst)
                shutil.move(src, full_dst)
                package.files.append(dst)
                name, ext = osp.splitext(dst)
                if create_bat_files and ext in ('', '.py'):
                    dst = name + '.bat'
                    if self.verbose:
                        print("file:  %s" % dst)
                    full_dst = osp.join(self.target, dst)
                    fd = open(full_dst, 'w')
                    fd.write(
                        """@echo off
python "%~dpn0"""
                        + ext
                        + """" %*"""
                    )
                    fd.close()
                    package.files.append(dst)

    def create_file(self, package, name, dstdir, contents):
        """Generate data file -- path is relative to distribution root dir"""
        dst = osp.join(dstdir, name)
        if self.verbose:
            print("create:  %s" % dst)
        full_dst = osp.join(self.target, dst)
        open(full_dst, 'w').write(contents)
        package.files.append(dst)

    def get_installed_packages(self, update=False):
        """Return installed packages"""

        # Include package installed via pip (not via WPPM)
        wppm = []
        try:
            if (
                os.path.dirname(sys.executable)
                == self.target
            ):
                #  win pip 22.2, we can use pip inspect API
                pip = piptree.pipdata()
                pip_list = pip.pip_list()
            else:
                #  indirect way: we use pip list (for now)
                cmdx = [
                    utils.get_python_executable(self.target), # PyPy !
                    '-m',
                    "pip", "list",
                ]
                pip_list_raw = utils.exec_run_cmd(cmdx).splitlines()
                # pip list gives 2 lines of titles to ignore
                pip_list = [ l.split()  for l in pip_list_raw[2:] ]
            # there are only Packages installed with pip now
            # create pip package list
            wppm = [
                Package(
                    '%s-%s-py2.py3-none-any.whl'
                    % (i[0].replace('-', '_').lower(), i[1])
                , update=update)
                for i in pip_list
            ]
        except:
            pass
        return sorted(
            wppm, key=lambda tup: tup.name.lower()
        )

    def find_package(self, name):
        """Find installed package"""
        for pack in self.get_installed_packages():
            if normalize(pack.name) == normalize(name):
                return pack

    def uninstall_existing(self, package):
        """Uninstall existing package (or package name)"""
        if isinstance(package, str):
            pack = self.find_package(package)
        else:
            pack = self.find_package(package.name)
        if pack is not None:
            self.uninstall(pack)

    def patch_all_shebang(
        self,
        to_movable=True,
        max_exe_size=999999,
        targetdir="",
    ):
        """make all python launchers relatives"""
        import glob
        import os

        for ffname in glob.glob(
            r'%s\Scripts\*.exe' % self.target
        ):
            size = os.path.getsize(ffname)
            if size <= max_exe_size:
                utils.patch_shebang_line(
                    ffname,
                    to_movable=to_movable,
                    targetdir=targetdir,
                )
        for ffname in glob.glob(
            r'%s\Scripts\*.py' % self.target
        ):
            utils.patch_shebang_line_py(
                ffname,
                to_movable=to_movable,
                targetdir=targetdir,
            )

    def install(self, package, install_options=None):
        """Install package in distribution"""
        assert package.is_compatible_with(self)

        # wheel addition
        if package.fname.endswith(
            ('.whl', '.tar.gz', '.zip')
        ):
            self.install_bdist_direct(
                package, install_options=install_options
            )
        self.handle_specific_packages(package)
        # minimal post-install actions
        self.patch_standard_packages(package.name)

    def do_pip_action(
        self, actions=None, install_options=None
    ):
        """Do pip action in a distribution"""
        my_list = install_options
        if my_list is None:
            my_list = []
        my_actions = actions
        if my_actions is None:
            my_actions = []
        executing = osp.join(
            self.target, '..', 'scripts', 'env.bat'
        )
        if osp.isfile(executing):
            complement = [
                r'&&',
                'cd',
                '/D',
                self.target,
                r'&&',
                utils.get_python_executable(self.target), 
                # Before PyPy: osp.join(self.target, 'python.exe')
            ]
            complement += ['-m', 'pip']
        else:
            executing = utils.get_python_executable(self.target)
            # Before PyPy: osp.join(self.target, 'python.exe')
            complement = ['-m', 'pip']
        try:
            fname = utils.do_script(
                this_script=None,
                python_exe=executing,
                architecture=self.architecture,
                verbose=self.verbose,
                install_options=complement
                + my_actions
                + my_list,
            )
        except RuntimeError:
            if not self.verbose:
                print("Failed!")
                raise

    def patch_standard_packages(
        self, package_name='', to_movable=True
    ):
        """patch Winpython packages in need"""
        import filecmp

        # Adpating to PyPy
        if 'pypy3' in osp.basename(utils.get_python_executable(self.target)):
            site_package_place="\\site-packages\\" 
        else:
            site_package_place="\\Lib\\site-packages\\"


        # 'pywin32' minimal post-install (pywin32_postinstall.py do too much)
        if (
            package_name.lower() == "pywin32"
            or package_name == ''
        ):
            origin = self.target + site_package_place + "pywin32_system32"

            destin = self.target
            if osp.isdir(origin):
                for name in os.listdir(origin):
                    here, there = (
                        osp.join(origin, name),
                        osp.join(destin, name),
                    )
                    if not os.path.exists(
                        there
                    ) or not filecmp.cmp(here, there):
                        shutil.copyfile(here, there)
        # 'pip' to do movable launchers (around line 100) !!!!
        # rational: https://github.com/pypa/pip/issues/2328
        if (
            package_name.lower() == "pip"
            or package_name == ''
        ):
            # ensure pip will create movable launchers
            # sheb_mov1 = classic way up to WinPython 2016-01
            # sheb_mov2 = tried  way, but doesn't work for pip (at least)
            sheb_fix = " executable = get_executable()"
            sheb_mov1 = " executable = os.path.join(os.path.basename(get_executable()))"
            sheb_mov2 = " executable = os.path.join('..',os.path.basename(get_executable()))"

            # Adpating to PyPy
            the_place=site_package_place + r"pip\_vendor\distlib\scripts.py"
            print(the_place)
            if to_movable:
                utils.patch_sourcefile(
                    self.target
                    + the_place,
                    sheb_fix,
                    sheb_mov1,
                )
                utils.patch_sourcefile(
                    self.target
                    + the_place,
                    sheb_mov2,
                    sheb_mov1,
                )
            else:
                utils.patch_sourcefile(
                    self.target
                    + the_place,
                    sheb_mov1,
                    sheb_fix,
                )
                utils.patch_sourcefile(
                    self.target
                    + the_place,
                    sheb_mov2,
                    sheb_fix,
                )
            # ensure pip wheel will register relative PATH in 'RECORD' files
            # will be in standard pip 8.0.3
            utils.patch_sourcefile(
                self.target
                + (site_package_place +r"pip\wheel.py"),
                " writer.writerow((f, h, l))",
                " writer.writerow((normpath(f, lib_dir), h, l))",
            )

            # create movable launchers for previous package installations
            self.patch_all_shebang(to_movable=to_movable)
        if (
            package_name.lower() == "spyder"
            or package_name == ''
        ):
            # spyder don't goes on internet without I ask
            utils.patch_sourcefile(
                self.target
                + (
                    site_package_place+r"spyderlib\config\main.py"
                ),
                "'check_updates_on_startup': True,",
                "'check_updates_on_startup': False,",
            )
            utils.patch_sourcefile(
                self.target
                + (
                    site_package_place+r"spyder\config\main.py"
                ),
                "'check_updates_on_startup': True,",
                "'check_updates_on_startup': False,",
            )
        # workaround bad installers
        if package_name.lower() == "numba":
            self.create_pybat(['numba'])
        else:
            self.create_pybat(package_name.lower())

    def create_pybat(
        self,
        names='',
        contents=r"""@echo off
..\python "%~dpn0" %*""",
    ):
        """Create launcher batch script when missing"""

        scriptpy = osp.join(
            self.target, 'Scripts'
        )  # std Scripts of python

        # PyPy has no initial Scipts directory
        if not osp.isdir(scriptpy):
            os.mkdir(scriptpy)

        if not list(names) == names:
            my_list = [
                f
                for f in os.listdir(scriptpy)
                if '.' not in f and f.startswith(names)
            ]
        else:
            my_list = names
        for name in my_list:
            if osp.isdir(scriptpy) and osp.isfile(
                osp.join(scriptpy, name)
            ):
                if not osp.isfile(
                    osp.join(scriptpy, name + '.exe')
                ) and not osp.isfile(
                    osp.join(scriptpy, name + '.bat')
                ):
                    fd = open(
                        osp.join(scriptpy, name + '.bat'),
                        'w',
                    )
                    fd.write(contents)
                    fd.close()

    def handle_specific_packages(self, package):
        """Packages requiring additional configuration"""
        if package.name.lower() in (
            'pyqt4',
            'pyqt5',
            'pyside2',
        ):
            # Qt configuration file (where to find Qt)
            name = 'qt.conf'
            contents = """[Paths]
Prefix = .
Binaries = ."""
            self.create_file(
                package,
                name,
                osp.join(
                    'Lib', 'site-packages', package.name
                ),
                contents,
            )
            self.create_file(
                package,
                name,
                '.',
                contents.replace(
                    '.',
                    './Lib/site-packages/%s' % package.name,
                ),
            )
            # pyuic script
            if package.name.lower() == 'pyqt5':
                # see http://code.activestate.com/lists/python-list/666469/
                tmp_string = r'''@echo off
if "%WINPYDIR%"=="" call "%~dp0..\..\scripts\env.bat"
"%WINPYDIR%\python.exe" -m PyQt5.uic.pyuic %1 %2 %3 %4 %5 %6 %7 %8 %9'''
            else:
                tmp_string = r'''@echo off
if "%WINPYDIR%"=="" call "%~dp0..\..\scripts\env.bat"
"%WINPYDIR%\python.exe" "%WINPYDIR%\Lib\site-packages\package.name\uic\pyuic.py" %1 %2 %3 %4 %5 %6 %7 %8 %9'''

            # PyPy adaption: python.exe or pypy3.exe
            my_exec = osp.basename(utils.get_python_executable(self.target))
            tmp_string = tmp_string.replace('python.exe', my_exec)

            self.create_file(
                package,
                'pyuic%s.bat' % package.name[-1],
                'Scripts',
                tmp_string.replace(
                    'package.name', package.name
                ),
            )
            # Adding missing __init__.py files (fixes Issue 8)
            uic_path = osp.join(
                'Lib', 'site-packages', package.name, 'uic'
            )
            for dirname in ('Loader', 'port_v2', 'port_v3'):
                self.create_file(
                    package,
                    '__init__.py',
                    osp.join(uic_path, dirname),
                    '',
                )

    def _print(self, package, action):
        """Print package-related action text (e.g. 'Installing')
        indicating progress"""
        text = " ".join(
            [action, package.name, package.version]
        )
        if self.verbose:
            utils.print_box(text)
        else:
            if self.indent:
                text = (' ' * 4) + text
            print(text + '...', end=" ")

    def _print_done(self):
        """Print OK at the end of a process"""
        if not self.verbose:
            print("OK")

    def uninstall(self, package):
        """Uninstall package from distribution"""
        self._print(package, "Uninstalling")
        if not package.name == 'pip':
            # trick to get true target (if not current)
            this_executable_path = self.target
            this_exec = utils.get_python_executable(self.target)  # PyPy !
            subprocess.call(
                [
                    this_exec,
                    '-m',
                    'pip',
                    'uninstall',
                    package.name,
                    '-y',
                ],
                cwd=this_executable_path,
            )
            # no more legacy, no package are installed by old non-pip means
        self._print_done()

    def install_bdist_direct(
        self, package, install_options=None
    ):
        """Install a package directly !"""
        self._print(
            package,
            "Installing %s" % package.fname.split(".")[-1],
        )
        try:
            fname = utils.direct_pip_install(
                package.fname,
                python_exe=utils.get_python_executable(self.target),  # PyPy !
                architecture=self.architecture,
                verbose=self.verbose,
                install_options=install_options,
            )
        except RuntimeError:
            if not self.verbose:
                print("Failed!")
                raise
        package = Package(fname)
        self._print_done()

    def install_script(self, script, install_options=None):
        try:
            fname = utils.do_script(
                script,
                python_exe=utils.get_python_executable(self.target), # PyPy3 !
                architecture=self.architecture,
                verbose=self.verbose,
                install_options=install_options,
            )
        except RuntimeError:
            if not self.verbose:
                print("Failed!")
                raise


def main(test=False):
    if test:
        #sbdir = osp.join(
        #    osp.dirname(__file__),
        #    os.pardir,
        #    os.pardir,
        #    os.pardir,
        #    'sandbox',
        #)
        sbdir = str(Path(__file__).parents[0].parent.parent.parent / 'sandbox')
        tmpdir = osp.join(sbdir, 'tobedeleted')

        # fname = osp.join(tmpdir, 'scipy-0.10.1.win-amd64-py2.7.exe')
        fname = osp.join(
            sbdir, 'VTK-5.10.0-Qt-4.7.4.win32-py2.7.exe'
        )
        print(Package(fname))
        sys.exit()
        target = osp.join(
            utils.BASE_DIR,
            'build',
            'winpython-2.7.3',
            'python-2.7.3',
        )
        fname = osp.join(
            utils.BASE_DIR,
            'packages.src',
            'docutils-0.9.1.tar.gz',
        )

        dist = Distribution(target, verbose=True)
        pack = Package(fname)
        print(pack.description)
        # dist.install(pack)
        # dist.uninstall(pack)
    else:
        bold = "\033[1m"
        unbold = "\033[0m"
        registerWinPythonHelp = f'''Register distribution
({bold}experimental{unbold})
This will associate file extensions, icons and
Windows explorer's context menu entries ('Edit with IDLE', ...)
with selected Python distribution in Windows registry.

Shortcuts for all WinPython launchers will be installed
in {unbold}WinPython{unbold} Start menu group (replacing existing
shortcuts).

{bold}Note{unbold}: these actions are similar to those performed 
when installing old Pythons with the official installer before 'py'
.
'''

        unregisterWinPythonHelp = '''Unregister distribution
({bold}experimental{unbold})
This will remove file extensions associations, icons and
Windows explorer's context menu entries ('Edit with IDLE', ...)
with selected Python distribution in Windows registry.

Shortcuts for all WinPython launchers will be removed
from {bold}WinPython{unbold} Start menu group."
.'''

        parser = ArgumentParser(
            description="WinPython Package Manager: view, install, "
            "uninstall or upgrade Python packages on a Windows "
            "Python distribution like WinPython.",
            formatter_class=RawTextHelpFormatter
        )
        parser.add_argument(
            'fname',
            metavar='package',
            nargs='?',
            default='',
            type=str,
            help='path to a Python package, or package name',
        )
        parser.add_argument(
            '-t',
            '--target',
            dest='target',
            default=sys.prefix,
            help='path to target Python distribution '
            '(default: "%s")' % sys.prefix,
        )
        parser.add_argument(
            '-i',
            '--install',
            dest='install',
            action='store_const',
            const=True,
            default=False,
            help='install package (this is the default action)',
        )
        parser.add_argument(
            '-u',
            '--uninstall',
            dest='uninstall',
            action='store_const',
            const=True,
            default=False,
            help='uninstall package',
        )
        parser.add_argument(
            '-r',
            '--reverse-tree',
            dest='pipup',
            action='store_const',
            const=True,
            default=False,
            help='show reverse dependancies of the package',
        )
        parser.add_argument(
            '-p',
            '--package-tree',
            dest='pipdown',
            action='store_const',
            const=True,
            default=False,
            help='show  dependancies of the package',
        )    
        parser.add_argument(
            '-l',
            '--levels_of_depth',
            dest='levels_of_depth', 
            type=int, default=2,
            help='show  l levels_of_depth',
        )
        parser.add_argument(
            '--register',
            dest='registerWinPython', 
            action='store_const',
            const=True,
            default=False,
            help=registerWinPythonHelp,
        )
        parser.add_argument(
            '--unregister',
            dest='unregisterWinPython', 
            action='store_const',
            const=True,
            default=False,
            help=unregisterWinPythonHelp,
        )
        
        args = parser.parse_args()

        if args.install and args.uninstall:
            raise RuntimeError(
                "Incompatible arguments: --install and --uninstall"
            )
        if args.registerWinPython and args.unregisterWinPython:
            raise RuntimeError(
                "Incompatible arguments: --install and --uninstall"
            )
        if args.pipdown:
           pip = piptree.pipdata()
           pack, extra, *other =(args.fname +"[").replace(']','[').split("[")
           pip.down(pack, extra, args.levels_of_depth)
           sys.exit()
        elif args.pipup:
           pip = piptree.pipdata()
           pack, extra, *other =(args.fname +"[").replace(']','[').split("[")
           pip.up(pack, extra, args.levels_of_depth)
           sys.exit()
        if args.registerWinPython:
            print(registerWinPythonHelp)
            if utils.is_python_distribution(args.target):
                dist = Distribution(args.target)
            else:
                raise WindowsError("Invalid Python distribution {args.target}")
            print(f'registering {args.target}')
            print('continue ? Y/N')
            theAnswer=input()
            if theAnswer=='Y':
                from winpython import associate
                associate.register(dist.target)
                sys.exit()
        if args.unregisterWinPython:
            print(unregisterWinPythonHelp)
            if utils.is_python_distribution(args.target):
                dist = Distribution(args.target)
            else:
                raise WindowsError("Invalid Python distribution {args.target}")
            print(f'unregistering {args.target}')
            print('continue ? Y/N')
            theAnswer=input()
            if theAnswer=='Y':
                from winpython import associate
                associate.unregister(dist.target)
                sys.exit()        
        elif not args.install and not args.uninstall:
            args.install = True
        if not osp.isfile(args.fname) and args.install:
            if args.fname=="":
                parser.print_help()
                sys.exit()
            else:
                raise IOError("File not found: %s" % args.fname)
        if utils.is_python_distribution(args.target):
            dist = Distribution(args.target)
            try:
                if args.uninstall:
                    package = dist.find_package(args.fname)
                    dist.uninstall(package)
                else:
                    package = Package(args.fname)
                    if (
                        args.install
                        and package.is_compatible_with(dist)
                    ):
                        dist.install(package)
                    else:
                        raise RuntimeError(
                            "Package is not compatible with Python "
                            "%s %dbit"
                            % (
                                dist.version,
                                dist.architecture,
                            )
                        )
            except NotImplementedError:
                raise RuntimeError(
                    "Package is not (yet) supported by WPPM"
                )
        else:
            raise WindowsError(
                "Invalid Python distribution %s"
                % args.target
            )


if __name__ == '__main__':
    main()
