CrunchBang Linux

filename = "/var/log/apt/history.log"
f = open(filename, "r")
text = f.read()

entries = text.split("\n\n")

i=0
for entry in entries:
    components = entry.split("\n")
    startDate = components[0][12:]
    commandLine = components[1][13:]
    print(str(i)+" -- "+startDate)
    print("      "+commandLine+"\n")
    i=i+1

rbIndex = input("Which entry would you like to roll back?\n")

entry = entries[rbIndex]
components = entry.split("\n")
startDate = components[0][12:]
commandLine = components[1][13:]
actions = []
packages = []
i = 2
while components[i][:10] != "End-Date: ":
    parts = components[i].split(": ")
    actions.append(parts[0])
    packages.append(parts[1].split('), '))
    i=i+1

print("\nTo roll back the install:\n")
print("   "+startDate)
print("   "+commandLine)
print("\nrun the following commands:\n-----------------------------------------\n")

c1 = ''
c2 = ''
c3 = ''

for i in range(0,len(actions)):
    if actions[i] == "Install":
        #handle uninstalls
        c3 = "apt-get remove --purge"
        for package in packages[i]:
            pname = package.split(":")[0]
            c3 = c3 + " " + pname
    if actions[i] == "Upgrade":
        #handle downgrades
        c1 = "apt-get install"
        for package in packages[i]:
            p = package.split(":")
            pname = p[0]
            version = p[1].split('(')[1].split(',')[0]
            c1 = c1 + " " + pname + "=" + version
            
    if actions[i] == "Remove":
        #handle installs
        c2 = "apt-get install"
        for package in packages[i]:
            p = package.split(":")
            pname = p[0]
            version = p[1].split('(')[1].split(')')[0]
            c2 = c2 + " " + pname + "=" + version

if c1 != "":
    print c1
    print
if c2 != "":
    print c2
    print
if c3 != "":
    print c3
    print
print "-----------------------------------------"
print "Note: Some of the packages in step 3 may already have been removed in the"
print "previous steps -- if this happens, just remove them from the command line"
print