Ticket #21540: pyport

#!/usr/bin/env python
#vim:set expandtab:

import os
import sys
#import copy.deepcopy

class Color:
    """ Ansi colors """
    black="\033[0;30m\]"
    red="\033[0;31m\]"
    green="\033[0;32m\]"
    brown="\033[0;33m\]"
    blue="\033[0;34m\]"
    purple="\033[0;35m\]"
    cyan="\033[0;36m\]"
    lightGray="\033[0;37m\]"

    darkGray="\033[1;30m\]"
    lightRed="\033[1;31m\]"
    lightGreen="\033[1;32m\]"
    yellow="\033[1;33m\]"
    lightBlue="\033[1;34m\]"
    lightPurple="\033[1;35m\]"
    lightCyan="\033[1;36m\]"
    white="\033[1;37m\]"

    default="\033[1;0m\]"

class Port:
    """ Collect info from port """
    _name = ""
    _variants = ""
    _installed = False

    _buildDeps = []
    _runDeps = []
    _fetchDeps = []
    _libDeps = []
    _extractDeps = []
    _allDeps = []

    def __init__(self, portName):
        self._name=portName

    def name(self):
        return self._name

    def variants(self):
        return self._variants
    def setVariants(self, x):
        self._variants = x
        self._allDeps += x

    def buildDeps(self):
        return self._buildDeps
    def setBuildDeps(self, x):
        self._buildDeps = x
        self._allDeps += x

    def runDeps(self):
        return self._runDeps
    def setRunDeps(self, x):
        self._runDeps = x
        self._allDeps += x
    
    def fetchDeps(self):
        return self._fetchDeps
    def setFetchDeps(self, x):
        self._fetchDeps = x
        self._allDeps += x

    def libDeps(self):
        return self._libDeps
    def setLibDeps(self, x):
        self._libDeps = x
        self._allDeps += x

    def extractDeps(self):
        return self._extractDeps
    def setExtractDeps(self, x):
        self._extractDeps = x
        self._allDeps += x

    def allDeps(self):
        return self._allDeps

    def isInstalled(self):

        return self._installed
    def setInstalled(self, b=True):
        self._installed = b

def parseInfoLine(portInfoLine):
    return portInfoLine.partition(":")[2].strip().split(',')

def getInfo(portName, allPortsStatic = []):
    """ Store results from 'port info' and 'port installed' in Port obj """
    # Avoid calling port(1) multiple times for same port
    port = next((p for p in allPortsStatic if p.name() == portName), None)
    if port is not None:
        return port
    # else...
    cmd = os.popen("port info " + portName)
    retCmd = cmd.readlines()
    port = Port(portName)
    for line in retCmd:
        if line.startswith("Variants:"):
            port.setVariants(parseInfoLine(line))
        elif line.startswith("Build Dependencies:"):
            port.setBuildDeps(parseInfoLine(line))
        elif line.startswith("Runtime Dependencies:"):
            port.setRunDeps(parseInfoLine(line))
        elif line.startswith("Fetch Dependencies:"):
            port.setFetchDeps(parseInfoLine(line))
        elif line.startswith("Library Dependencies:"):
            port.setLibDeps(parseInfoLine(line))
        elif line.startswith("Extract Dependencies:"):
            port.setExtractDeps(parseInfoLine(line))
    cmd = os.popen("port installed " + portName)
    retCmd = cmd.readlines()
    for line in retCmd:
        if line.startswith("The following ports"):
            port.setInstalled()
            break
    allPortsStatic.append(port)
    return port

def printOutput(port, depth):
    """ Format and display output """
    line = depth*"   " + port.name()
    if port.isInstalled():
        line += "*"
    print line

def makeTree(portName, n=0):
    """ Build dependency tree """
    port = getInfo(portName)
    printOutput(port, n)
    for dep in port.libDeps():
        if dep:
            makeTree(dep, n+1)

def main(argv=None):
    #try:
        arg = sys.argv[1]
        makeTree(arg)
    #except:
    #    sys.exit(2)

if __name__ == "__main__":
    main()