init_env.py

import os
import tarfile
import requests


def init_env():
    # download geckodriver
    print("Downloading geckodriver...")
    r = requests.get('https://github.com/mozilla/geckodriver/releases/download/v0.34.0/geckodriver-v0.34.0-linux64.tar.gz')
    with open('geckodriver-v0.34.0-linux64.tar.gz', 'wb') as f:
        f.write(r.content)
    # extract geckodriver
    tar = tarfile.open('geckodriver-v0.34.0-linux64.tar.gz')
    tar.extractall()
    tar.close()

    # add geckodriver to PATH variable
    os.environ["PATH"] += os.pathsep + os.getcwd()