paul@22 | 1 | #!/usr/bin/env python |
paul@22 | 2 | |
paul@22 | 3 | templates = { |
paul@22 | 4 | |
paul@22 | 5 | "Debian" : """ |
paul@22 | 6 | deb http://ftp.%(country)s.debian.org/debian %(codename)s main |
paul@22 | 7 | deb-src http://ftp.%(country)s.debian.org/debian %(codename)s main |
paul@22 | 8 | |
paul@22 | 9 | deb http://security.debian.org/ %(codename)s/updates main |
paul@22 | 10 | deb-src http://security.debian.org/ %(codename)s/updates main |
paul@22 | 11 | """, |
paul@22 | 12 | |
paul@22 | 13 | "Ubuntu" : """ |
paul@22 | 14 | deb http://%(country)s.archive.ubuntu.com/ubuntu/ %(codename)s main universe |
paul@22 | 15 | deb-src http://%(country)s.archive.ubuntu.com/ubuntu/ %(codename)s main universe |
paul@22 | 16 | |
paul@22 | 17 | deb http://%(country)s.archive.ubuntu.com/ubuntu/ %(codename)s-security main universe |
paul@22 | 18 | deb-src http://%(country)s.archive.ubuntu.com/ubuntu/ %(codename)s-security main universe |
paul@22 | 19 | |
paul@22 | 20 | deb http://%(country)s.archive.ubuntu.com/ubuntu/ %(codename)s-updates main universe |
paul@22 | 21 | deb-src http://%(country)s.archive.ubuntu.com/ubuntu/ %(codename)s-updates main universe |
paul@22 | 22 | """ |
paul@22 | 23 | |
paul@22 | 24 | } |
paul@22 | 25 | |
paul@22 | 26 | if __name__ == "__main__": |
paul@22 | 27 | import os, sys |
paul@22 | 28 | |
paul@22 | 29 | if len(sys.argv) < 2: |
paul@22 | 30 | print "Need a filename for the generated file." |
paul@22 | 31 | sys.exit(1) |
paul@22 | 32 | |
paul@22 | 33 | elif len(sys.argv) < 3: |
paul@22 | 34 | print "Need a country to generate source lists." |
paul@22 | 35 | sys.exit(1) |
paul@22 | 36 | |
paul@22 | 37 | filename, country = sys.argv[1:3] |
paul@22 | 38 | distribution = os.environ["USERINSTALL_ID"] |
paul@22 | 39 | codename = os.environ["USERINSTALL_CODENAME"] |
paul@22 | 40 | |
paul@22 | 41 | try: |
paul@22 | 42 | template = templates[distribution] |
paul@22 | 43 | except KeyError: |
paul@22 | 44 | print "Distribution %s not supported." % distribution |
paul@22 | 45 | sys.exit(1) |
paul@22 | 46 | |
paul@22 | 47 | f = open(filename, "w") |
paul@22 | 48 | try: |
paul@22 | 49 | f.write(template % {"codename" : codename, "country" : country}) |
paul@22 | 50 | finally: |
paul@22 | 51 | f.close() |
paul@22 | 52 | |
paul@22 | 53 | # vim: tabstop=4 expandtab shiftwidth=4 |