enter_venv.py

#! /usr/local/bin/python3.6
''' 
Written by: Ashfaq Rahman
This is a simple script that lets you enter into venv by pressing "enter <path/folder containing activate file>.
It looks for activate file in current directory, previous directory, and explores the folders upto depth of 1
for "activate" file. 
'''
import os

try:
    os.chdir(os.sys.argv[1])
except:
    os.chdir('.')

def get_files(list_of_dirs, prev=False):
    if prev:
        current_dir=os.getcwd()
        os.chdir('../')
    l=[]
    for each_dir in list_of_dirs:
        if os.path.isdir(each_dir):
            os.chdir(each_dir)
            for i in os.listdir():
                l.append(os.path.abspath(i))
            os.chdir('../')
    if prev:
        os.chdir(current_dir)
    return l

search_lists = [os.listdir(), os.listdir('../'), get_files(os.listdir()), get_files(os.listdir('../'), True)]

target=None
src_target=None
target_found=False
break_again=False
src_found=False
for index,each_search_list in enumerate(search_lists):
    for index2,path in enumerate(each_search_list):
        if path.split('/')[-1] == 'activate':
            target=index, index2
            target_found=True

        if path.split('/')[-1] == 'src':
            src_target=index, index2
            src_found = True

        if target_found and src_found:
            break_again = True
            break
    if break_again: break

try:
    os.chdir(os.path.abspath(search_lists[src_target[0]][src_target[1]]))
except:
    print('unable to switch to root directory of django, root directory must be named as "src".')

try:
    activate_path=search_lists[target[0]][target[1]]
    print(activate_path)
    os.system('/bin/bash --rcfile '+activate_path)
except:
    print('no activate file found')