Python version of https://github.com/rahiel/pwds

pwds.py

#!/usr/bin/python3
import os


def main():
    try:
        MAX_LENGTH = os.environ["PWDS_LENGTH"]
    except KeyError:
        MAX_LENGTH = 16

    p = os.getcwd().replace("/home/" + os.environ["USER"], '~')
    root = True if p[0] == '/' else False
    length = len(p)

    if length > MAX_LENGTH:
        dirs = p.split('/')
        if root:
            dirs = dirs[1:]
        last = len(dirs) - 1
        i = 0
        while length > MAX_LENGTH and i < last:
            if len(dirs[i]) > 1:
                length = length - len(dirs[i][1:])
                dirs[i] = dirs[i][0]
            i += 1
        p = '/'.join(dirs)
        if root:
            p = '/' + p

    print(p)


if __name__ == "__main__":
    main()