install_requirements.py

# -*- coding: utf-8 -*-
import platform
import re
import subprocess as sb

requirements_file = 'requirements.txt'
poetry_requirements = 'pyproject.toml'
exceptions = ['python']

Python36 = {
    "opencv-python": "4.3.0.38",
    "rich": "12.6.0"
}
def write(text, mode='w'):
    with open(requirements_file, mode) as f:
        f.write(text)

def old_system_package_version(package):
    for pkg, version in Python36.items():
        if package.lower() == pkg.lower():
            return version
    return ''

def create_requirements():
    with open(poetry_requirements) as t:
        for package, version in tomlkit.parse(t.read())['tool']['poetry']['dependencies'].items():
            if package.lower() not in exceptions:
                if isinstance(version, dict):
                    if 'git' in version:
                        write(f"git+{version.get('git')}@{version.get('branch')}", 'a')
                    continue
                elif int(platform.python_version().rsplit(".", 1)[0].replace('.', '')) < 39:
                    version = old_system_package_version(package)
                else:
                    version = re.sub(r'[*^]', '', version)
                write(f"{package}{('==' + version) if version else ''}\n", 'a')

def upgrade_pip():
    sb.call('python -m pip install --upgrade pip', shell=True)

def install_requirements():
    sb.call(f'pip install -r {requirements_file}', shell=True)

def install_tomlkit():
    sb.call('pip install tomlkit==0.11.6', shell=True)

if __name__ == "__main__":
    install_tomlkit()
    import tomlkit
    upgrade_pip()
    write('# -*- coding: utf-8 -*-\n', 'w')
    create_requirements()
    install_requirements()