projects / apt.git / blame
| Commit | Line | Data |
|---|---|---|
| b1a8717a MV |
1 | #!/usr/bin/python |
| 2 | ||
| 3 | from optparse import OptionParser | |
| 4 | ||
| 5 | try: | |
| 6 | import apt_pkg | |
| 7 | except ImportError: | |
| 8 | print "Error importing apt_pkg, is python-apt installed?" | |
| 9 | ||
| 10 | import sys | |
| 11 | import os.path | |
| 12 | ||
| 13 | actions = { "markauto" : 1, | |
| 14 | "unmarkauto": 0 | |
| 15 | } | |
| 16 | ||
| 17 | if __name__ == "__main__": | |
| 18 | apt_pkg.init() | |
| 19 | ||
| 20 | # option parsing | |
| 21 | parser = OptionParser() | |
| 22 | parser.usage = "%prog [options] {markauto|unmarkauto} packages..." | |
| 23 | parser.add_option("-f", "--file", action="store", type="string", | |
| 24 | dest="filename", | |
| 25 | help="read/write a different file") | |
| 26 | parser.add_option("-v", "--verbose", | |
| 27 | action="store_true", dest="verbose", default=False, | |
| 28 | help="print verbose status messages to stdout") | |
| 29 | (options, args) = parser.parse_args() | |
| 30 | if len(args) < 2: | |
| 31 | parser.error("not enough argument") | |
| 32 | ||
| 33 | # get pkgs to change | |
| 34 | if args[0] not in actions.keys(): | |
| 35 | parser.error("first argument must be 'markauto' or 'unmarkauto'") | |
| 36 | pkgs = args[1:] | |
| 37 | action = actions[args[0]] | |
| 38 | ||
| 39 | # get the state-file | |
| 40 | if not options.filename: | |
| 41 | STATE_FILE = apt_pkg.Config.FindDir("Dir::State") + "extended_states" | |
| 42 | else: | |
| 43 | STATE_FILE=options.state_file | |
| 44 | ||
| 45 | # open the statefile | |
| 46 | if os.path.exists(STATE_FILE): | |
| 47 | tagfile = apt_pkg.ParseTagFile(open(STATE_FILE)) | |
| 48 | outfile = open(STATE_FILE+".tmp","w") | |
| 49 | while tagfile.Step(): | |
| 50 | pkgname = tagfile.Section.get("Package") | |
| 51 | autoInst = tagfile.Section.get("Auto-Installed") | |
| 52 | if pkgname in pkgs: | |
| 53 | if options.verbose: | |
| 54 | print "changing %s to %s" % (pkgname,action) | |
| 55 | newsec = apt_pkg.RewriteSection(tagfile.Section, | |
| 56 | [], | |
| 57 | [ ("Auto-Installed",str(action)) ] | |
| 58 | ) | |
| 59 | outfile.write(newsec+"\n") | |
| 60 | else: | |
| 61 | outfile.write(str(tagfile.Section)+"\n") | |
| 62 | # all done, rename the tmpfile | |
| 63 | os.rename(outfile.name, STATE_FILE) |