import_it.py

#!/usr/bin/env python3.7
import argparse
import sys
from ripgrepy import Ripgrepy

parser = argparse.ArgumentParser()
parser.add_argument('symbol')
parser.add_argument('repository')
args = parser.parse_args()

symbol = args.symbol
repository = args.repository

search = Ripgrepy(f'class {symbol}', repository).json().run()
result = search.as_dict[0]

absolute_path = result['data']['path']['text']

def trim_prefix(s, prefix):
    if s.startswith(prefix):
        return s[len(prefix):]
    return s

relative_path = trim_prefix(absolute_path, repository)

import_path = trim_prefix(relative_path, '/').replace('.py', '').replace('/', '.')

print(f'from {import_path} import {symbol}')