• Home
  • Categories
  • WIKI
  • LANGUAGE: en
    • 求3个文件
    deepin Talks 1219 views · 4 replies ·
    Tofloor
    poster avatar
    Bluek404
    deepin
    2014-05-17 04:30
    Author
    求发
    /usr/bin/add-apt-repository
    /usr/lib/python3/dist-packages/softwareproperties/SoftwareProperties.py
    /usr/lib/python3/dist-packages/aptsources/distro.py
    这三个文件里的内容
    直接复制粘贴回复就行了
    万分感谢
    可以直接用
    1. gedit /usr/bin/add-apt-repository /usr/lib/python3/dist-packages/softwareproperties/SoftwareProperties.py /usr/lib/python3/dist-packages/aptsources/distro.py
    Copy the Code
    这条命令打开
    Reply Favorite View the author
    All Replies
    foundix
    deepin
    2014-05-17 05:23
    #1
    /usr/bin/add-apt-repository
    1. #! /usr/bin/python3
    3. from __future__ import print_function
    5. import io
    6. import os
    7. import sys
    8. import gettext
    9. import locale
    11. from softwareproperties.SoftwareProperties import SoftwareProperties, shortcut_handler
    12. from softwareproperties.shortcuts import ShortcutException
    13. from softwareproperties.ppa import DEFAULT_KEYSERVER
    14. import aptsources
    15. from aptsources.sourceslist import SourceEntry
    16. from optparse import OptionParser
    17. from gettext import gettext as _
    19. if __name__ == "__main__":
    20.     # Force encoding to UTF-8 even in non-UTF-8 locales.
    21.     sys.stdout = io.TextIOWrapper(
    22.         sys.stdout.detach(), encoding="UTF-8", line_buffering=True)
    24.     try:
    25.         locale.setlocale(locale.LC_ALL, "")
    26.     except:
    27.         pass
    28.     gettext.textdomain("software-properties")
    29.     usage = """Usage: %prog
    31. %prog is a script for adding apt sources.list entries.
    32. It can be used to add any repository and also provides a shorthand
    33. syntax for adding a Launchpad PPA (Personal Package Archive)
    34. repository.
    36. - The apt repository source line to add. This is one of:
    37.   a complete apt line in quotes,
    38.   a repo url and areas in quotes (areas defaults to 'main')
    39.   a PPA shortcut.
    40.   a distro component
    42.   Examples:
    43.     apt-add-repository 'deb http://myserver/path/to/repo stable myrepo'
    44.     apt-add-repository 'http://myserver/path/to/repo myrepo'
    45.     apt-add-repository 'https://packages.medibuntu.org free non-free'
    46.     apt-add-repository http://extras.ubuntu.com/ubuntu
    47.     apt-add-repository ppa:user/repository
    48.     apt-add-repository multiverse
    50. If --remove is given the tool will remove the given sourceline from your
    51. sources.list
    52. """
    53.     parser = OptionParser(usage)
    54.     # FIXME: provide a --sources-list-file= option that
    55.     #        puts the line into a specific file in sources.list.d
    56.     parser.add_option ("-m", "--massive-debug", action="store_true",
    57.         dest="massive_debug", default=False,
    58.         help=_("Print a lot of debug information to the command line"))
    59.     parser.add_option("-r", "--remove", action="store_true",
    60.         dest="remove", default=False,
    61.         help=_("remove repository from sources.list.d directory"))
    62.     parser.add_option("-k", "--keyserver",
    63.         dest="keyserver", default=DEFAULT_KEYSERVER,
    64.         help=_("URL of keyserver. Default: %default"))
    65.     parser.add_option("-s", "--enable-source", action="store_true",
    66.         dest="enable_source", default=False,
    67.         help=_("Allow downloading of the source packages from the repository"))
    68.     parser.add_option("-y", "--yes", action="store_true",
    69.         dest="assume_yes", default=False,
    70.         help=_("Assume yes to all queries"))
    71.     (options, args) = parser.parse_args()
    73.     if os.geteuid() != 0:
    74.         print(_("Error: must run as root"))
    75.         sys.exit(1)
    77.     if len(args) == 0:
    78.         print(_("Error: need a repository as argument"))
    79.         sys.exit(1)
    80.     elif len(args) > 1:
    81.         print(_("Error: need a single repository as argument"))
    82.         sys.exit(1)
    84.     # force new ppa file to be 644 (LP: #399709)
    85.     os.umask(0o022)
    87.     # get the line
    88.     line = args[0]
    90.     # add it
    91.     sp = SoftwareProperties(options=options)
    92.     distro = aptsources.distro.get_distro()
    93.     distro.get_sources(sp.sourceslist)
    95.     # check if its a component that should be added/removed
    96.     components = [comp.name for comp in distro.source_template.components]
    97.     if line in components:
    98.         if options.remove:
    99.             if line in distro.enabled_comps:
    100.                 distro.disable_component(line)
    101.                 print(_("'%s' distribution component disabled for all sources.") % line)
    102.             else:
    103.                 print(_("'%s' distribution component is already disabled for all sources.") % line)
    104.                 sys.exit(0)
    105.         else:
    106.             if line not in distro.enabled_comps:
    107.                 distro.enable_component(line)
    108.                 print(_("'%s' distribution component enabled for all sources.") % line)
    109.             else:
    110.                 print(_("'%s' distribution component is already enabled for all sources.") % line)
    111.                 sys.exit(0)
    112.         sp.sourceslist.save()
    113.         sys.exit(0)
    115.     # this wasn't a component name ('multiverse', 'backports'), so its either
    116.     # a actual line to be added or a shortcut.
    117.     try:
    118.         shortcut = shortcut_handler(line)
    119.     except ShortcutException as e:
    120.         print(e)
    121.         sys.exit(1)
    123.     # display more information about the shortcut / ppa info
    124.     if not options.assume_yes and shortcut.should_confirm():
    125.         try:
    126.             info = shortcut.info()
    127.         except ShortcutException as e:
    128.             print(e)
    129.             sys.exit(1)
    131.         print(" %s" % (info["description"] or ""))
    132.         print(_(" More info: %s") % str(info["web_link"]))
    133.         if (sys.stdin.isatty() and
    134.             not "FORCE_ADD_APT_REPOSITORY" in os.environ):
    135.             if options.remove:
    136.                 print(_("Press [ENTER] to continue or ctrl-c to cancel removing it"))
    137.             else:
    138.                 print(_("Press [ENTER] to continue or ctrl-c to cancel adding it"))
    139.             sys.stdin.readline()
    142.     if options.remove:
    143.         try:
    144.             (line, file) = shortcut.expand(sp.distro.codename)
    145.         except ShortcutException as e:
    146.             print(e)
    147.             sys.exit(1)
    148.         deb_line = sp.expand_http_line(line)
    149.         debsrc_line = 'deb-src' + deb_line[3:]
    150.         deb_entry = SourceEntry(deb_line, file)
    151.         debsrc_entry = SourceEntry(debsrc_line, file)
    152.         try:
    153.             sp.remove_source(deb_entry)
    154.         except ValueError:
    155.             print(_("Error: '%s' doesn't exist in a sourcelist file") % deb_line)
    156.         try:
    157.             sp.remove_source(debsrc_entry)
    158.         except ValueError:
    159.             print(_("Error: '%s' doesn't exist in a sourcelist file") % debsrc_line)
    161.     else:
    162.         try:
    163.             if not sp.add_source_from_shortcut(shortcut, options.enable_source):
    164.                 print(_("Error: '%s' invalid") % line)
    165.                 sys.exit(1)
    166.         except ShortcutException as e:
    167.             print(e)
    168.             sys.exit(1)
    169.         
    170.         sp.sourceslist.save()
    Copy the Code
    Reply View the author
    foundix
    deepin
    2014-05-17 05:24
    #2
    /usr/lib/python3/dist-packages/softwareproperties/SoftwareProperties.py
    1. #  software-properties backend
    2. #
    3. #  Copyright (c) 2004-2007 Canonical Ltd.
    4. #                2004-2005 Michiel Sikkes
    5. #
    6. #  Author: Michiel Sikkes
    7. #          Michael Vogt
    8. #          Sebastian Heinlein
    9. #
    10. #  This program is free software; you can redistribute it and/or
    11. #  modify it under the terms of the GNU General Public License as
    12. #  published by the Free Software Foundation; either version 2 of the
    13. #  License, or (at your option) any later version.
    14. #
    15. #  This program is distributed in the hope that it will be useful,
    16. #  but WITHOUT ANY WARRANTY; without even the implied warranty of
    17. #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    18. #  GNU General Public License for more details.
    19. #
    20. #  You should have received a copy of the GNU General Public License
    21. #  along with this program; if not, write to the Free Software
    22. #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
    23. #  USA
    25. from __future__ import absolute_import, print_function
    27. import apt_pkg
    28. import copy
    29. from hashlib import md5
    30. import re
    31. import os
    32. import glob
    33. import shutil
    34. import threading
    35. import atexit
    36. import tempfile
    37. try:
    38.   from string import maketrans
    39. except ImportError:
    40.   maketrans = str.maketrans
    41. import stat
    43. from tempfile import NamedTemporaryFile
    44. from xml.sax.saxutils import escape
    45. try:
    46.   from configparser import ConfigParser
    47. except ImportError:
    48.   from ConfigParser import ConfigParser
    49. from gettext import gettext as _
    51. import aptsources
    52. import aptsources.distro
    53. import softwareproperties
    55. from .AptAuth import AptAuth
    56. from aptsources.sourceslist import SourcesList, SourceEntry
    57. from . import shortcuts
    58. from . import ppa
    59. from . import cloudarchive
    61. _SHORTCUT_FACTORIES = [
    62.     ppa.shortcut_handler,
    63.     cloudarchive.shortcut_handler,
    64.     shortcuts.shortcut_handler,
    65. ]
    68. class SoftwareProperties(object):
    70.   # known (whitelisted) channels
    71.   CHANNEL_PATH="/usr/share/app-install/channels/"
    73.   # release upgrades policy
    74.   RELEASE_UPGRADES_CONF = "/etc/update-manager/release-upgrades"
    75.   #RELEASE_UPGRADES_CONF = "/tmp/release-upgrades"
    76.   (
    77.     RELEASE_UPGRADES_NORMAL,
    78.     RELEASE_UPGRADES_LTS,
    79.     RELEASE_UPGRADES_NEVER
    80.   ) = list(range(3))
    81.   release_upgrades_policy_map = {
    82.     RELEASE_UPGRADES_NORMAL : 'normal',
    83.     RELEASE_UPGRADES_LTS    : 'lts',
    84.     RELEASE_UPGRADES_NEVER  : 'never',
    85.   }
    86.   
    87.   def __init__(self, datadir=None, options=None, rootdir="/"):
    88.     """ Provides the core functionality to configure the used software
    89.         repositories, the corresponding authentication keys and
    90.         update automation """
    91.     self.popconfile = rootdir+"/etc/popularity-contest.conf"
    93.     self.rootdir = rootdir
    94.     if rootdir != "/":
    95.       apt_pkg.config.set("Dir", rootdir)
    97.     # FIXME: some saner way is needed here
    98.     if datadir == None:
    99.       datadir = "/usr/share/software-properties/"
    100.     self.options = options
    101.     self.datadir = datadir
    103.     self.sourceslist = SourcesList()
    104.     self.distro = aptsources.distro.get_distro()
    105.    
    106.     self.seen_server = []
    107.     self.modified_sourceslist = False
    109.     self.reload_sourceslist()
    110.     self.backup_sourceslist()
    112.     self.backup_apt_conf()
    114.     # FIXME: we need to store this value in a config option
    115.     #self.custom_mirrors = ["http://adasdwww.de/ubuntu"]
    116.     self.custom_mirrors= []
    118.     # apt-key stuff
    119.     self.apt_key = AptAuth(rootdir=rootdir)
    121.     atexit.register(self.wait_for_threads)
    123.   def wait_for_threads(self):
    124.     " wait for all running threads (PPA key fetchers) to exit "
    125.     for t in threading.enumerate():
    126.       if t.ident != threading.current_thread().ident:
    127.         t.join()
    129.   def backup_apt_conf(self):
    130.     """Backup all apt configuration options"""
    131.     self.apt_conf_backup = {}
    132.     for option in softwareproperties.CONF_MAP.keys():
    133.         value = apt_pkg.config.find_i(softwareproperties.CONF_MAP[option])
    134.         self.apt_conf_backup[option] = value
    136.   def restore_apt_conf(self):
    137.     """Restore the stored apt configuration"""
    138.     for option in self.apt_conf_backup.keys():
    139.         apt_pkg.config.set(softwareproperties.CONF_MAP[option],
    140.                            str(self.apt_conf_backup[option]))
    141.     self.write_config()
    143.   def get_update_automation_level(self):
    144.     """ Parse the apt cron configuration. Try to fit a predefined use case
    145.         and return it. Special case: if the user made a custom
    146.         configurtation, that we cannot represent it will return None """
    147.     if apt_pkg.config.find_i(softwareproperties.CONF_MAP["autoupdate"]) > 0:
    148.         # Autodownload
    149.         if apt_pkg.config.find_i(softwareproperties.CONF_MAP["unattended"]) == 1\
    150.            and apt_pkg.config.find_i(softwareproperties.CONF_MAP["autodownload"]) == 1 :
    151.             return softwareproperties.UPDATE_INST_SEC
    152.         elif apt_pkg.config.find_i(softwareproperties.CONF_MAP["autodownload"]) == 1 and  \
    153.              apt_pkg.config.find_i(softwareproperties.CONF_MAP["unattended"]) == 0:
    154.             return softwareproperties.UPDATE_DOWNLOAD
    155.         elif apt_pkg.config.find_i(softwareproperties.CONF_MAP["unattended"]) == 0 and \
    156.              apt_pkg.config.find_i(softwareproperties.CONF_MAP["autodownload"]) == 0:
    157.             return softwareproperties.UPDATE_NOTIFY
    158.         else:
    159.             return None
    160.     elif apt_pkg.config.find_i(softwareproperties.CONF_MAP["unattended"]) == 0 and \
    161.          apt_pkg.config.find_i(softwareproperties.CONF_MAP["autodownload"]) == 0:
    162.         return softwareproperties.UPDATE_MANUAL
    163.     else:
    164.         return None
    166.   def set_update_automation_level(self, state):
    167.     """ Set the apt periodic configurtation to the selected
    168.         update automation level. To synchronize the cache update and the
    169.         actual upgrading function, the upgrade function, e.g. unattended,
    170.         will run every day, if enabled. """
    171.     if state == softwareproperties.UPDATE_INST_SEC:
    172.         apt_pkg.config.set(softwareproperties.CONF_MAP["unattended"], str(1))
    173.         apt_pkg.config.set(softwareproperties.CONF_MAP["autodownload"], str(1))
    174.     elif state == softwareproperties.UPDATE_DOWNLOAD:
    175.         apt_pkg.config.set(softwareproperties.CONF_MAP["autodownload"], str(1))
    176.         apt_pkg.config.set(softwareproperties.CONF_MAP["unattended"], str(0))
    177.     elif state == softwareproperties.UPDATE_NOTIFY:
    178.         apt_pkg.config.set(softwareproperties.CONF_MAP["autodownload"], str(0))
    179.         apt_pkg.config.set(softwareproperties.CONF_MAP["unattended"], str(0))
    180.     else:
    181.         apt_pkg.config.set(softwareproperties.CONF_MAP["autoupdate"], str(0))
    182.         apt_pkg.config.set(softwareproperties.CONF_MAP["unattended"], str(0))
    183.         apt_pkg.config.set(softwareproperties.CONF_MAP["autodownload"], str(0))
    184.     self.set_modified_config()
    186.   def set_update_interval(self, days):
    187.       """Set the interval in which we check for available updates"""
    188.       # Only write the key if it has changed
    189.       if not days == apt_pkg.config.find_i(softwareproperties.CONF_MAP["autoupdate"]):
    190.           apt_pkg.config.set(softwareproperties.CONF_MAP["autoupdate"], str(days))
    191.           self.set_modified_config()
    193.   def get_update_interval(self):
    194.     """ Returns the interval of the apt periodic cron job """
    195.     return apt_pkg.config.find_i(softwareproperties.CONF_MAP["autoupdate"])
    197.   def get_release_upgrades_policy(self):
    198.     """
    199.     return the release upgrade policy:
    200.      RELEASE_UPGRADES_NORMAL,
    201.      RELEASE_UPGRADES_LTS,
    202.      RELEASE_UPGRADES_NEVER
    203.     """
    204.     # default (if no option is set) is NORMAL
    205.     if not os.path.exists(self.RELEASE_UPGRADES_CONF):
    206.       return self.RELEASE_UPGRADES_NORMAL
    207.     parser = ConfigParser()
    208.     parser.read(self.RELEASE_UPGRADES_CONF)
    209.     if parser.has_option("DEFAULT","Prompt"):
    210.       type = parser.get("DEFAULT","Prompt").lower()
    211.       for k, v in self.release_upgrades_policy_map.items():
    212.         if v == type:
    213.           return k
    214.     return self.RELEASE_UPGRADES_NORMAL
    216.   def set_release_upgrades_policy(self, i):
    217.     """
    218.     set the release upgrade policy:
    219.      RELEASE_UPGRADES_NORMAL,
    220.      RELEASE_UPGRADES_LTS,
    221.      RELEASE_UPGRADES_NEVER
    222.      """
    223.     # we are note using ConfigParser.write() as it removes comments
    224.     if not os.path.exists(self.RELEASE_UPGRADES_CONF):
    225.       f = open(self.RELEASE_UPGRADES_CONF,"w")
    226.       f.write("[DEFAULT]\nPrompt=%s\n"% self.release_upgrades_policy_map[i])
    227.       return True
    228.     f = open(self.RELEASE_UPGRADES_CONF,"r")
    229.     out = NamedTemporaryFile(mode="w+")
    230.     for line in f:
    231.       line = line.strip()
    232.       if line.lower().startswith("prompt"):
    233.         out.write("Prompt=%s\n" % self.release_upgrades_policy_map[i])
    234.       else:
    235.         out.write(line+"\n")
    236.     out.flush()
    237.     shutil.copymode(self.RELEASE_UPGRADES_CONF, out.name)
    238.     shutil.copy(out.name, self.RELEASE_UPGRADES_CONF)
    239.     return True
    241.   def get_popcon_participation(self):
    242.     """ Will return True if the user wants to participate in the popularity
    243.         contest. Otherwise it will return False. Special case: if no
    244.         popcon is installed it will return False """
    245.     if os.path.exists(self.popconfile):
    246.         lines = open(self.popconfile).read().split("\n")
    247.         active = False
    248.         for line in lines:
    249.             try:
    250.                 (key,value) = line.split("=")
    251.                 if key == "PARTICIPATE" and value.strip('"').lower() == "yes":
    252.                     active = True
    253.             except ValueError:
    254.                 continue
    255.         return active
    256.     else:
    257.         return False
    259.   def set_popcon_pariticipation(self, is_helpful):
    260.     """ Enable or disable the participation in the popularity contest """
    261.     if is_helpful == True:
    262.         value = "yes"
    263.     else:
    264.         value = "no"
    265.     if os.path.exists(self.popconfile):
    266.         # read the current config and replace the corresponding settings
    267.         # FIXME: should we check the other values, too?
    268.         with open(self.popconfile, "r") as popconfile:
    269.             lines = [re.sub(r'^(PARTICIPATE=)(".+?")', '\\1"%s"' % value, line)
    270.                      for line in popconfile]
    271.     else:
    272.         # create a new popcon config file
    273.         m = md5()
    274.         m.update(open("/dev/urandom", "rb").read(1024))
    275.         id = m.hexdigest()
    276.         lines = []
    277.         lines.append("MY_HOSTID=\"%s\"\n" % id)
    278.         lines.append("PARTICIPATE=\"%s\"\n" % str(value))
    279.         lines.append("USE_HTTP=\"yes\"\n")
    280.     open(self.popconfile, "w").writelines(lines)
    282.   def get_source_code_state(self):
    283.     """Return True if all distro componets are also available as
    284.        source code. Otherwise return Flase. Special case: If the
    285.        configuration cannot be represented return None"""
    286.   
    287.     if len(self.distro.source_code_sources) < 1:
    288.         # we don't have any source code sources, so
    289.         # uncheck the button
    290.         self.distro.get_source_code = False
    291.         return False
    292.     else:
    293.         # there are source code sources, so we check the button
    294.         self.distro.get_source_code = True
    295.         # check if there is a corresponding source code source for
    296.         # every binary source. if not set the checkbutton to inconsistent
    297.         templates = {}
    298.         sources = []
    299.         sources.extend(self.distro.main_sources)
    300.         sources.extend(self.distro.child_sources)
    301.         for source in sources:
    302.             if source.template in templates:
    303.                 for comp in source.comps:
    304.                     templates[source.template].add(comp)
    305.             else:
    306.                 templates[source.template] = set(source.comps)
    307.         # add fake http sources for the cdrom, since the sources
    308.         # for the cdrom are only available in the internet
    309.         if len(self.distro.cdrom_sources) > 0:
    310.             templates[self.distro.source_template] = self.distro.cdrom_comps
    311.         for source in self.distro.source_code_sources:
    312.             if source.template not in templates or \
    313.                (source.template in templates and not \
    314.                 (len(set(templates[source.template]) ^ set(source.comps)) == 0\
    315.                  or (len(set(source.comps) ^ self.distro.enabled_comps) == 0))):
    316.                 self.distro.get_source_code = False
    317.                 return None
    318.                 break
    319.     return True
    321.   def print_source_entry(self, source):
    322.     """Print the data of a source entry to the command line"""
    323.     for (label, value) in [("URI:", source.uri),
    324.                            ("Comps:", source.comps),
    325.                            ("Enabled:", not source.disabled),
    326.                            ("Valid:", not source.invalid)]:
    327.         print(" %s %s" % (label, value))
    328.     if source.template:
    329.         for (label, value) in [("MatchURI:", source.template.match_uri),
    330.                                ("BaseURI:", source.template.base_uri)]:
    331.             print(" %s %s" % (label, value))
    332.     print("\n")
    334.   def massive_debug_output(self):
    335.     """Print the complete sources.list"""
    336.     print("START SOURCES.LIST:")
    337.     for source in self.sourceslist:
    338.         print(source.str())
    339.     print("END SOURCES.LIST\n")
    341.   def change_main_download_server(self, server):
    342.     """ change the main download server """
    343.     self.distro.default_server = server
    344.     res = self.distro.change_server(server)
    345.     self.set_modified_sourceslist()
    346.     return res
    348.   def enable_component(self, comp):
    349.     """Enable a component of the distro"""
    350.     self.distro.enable_component(comp)
    351.     self.set_modified_sourceslist()
    353.   def disable_component(self, comp):
    354.     """Disable a component of the distro"""
    355.     self.distro.disable_component(comp)
    356.     self.set_modified_sourceslist()
    358.   def _find_template_from_string(self, name):
    359.     for template in self.distro.source_template.children:
    360.       if template.name == name:
    361.         return template
    363.   def disable_child_source(self, template):
    364.     """Enable a child repo of the distribution main repository"""
    365.     if isinstance(template, str):
    366.       template = self._find_template_from_string(template)
    368.     for source in self.distro.child_sources:
    369.         if source.template == template:
    370.             self.sourceslist.remove(source)
    371.     for source in self.distro.source_code_sources:
    372.         if source.template == template:
    373.             self.sourceslist.remove(source)
    374.     self.set_modified_sourceslist()
    376.   def enable_child_source(self, template):
    377.     """Enable a child repo of the distribution main repository"""
    378.     if isinstance(template, str):
    379.       template = self._find_template_from_string(template)
    381.     # Use the currently selected mirror only if the child source
    382.     # did not override the server
    383.     if template.base_uri == None:
    384.         child_uri = self.distro.default_server
    385.     else:
    386.         child_uri = template.base_uri
    387.     self.distro.add_source(uri=child_uri, dist=template.name)
    388.     self.set_modified_sourceslist()
    390.   def disable_source_code_sources(self):
    391.     """Remove all distro source code sources"""
    392.     sources = []
    393.     sources.extend(self.distro.main_sources)
    394.     sources.extend(self.distro.child_sources)
    395.     # remove all exisiting sources
    396.     for source in self.distro.source_code_sources:
    397.         self.sourceslist.remove(source)
    398.     self.set_modified_sourceslist()
    399.   
    400.   def enable_source_code_sources(self):
    401.     """Enable source code source for all distro sources"""
    402.     sources = []
    403.     sources.extend(self.distro.main_sources)
    404.     sources.extend(self.distro.child_sources)
    406.     # remove all exisiting sources
    407.     for source in self.distro.source_code_sources:
    408.         self.sourceslist.remove(source)
    410.     for source in sources:
    411.         self.sourceslist.add("deb-src",
    412.                              source.uri,
    413.                              source.dist,
    414.                              source.comps,
    415.                              "Added by software-properties",
    416.                              self.sourceslist.list.index(source)+1,
    417.                              source.file)
    418.     for source in self.distro.cdrom_sources:
    419.         self.sourceslist.add("deb-src",
    420.                              self.distro.source_template.base_uri,
    421.                              self.distro.source_template.name,
    422.                              source.comps,
    423.                              "Added by software-properties",
    424.                              self.sourceslist.list.index(source)+1,
    425.                              source.file)
    426.     self.set_modified_sourceslist()
    428.   def backup_sourceslist(self):
    429.     """Store a backup of the source.list in memory"""
    430.     self.sourceslist_backup = []
    431.     for source in self.sourceslist.list:
    432.         source_bkp = SourceEntry(line=source.line,file=source.file)
    433.         self.sourceslist_backup.append(source_bkp)
    434.   
    435.   def _find_source_from_string(self, line):
    436.     # ensure that we have a current list, it might have been changed underneath
    437.     # us
    438.     self.reload_sourceslist()
    440.     for source in self.sourceslist.list:
    441.       if str(source) == line:
    442.         return source
    443.     return None
    445.   def toggle_source_use(self, source):
    446.     """Enable or disable the selected channel"""
    447.     #FIXME cdroms need to disable the comps in the childs and sources
    448.     if isinstance(source, str):
    449.       source = self._find_source_from_string(source)
    450.     source.disabled = not source.disabled
    451.     self.set_modified_sourceslist()
    453.   def replace_source_entry(self, old_entry, new_entry):
    454.     # find and replace, then write
    455.     for (index, entry) in enumerate(self.sourceslist.list):
    456.       if str(entry) == old_entry:
    457.         file = self.sourceslist.list[index].file
    458.         self.sourceslist.list[index] = SourceEntry(new_entry, file)
    459.         self.set_modified_sourceslist()
    460.         return True
    461.     return False
    463.   def revert(self):
    464.     """Revert all settings to the state when software-properties
    465.        was launched"""
    466.     #FIXME: GPG keys are still missing
    467.     self.restore_apt_conf()
    468.     self.revert_sourceslist()
    470.   def revert_sourceslist(self):
    471.     """Restore the source list from the startup of the dialog"""
    472.     self.sourceslist.list = []
    473.     for source in self.sourceslist_backup:
    474.         source_reset = SourceEntry(line=source.line,file=source.file)
    475.         self.sourceslist.list.append(source_reset)
    476.     self.save_sourceslist()
    477.     self.reload_sourceslist()
    479.   def set_modified_sourceslist(self):
    480.     """The sources list was changed and now needs to be saved and reloaded"""
    481.     self.modified_sourceslist = True
    482.     if self.options and self.options.massive_debug:
    483.         self.massive_debug_output()
    484.     self.save_sourceslist()
    485.     self.reload_sourceslist()
    487.   def set_modified_config(self):
    488.     """Write the changed apt configuration to file"""
    489.     self.write_config()
    491.   def render_source(self, source):
    492.     """Render a nice output to show the source in a treeview"""
    493.     if source.template == None:
    494.         if source.comment:
    495.             contents = "%s %s" % (escape(source.comment).strip(),
    496.                                          source.dist)
    497.             # Only show the components if there are more than one
    498.             if len(source.comps) > 1:
    499.                 for c in source.comps:
    500.                     contents += " %s" % c
    501.             if source.type in ("deb-src", "rpm-src"):
    502.                 contents += " %s" % _("(Source Code)")
    503.             contents += "\n%s" % source.uri
    504.         else:
    505.             contents = "%s %s" % (source.uri, source.dist)
    506.             for c in source.comps:
    507.                 contents += " %s" % c
    508.             if source.type in ("deb-src", "rpm-src"):
    509.                 contents += " %s" % _("(Source Code)")
    510.         return contents
    511.     else:
    512.         # try to make use of a corresponding template
    513.         contents = "%s" % source.template.description
    514.         if source.type in ("deb-src", "rpm-src"):
    515.             contents += " (%s)" % _("Source Code")
    516.         if source.comment:
    517.             contents +=" %s" % source.comment
    518.         if source.template.child == False:
    519.             for comp in source.comps:
    520.                 if source.template.has_component(comp):
    521.                     # fixme: move something like this into distinfo.Template
    522.                     #        (why not use a dictionary again?)
    523.                     for c in source.template.components:
    524.                         if c.name == comp:
    525.                             contents += "\n%s" % c.description
    526.                 else:
    527.                     contents += "\n%s" % comp
    528.         return contents
    530.   def get_comparable(self, source):
    531.       """extract attributes to sort the sources"""
    532.       cur_sys = 1
    533.       has_template = 1
    534.       has_comment = 1
    535.       is_source = 1
    536.       revert_numbers = maketrans("0123456789", "9876543210")
    537.       if source.template:
    538.         has_template = 0
    539.         desc = source.template.description
    540.         if source.template.distribution == self.distro:
    541.             cur_sys = 0
    542.       else:
    543.           desc = "%s %s %s" % (source.uri, source.dist, source.comps)
    544.           if source.comment:
    545.               has_comment = 0
    546.       if source.type.find("src"):
    547.           is_source = 0
    548.       return (cur_sys, has_template, has_comment, is_source,
    549.               desc.translate(revert_numbers))
    551.   def get_isv_sources(self):
    552.     """Return a list of sources that are not part of the distribution"""
    553.     isv_sources = []
    554.     for source in self.sourceslist.list:
    555.         if not source.invalid and\
    556.            (source not in self.distro.main_sources and\
    557.             source not in self.distro.cdrom_sources and\
    558.             source not in self.distro.child_sources and\
    559.             source not in self.distro.disabled_sources) and\
    560.            source not in self.distro.source_code_sources:
    561.             isv_sources.append(source)
    562.     return isv_sources
    564.   def get_cdrom_sources(self):
    565.     """Return the list of CDROM based distro sources"""
    566.     return self.distro.cdrom_sources
    567.       
    568.   def get_comp_download_state(self, comp):
    569.     """Return a tuple: the first value describes if a component is enabled
    570.        in the Internet repositories. The second value describes if the
    571.        first value is inconsistent."""
    572.     #FIXME: also return a correct inconsistent value
    573.     return (comp.name in self.distro.download_comps, False)
    575.   def get_comp_child_state(self, template):
    576.     """Return a tuple: the first value describes if a component is enabled
    577.        in one of the child source that matcth the given template.
    578.        The second value describes if the first value is inconsistent."""
    579.     comps = []
    580.     for child in self.distro.child_sources:
    581.         if child.template == template:
    582.             comps.extend(child.comps)
    583.     if len(comps) > 0 and \
    584.         len(self.distro.enabled_comps ^ set(comps)) == 0:
    585.         # All enabled distro components are also enabled for the child source
    586.         return (True, False)
    587.     elif len(comps) > 0 and\
    588.         len(self.distro.enabled_comps ^ set(comps)) != 0:
    589.         # A matching child source does exist but doesn't include all
    590.         # enabled distro components
    591.         return(False, True)
    592.     else:
    593.         # There is no corresponding child source at all
    594.         return (False, False)
    595.   
    596.   def reload_sourceslist(self):
    597.     self.sourceslist.refresh()
    598.     self.sourceslist_visible=[]
    599.     self.distro.get_sources(self.sourceslist)   
    601.   def write_config(self):
    602.     """Write the current apt configuration to file"""
    603.     # update the adept file as well if it is there
    604.     conffiles = [self.rootdir+"/etc/apt/apt.conf.d/10periodic",
    605.                  self.rootdir+"/etc/apt/apt.conf.d/20auto-upgrades",
    606.                  self.rootdir+"/etc/apt/apt.conf.d/15adept-periodic-update"]
    608.     # check (beforehand) if one exists, if not create one
    609.     for f in conffiles:
    610.       if os.path.isfile(f):
    611.         break
    612.     else:
    613.       print("No config found, creating one")
    614.       open(conffiles[0], "w")
    616.     # ensure /etc/cron.daily/apt is executable
    617.     ac = "/etc/cron.daily/apt"
    618.     if os.path.exists(ac):
    619.       perm = os.stat(ac)[stat.ST_MODE]
    620.       if not (perm & stat.S_IXUSR):
    621.         print("file '%s' not executable, fixing" % ac)
    622.         os.chmod(ac, 0o755)
    624.     # now update them
    625.     for periodic in conffiles:
    626.       # read the old content first
    627.       content = []
    628.       if os.path.isfile(periodic):
    629.         content = open(periodic, "r").readlines()
    630.         cnf = apt_pkg.config.subtree("APT::Periodic")
    632.         # then write a new file without the updated keys
    633.         f = open(periodic, "w")
    634.         for line in content:
    635.           for key in cnf.list():
    636.             if line.find("APT::Periodic::%s" % (key)) >= 0:
    637.               break
    638.           else:
    639.             f.write(line)
    641.         # and append the updated keys
    642.         for i in cnf.list():
    643.           f.write("APT::Periodic::%s \"%s\";\n" % (i, cnf.find_i(i)))
    644.         f.close()   
    646.   def save_sourceslist(self):
    647.     """Backup the existing sources.list files and write the current
    648.        configuration"""
    649.     self.sourceslist.backup(".save")
    650.     self.sourceslist.save()
    652.   def _is_line_in_whitelisted_channel(self, srcline):
    653.     """
    654.     helper that checks if a given line is in the source list
    655.     return the channel name or None if not found
    656.     """
    657.     srcentry = SourceEntry(srcline)
    658.     if os.path.exists(self.CHANNEL_PATH):
    659.       for f in glob.glob("%s/*.list" % self.CHANNEL_PATH):
    660.         for line in open(f):
    661.           if line.strip().startswith("#"):
    662.             continue
    663.           if srcentry == SourceEntry(line):
    664.             return os.path.splitext(os.path.basename(f))[0]
    665.     return None
    667.   def check_and_add_key_for_whitelisted_channels(self, srcline):
    668.     # This is maintained for any legacy callers
    669.     return self.check_and_add_key_for_whitelisted_shortcut(shortcut_handler(srcline))
    671.   def check_and_add_key_for_whitelisted_shortcut(self, shortcut):
    672.     """
    673.     helper that adds the gpg key of the channel to the apt
    674.     keyring *if* the channel is in the whitelist
    675.     /usr/share/app-install/channels or it is a public Launchpad PPA.
    676.     """
    677.     (srcline, _fname) = shortcut.expand(codename=self.distro.codename)
    678.     channel = self._is_line_in_whitelisted_channel(srcline)
    679.     if channel:
    680.       keyp = "%s/%s.key" % (self.CHANNEL_PATH, channel)
    681.       self.add_key(keyp)
    683.     cdata = (shortcut.add_key, {'keyserver': (self.options and
    684.                                               self.options.keyserver)})
    685.     def addkey_func():
    686.         func, kwargs = cdata
    687.         func(**kwargs)
    689.     worker = threading.Thread(target=addkey_func)
    690.     worker.start()
    691.     return worker
    693.   def update_interface(self):
    694.     " abstract interface to keep the UI alive "
    696.   def expand_http_line(self, line):
    697.     """
    698.     short cut - this:
    699.       apt-add-repository http://packages.medibuntu.org free non-free
    700.     same as
    701.       apt-add-repository 'deb http://packages.medibuntu.org/ '$(lsb_release -cs)' free non-free'
    702.     """
    703.     if not line.startswith("http"):
    704.       return line
    705.     repo = line.split()[0]
    706.     try:
    707.         areas = line.split(" ",1)[1]
    708.     except IndexError:
    709.         areas = "main"
    710.     line = "deb %s %s %s" % ( repo, self.distro.codename, areas )
    711.     return line
    713.   def add_source_from_line(self, line, enable_source_code=False):
    714.     """
    715.     Add a source for the given line.
    716.     """
    717.     return self.add_source_from_shortcut(
    718.         shortcut=shortcut_handler(line.strip()),
    719.         enable_source_code=enable_source_code)
    721.   def add_source_from_shortcut(self, shortcut, enable_source_code=False):
    722.     """
    723.     Add a source with the given shortcut and add the signing key if the
    724.     site is in whitelist or the shortcut implementer adds it.
    725.     """
    727.     (deb_line, file) = shortcut.expand(codename=self.distro.codename)
    728.     deb_line = self.expand_http_line(deb_line)
    729.     debsrc_entry_type = 'deb-src' if enable_source_code else '# deb-src'
    730.     debsrc_line = debsrc_entry_type + deb_line[3:]
    731.     new_deb_entry = SourceEntry(deb_line, file)
    732.     new_debsrc_entry = SourceEntry(debsrc_line, file)
    733.     if new_deb_entry.invalid or new_debsrc_entry.invalid:
    734.       return False
    735.     worker = self.check_and_add_key_for_whitelisted_shortcut(shortcut)
    736.     self.sourceslist.add(new_deb_entry.type,
    737.                          new_deb_entry.uri,
    738.                          new_deb_entry.dist,
    739.                          new_deb_entry.comps,
    740.                          comment=new_deb_entry.comment,
    741.                          file=new_deb_entry.file,
    742.                          architectures=new_deb_entry.architectures)
    743.     self.sourceslist.add(debsrc_entry_type,
    744.                          new_debsrc_entry.uri,
    745.                          new_debsrc_entry.dist,
    746.                          new_debsrc_entry.comps,
    747.                          comment=new_debsrc_entry.comment,
    748.                          file=new_debsrc_entry.file,
    749.                          architectures=new_debsrc_entry.architectures)
    750.     self.set_modified_sourceslist()
    751.     if worker:
    752.         # wait for GPG key to be downloaded
    753.         worker.join(30)
    754.     return True
    756.   def remove_source(self, source, remove_source_code=True):
    757.     """Remove the given source"""
    758.     if remove_source_code:
    759.       if isinstance(source, str):
    760.         # recall this method giving a SourceEntry as an argument
    761.         source = self._find_source_from_string(source)
    762.         self.remove_source(source, True)
    763.       elif source is not None:
    764.         self.remove_source(source, False)
    765.         # remove the deb-src lines (enabled or not) associated to
    766.         # this source entry
    767.         source = copy.copy(source)
    768.         source.type = 'deb-src'
    769.         source.disabled = True
    770.         self.remove_source(source, False)
    771.         source.disabled = False
    772.         self.remove_source(source, False)
    773.       return
    775.     # first find the source object if we got a string
    776.     if isinstance(source, str):
    777.       source = self._find_source_from_string(source)
    778.     if source is None:
    779.       return
    780.     # if its a sources.list.d file and it contains only a single line
    781.     # (the line that we just remove), then all references to that
    782.     # file are gone and the file is not saved. we work around that
    783.     # here
    784.     if source.file != apt_pkg.config.find_file("Dir::Etc::sourcelist"):
    785.       self.sourceslist.list.append(SourceEntry("", file=source.file))
    786.     try:
    787.       self.sourceslist.remove(source)
    788.     except ValueError:
    789.       # this exception is raised if trying to remove an entry that does
    790.       # not exist. in this case we suppress the error because there's no
    791.       # need to propagate it (the aim of this method is to ensure that
    792.       # the given entries are not listed in the sourceslist)
    793.       pass
    794.     self.set_modified_sourceslist()
    796.   def add_key(self, path):
    797.     """Add a gnupg key to the list of trusted software vendors"""
    798.     if not isinstance(path, str):
    799.       path = str(path) # allows non-ascii filenames
    800.     if not os.path.exists(path):
    801.         return False
    802.     try:
    803.         res = self.apt_key.add(path)
    804.         self.KeysModified()
    805.         return res
    806.     except:
    807.         return False
    809.   def add_key_from_data(self, keydata):
    810.     "Add a gnupg key from a data string (e.g. copy-n-paste)"
    811.     tmp = tempfile.NamedTemporaryFile()
    812.     tmp.write(keydata.encode())
    813.     tmp.flush()
    814.     return self.add_key(tmp.name)
    816.   def remove_key(self, keyid):
    817.     """Remove a gnupg key from the list of trusted software vendors"""
    818.     try:
    819.         self.apt_key.rm(keyid)
    820.         self.KeysModified()
    821.         return True
    822.     except:
    823.         return False
    825.   def update_keys(self):
    826.     """ Run apt-key update """
    827.     try:
    828.       self.apt_key.update()
    829.       self.KeysModified()
    830.       return True
    831.     except:
    832.       return False
    835. def shortcut_handler(shortcut):
    836.     for factory in _SHORTCUT_FACTORIES:
    837.         ret = factory(shortcut)
    838.         if ret is not None:
    839.             return ret
    841.     raise shortcuts.ShortcutException("Unable to handle input '%s'" % shortcut)
    844. if __name__ == "__main__":
    845.   sp = SoftwareProperties()
    846.   print(sp.get_release_upgrades_policy())
    847.   sp.set_release_upgrades_policy(0)
    Copy the Code
    Reply View the author
    foundix
    deepin
    2014-05-17 05:24
    #3
    /usr/lib/python3/dist-packages/aptsources/distro.py
    1. #  distro.py - Provide a distro abstraction of the sources.list
    2. #
    3. #  Copyright (c) 2004-2009 Canonical Ltd.
    4. #  Copyright (c) 2006-2007 Sebastian Heinlein
    5. #
    6. #  Authors: Sebastian Heinlein
    7. #           Michael Vogt
    8. #
    9. #  This program is free software; you can redistribute it and/or
    10. #  modify it under the terms of the GNU General Public License as
    11. #  published by the Free Software Foundation; either version 2 of the
    12. #  License, or (at your option) any later version.
    13. #
    14. #  This program is distributed in the hope that it will be useful,
    15. #  but WITHOUT ANY WARRANTY; without even the implied warranty of
    16. #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    17. #  GNU General Public License for more details.
    18. #
    19. #  You should have received a copy of the GNU General Public License
    20. #  along with this program; if not, write to the Free Software
    21. #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
    22. #  USA
    24. import gettext
    25. import logging
    26. import re
    27. import os
    29. from xml.etree.ElementTree import ElementTree
    31. from apt_pkg import gettext as _
    34. class NoDistroTemplateException(Exception):
    35.     pass
    38. class Distribution(object):
    40.     def __init__(self, id, codename, description, release):
    41.         """ Container for distribution specific informations """
    42.         # LSB information
    43.         self.id = id
    44.         self.codename = codename
    45.         self.description = description
    46.         self.release = release
    48.         self.binary_type = "deb"
    49.         self.source_type = "deb-src"
    51.     def get_sources(self, sourceslist):
    52.         """
    53.         Find the corresponding template, main and child sources
    54.         for the distribution
    55.         """
    57.         self.sourceslist = sourceslist
    58.         # corresponding sources
    59.         self.source_template = None
    60.         self.child_sources = []
    61.         self.main_sources = []
    62.         self.disabled_sources = []
    63.         self.cdrom_sources = []
    64.         self.download_comps = []
    65.         self.enabled_comps = []
    66.         self.cdrom_comps = []
    67.         self.used_media = []
    68.         self.get_source_code = False
    69.         self.source_code_sources = []
    71.         # location of the sources
    72.         self.default_server = ""
    73.         self.main_server = ""
    74.         self.nearest_server = ""
    75.         self.used_servers = []
    77.         # find the distro template
    78.         for template in self.sourceslist.matcher.templates:
    79.             if (self.is_codename(template.name) and
    80.                 template.distribution == self.id):
    81.                 #print "yeah! found a template for %s" % self.description
    82.                 #print template.description, template.base_uri, \
    83.                 #    template.components
    84.                 self.source_template = template
    85.                 break
    86.         if self.source_template is None:
    87.             raise NoDistroTemplateException(
    88.                 "Error: could not find a distribution template for %s/%s" %
    89.                 (self.id, self.codename))
    91.         # find main and child sources
    92.         media = []
    93.         comps = []
    94.         cdrom_comps = []
    95.         enabled_comps = []
    96.         #source_code = []
    97.         for source in self.sourceslist.list:
    98.             if (source.invalid == False and
    99.                 self.is_codename(source.dist) and
    100.                 source.template and
    101.                 source.template.official == True and
    102.                 self.is_codename(source.template.name)):
    103.                 #print "yeah! found a distro repo:  %s" % source.line
    104.                 # cdroms need do be handled differently
    105.                 if (source.uri.startswith("cdrom:") and
    106.                     source.disabled == False):
    107.                     self.cdrom_sources.append(source)
    108.                     cdrom_comps.extend(source.comps)
    109.                 elif (source.uri.startswith("cdrom:") and
    110.                       source.disabled == True):
    111.                     self.cdrom_sources.append(source)
    112.                 elif (source.type == self.binary_type and
    113.                       source.disabled == False):
    114.                     self.main_sources.append(source)
    115.                     comps.extend(source.comps)
    116.                     media.append(source.uri)
    117.                 elif (source.type == self.binary_type and
    118.                       source.disabled == True):
    119.                     self.disabled_sources.append(source)
    120.                 elif (source.type == self.source_type
    121.                         and source.disabled == False):
    122.                     self.source_code_sources.append(source)
    123.                 elif (source.type == self.source_type and
    124.                       source.disabled == True):
    125.                     self.disabled_sources.append(source)
    126.             if (source.invalid == False and
    127.                 source.template in self.source_template.children):
    128.                 if (source.disabled == False
    129.                     and source.type == self.binary_type):
    130.                     self.child_sources.append(source)
    131.                 elif (source.disabled == False
    132.                       and source.type == self.source_type):
    133.                     self.source_code_sources.append(source)
    134.                 else:
    135.                     self.disabled_sources.append(source)
    136.         self.download_comps = set(comps)
    137.         self.cdrom_comps = set(cdrom_comps)
    138.         enabled_comps.extend(comps)
    139.         enabled_comps.extend(cdrom_comps)
    140.         self.enabled_comps = set(enabled_comps)
    141.         self.used_media = set(media)
    142.         self.get_mirrors()
    144.     def get_mirrors(self, mirror_template=None):
    145.         """
    146.         Provide a set of mirrors where you can get the distribution from
    147.         """
    148.         # the main server is stored in the template
    149.         self.main_server = self.source_template.base_uri
    151.         # other used servers
    152.         for medium in self.used_media:
    153.             if not medium.startswith("cdrom:"):
    154.                 # seems to be a network source
    155.                 self.used_servers.append(medium)
    157.         if len(self.main_sources) == 0:
    158.             self.default_server = self.main_server
    159.         else:
    160.             self.default_server = self.main_sources[0].uri
    162.         # get a list of country codes and real names
    163.         self.countries = {}
    164.         fname = "/usr/share/xml/iso-codes/iso_3166.xml"
    165.         if os.path.exists(fname):
    166.             et = ElementTree(file=fname)
    167.             # python2.6 compat, the next two lines can get removed
    168.             # once we do not use py2.6 anymore
    169.             if getattr(et, "iter", None) is None:
    170.                 et.iter = et.getiterator
    171.             it = et.iter('iso_3166_entry')
    172.             for elm in it:
    173.                 try:
    174.                     descr = elm.attrib["common_name"]
    175.                 except KeyError:
    176.                     descr = elm.attrib["name"]
    177.                 try:
    178.                     code = elm.attrib["alpha_2_code"]
    179.                 except KeyError:
    180.                     code = elm.attrib["alpha_3_code"]
    181.                 self.countries[code.lower()] = gettext.dgettext('iso_3166',
    182.                                                                 descr)
    184.         # try to guess the nearest mirror from the locale
    185.         self.country = None
    186.         self.country_code = None
    187.         locale = os.getenv("LANG", default="en_UK")
    188.         a = locale.find("_")
    189.         z = locale.find(".")
    190.         if z == -1:
    191.             z = len(locale)
    192.         country_code = locale[a+1:z].lower()
    194.         if mirror_template:
    195.             self.nearest_server = mirror_template % country_code
    197.         if country_code in self.countries:
    198.             self.country = self.countries[country_code]
    199.             self.country_code = country_code
    201.     def _get_mirror_name(self, server):
    202.         ''' Try to get a human readable name for the main mirror of a country
    203.             Customize for different distributions '''
    204.         country = None
    205.         i = server.find("://")
    206.         l = server.find(".archive.ubuntu.com")
    207.         if i != -1 and l != -1:
    208.             country = server[i+len("://"):l]
    209.         if country in self.countries:
    210.             # TRANSLATORS: %s is a country
    211.             return _("Server for %s") % self.countries[country]
    212.         else:
    213.             return("%s" % server.rstrip("/ "))
    215.     def get_server_list(self):
    216.         ''' Return a list of used and suggested servers '''
    218.         def compare_mirrors(mir1, mir2):
    219.             ''' Helper function that handles comaprision of mirror urls
    220.                 that could contain trailing slashes'''
    221.             return re.match(mir1.strip("/ "), mir2.rstrip("/ "))
    223.         # Store all available servers:
    224.         # Name, URI, active
    225.         mirrors = []
    226.         if (len(self.used_servers) < 1 or
    227.             (len(self.used_servers) == 1 and
    228.              compare_mirrors(self.used_servers[0], self.main_server))):
    229.             mirrors.append([_("Main server"), self.main_server, True])
    230.             if self.nearest_server:
    231.                 mirrors.append([self._get_mirror_name(self.nearest_server),
    232.                                 self.nearest_server, False])
    233.         elif (len(self.used_servers) == 1 and not
    234.               compare_mirrors(self.used_servers[0], self.main_server)):
    235.             mirrors.append([_("Main server"), self.main_server, False])
    236.             # Only one server is used
    237.             server = self.used_servers[0]
    239.             # Append the nearest server if it's not already used
    240.             if self.nearest_server:
    241.                 if not compare_mirrors(server, self.nearest_server):
    242.                     mirrors.append([self._get_mirror_name(self.nearest_server),
    243.                                     self.nearest_server, False])
    244.             if server:
    245.                 mirrors.append([self._get_mirror_name(server), server, True])
    247.         elif len(self.used_servers) > 1:
    248.             # More than one server is used. Since we don't handle this case
    249.             # in the user interface we set "custom servers" to true and
    250.             # append a list of all used servers
    251.             mirrors.append([_("Main server"), self.main_server, False])
    252.             if self.nearest_server:
    253.                 mirrors.append([self._get_mirror_name(self.nearest_server),
    254.                                 self.nearest_server, False])
    255.             mirrors.append([_("Custom servers"), None, True])
    256.             for server in self.used_servers:
    257.                 mirror_entry = [self._get_mirror_name(server), server, False]
    258.                 if (compare_mirrors(server, self.nearest_server) or
    259.                     compare_mirrors(server, self.main_server)):
    260.                     continue
    261.                 elif not mirror_entry in mirrors:
    262.                     mirrors.append(mirror_entry)
    264.         return mirrors
    266.     def add_source(self, type=None,
    267.                  uri=None, dist=None, comps=None, comment=""):
    268.         """
    269.         Add distribution specific sources
    270.         """
    271.         if uri is None:
    272.             # FIXME: Add support for the server selector
    273.             uri = self.default_server
    274.         if dist is None:
    275.             dist = self.codename
    276.         if comps is None:
    277.             comps = list(self.enabled_comps)
    278.         if type is None:
    279.             type = self.binary_type
    280.         new_source = self.sourceslist.add(type, uri, dist, comps, comment)
    281.         # if source code is enabled add a deb-src line after the new
    282.         # source
    283.         if self.get_source_code == True and type == self.binary_type:
    284.             self.sourceslist.add(self.source_type, uri, dist, comps, comment,
    285.                                  file=new_source.file,
    286.                                  pos=self.sourceslist.list.index(new_source)+1)
    288.     def enable_component(self, comp):
    289.         """
    290.         Enable a component in all main, child and source code sources
    291.         (excluding cdrom based sources)
    293.         comp:         the component that should be enabled
    294.         """
    295.         comps = set([comp])
    296.         # look for parent components that we may have to add
    297.         for source in self.main_sources:
    298.             for c in source.template.components:
    299.                 if c.name == comp and c.parent_component:
    300.                     comps.add(c.parent_component)
    301.         for c in comps:
    302.             self._enable_component(c)
    304.     def _enable_component(self, comp):
    306.         def add_component_only_once(source, comps_per_dist):
    307.             """
    308.             Check if we already added the component to the repository, since
    309.             a repository could be splitted into different apt lines. If not
    310.             add the component
    311.             """
    312.             # if we don't have that distro, just return (can happen for e.g.
    313.             # dapper-update only in deb-src
    314.             if source.dist not in comps_per_dist:
    315.                 return
    316.             # if we have seen this component already for this distro,
    317.             # return (nothing to do)
    318.             if comp in comps_per_dist[source.dist]:
    319.                 return
    320.             # add it
    321.             source.comps.append(comp)
    322.             comps_per_dist[source.dist].add(comp)
    324.         sources = []
    325.         sources.extend(self.main_sources)
    326.         sources.extend(self.child_sources)
    327.         # store what comps are enabled already per distro (where distro is
    328.         # e.g. "dapper", "dapper-updates")
    329.         comps_per_dist = {}
    330.         comps_per_sdist = {}
    331.         for s in sources:
    332.             if s.type == self.binary_type:
    333.                 if s.dist not in comps_per_dist:
    334.                     comps_per_dist[s.dist] = set()
    335.                 for c in s.comps:
    336.                     comps_per_dist[s.dist].add(c)
    337.         for s in self.source_code_sources:
    338.             if s.type == self.source_type:
    339.                 if s.dist not in comps_per_sdist:
    340.                     comps_per_sdist[s.dist] = set()
    341.                 for c in s.comps:
    342.                     comps_per_sdist[s.dist].add(c)
    344.         # check if there is a main source at all
    345.         if len(self.main_sources) < 1:
    346.             # create a new main source
    347.             self.add_source(comps=["%s"%comp])
    348.         else:
    349.             # add the comp to all main, child and source code sources
    350.             for source in sources:
    351.                 add_component_only_once(source, comps_per_dist)
    353.             for source in self.source_code_sources:
    354.                 add_component_only_once(source, comps_per_sdist)
    356.         # check if there is a main source code source at all
    357.         if self.get_source_code == True:
    358.             if len(self.source_code_sources) < 1:
    359.                 # create a new main source
    360.                 self.add_source(type=self.source_type, comps=["%s"%comp])
    361.             else:
    362.                 # add the comp to all main, child and source code sources
    363.                 for source in self.source_code_sources:
    364.                     add_component_only_once(source, comps_per_sdist)
    366.     def disable_component(self, comp):
    367.         """
    368.         Disable a component in all main, child and source code sources
    369.         (excluding cdrom based sources)
    370.         """
    371.         sources = []
    372.         sources.extend(self.main_sources)
    373.         sources.extend(self.child_sources)
    374.         sources.extend(self.source_code_sources)
    375.         if comp in self.cdrom_comps:
    376.             sources = []
    377.             sources.extend(self.main_sources)
    378.         for source in sources:
    379.             if comp in source.comps:
    380.                 source.comps.remove(comp)
    381.                 if len(source.comps) < 1:
    382.                     self.sourceslist.remove(source)
    384.     def change_server(self, uri):
    385.         ''' Change the server of all distro specific sources to
    386.             a given host '''
    388.         def change_server_of_source(source, uri, seen):
    389.             # Avoid creating duplicate entries
    390.             source.uri = uri
    391.             for comp in source.comps:
    392.                 if [source.uri, source.dist, comp] in seen:
    393.                     source.comps.remove(comp)
    394.                 else:
    395.                     seen.append([source.uri, source.dist, comp])
    396.             if len(source.comps) < 1:
    397.                 self.sourceslist.remove(source)
    399.         seen_binary = []
    400.         seen_source = []
    401.         self.default_server = uri
    402.         for source in self.main_sources:
    403.             change_server_of_source(source, uri, seen_binary)
    404.         for source in self.child_sources:
    405.             # Do not change the forces server of a child source
    406.             if (source.template.base_uri is None or
    407.                 source.template.base_uri != source.uri):
    408.                 change_server_of_source(source, uri, seen_binary)
    409.         for source in self.source_code_sources:
    410.             change_server_of_source(source, uri, seen_source)
    412.     def is_codename(self, name):
    413.         ''' Compare a given name with the release codename. '''
    414.         if name == self.codename:
    415.             return True
    416.         else:
    417.             return False
    420. class DebianDistribution(Distribution):
    421.     ''' Class to support specific Debian features '''
    423.     def is_codename(self, name):
    424.         ''' Compare a given name with the release codename and check if
    425.             if it can be used as a synonym for a development releases '''
    426.         if name == self.codename or self.release in ("testing", "unstable"):
    427.             return True
    428.         else:
    429.             return False
    431.     def _get_mirror_name(self, server):
    432.         ''' Try to get a human readable name for the main mirror of a country
    433.             Debian specific '''
    434.         country = None
    435.         i = server.find("://ftp.")
    436.         l = server.find(".debian.org")
    437.         if i != -1 and l != -1:
    438.             country = server[i+len("://ftp."):l]
    439.         if country in self.countries:
    440.             # TRANSLATORS: %s is a country
    441.             return _("Server for %s") % gettext.dgettext(
    442.                 "iso_3166", self.countries[country].rstrip()).rstrip()
    443.         else:
    444.             return("%s" % server.rstrip("/ "))
    446.     def get_mirrors(self):
    447.         Distribution.get_mirrors(
    448.             self, mirror_template="http://ftp.%s.debian.org/debian/")
    451. class UbuntuDistribution(Distribution):
    452.     ''' Class to support specific Ubuntu features '''
    454.     def get_mirrors(self):
    455.         Distribution.get_mirrors(
    456.             self, mirror_template="http://%s.archive.ubuntu.com/ubuntu/")
    459. def _lsb_release():
    460.     """Call lsb_release --idrc and return a mapping."""
    461.     from subprocess import Popen, PIPE
    462.     import errno
    463.     result = {'Codename': 'sid', 'Distributor ID': 'Debian',
    464.               'Description': 'Debian GNU/Linux unstable (sid)',
    465.               'Release': 'unstable'}
    466.     try:
    467.         out = Popen(['lsb_release', '-idrc'], stdout=PIPE).communicate()[0]
    468.         # Convert to unicode string, needed for Python 3.1
    469.         out = out.decode("utf-8")
    470.         result.update(l.split(":\t") for l in out.split("\n") if ':\t' in l)
    471.     except OSError as exc:
    472.         if exc.errno != errno.ENOENT:
    473.             logging.warning('lsb_release failed, using defaults:' % exc)
    474.     return result
    477. def get_distro(id=None, codename=None, description=None, release=None):
    478.     """
    479.     Check the currently used distribution and return the corresponding
    480.     distriubtion class that supports distro specific features.
    482.     If no paramter are given the distro will be auto detected via
    483.     a call to lsb-release
    484.     """
    485.     # make testing easier
    486.     if not (id and codename and description and release):
    487.         result = _lsb_release()
    488.         id = result['Distributor ID']
    489.         codename = result['Codename']
    490.         description = result['Description']
    491.         release = result['Release']
    492.     if id == "Ubuntu":
    493.         return UbuntuDistribution(id, codename, description, release)
    494.     elif id == "Debian":
    495.         return DebianDistribution(id, codename, description, release)
    496.     else:
    497.         return Distribution(id, codename, description, release)
    Copy the Code
    Reply View the author
    Bluek404
    deepin
    2014-05-17 06:53
    #4
    /usr/lib/python3/dist-packages/aptsources/distro.py
    1. #  distro.py - Provide a distro abstraction of the sources.list
    2. #
    3. #  Copyright (c) 2004-2009 Canonical Ltd.
    4. #  Copyright (c) 2006-2007 Sebastian Heinlein
    5. #
    6. #  Authors: Sebastian Heinlein
    7. #           Michael Vogt
    8. #
    9. #  This program is free software; you can redistribute it and/or
    10. #  modify it under the terms of the GNU General Public License as
    11. #  published by the Free Software Foundation; either version 2 of the
    12. #  License, or (at your option) any later version.
    13. #
    14. #  This program is distributed in the hope that it will be useful,
    15. #  but WITHOUT ANY WARRANTY; without even the implied warranty of
    16. #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    17. #  GNU General Public License for more details.
    18. #
    19. #  You should have received a copy of the GNU General Public License
    20. #  along with this program; if not, write to the Free Software
    21. #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
    22. #  USA
    24. import gettext
    25. import logging
    26. import re
    27. import os
    29. from xml.etree.ElementTree import ElementTree
    31. from apt_pkg import gettext as _
    34. class NoDistroTemplateException(Exception):
    35.     pass
    38. class Distribution(object):
    40.     def __init__(self, id, codename, description, release):
    41.         """ Container for distribution specific informations """
    42.         # LSB information
    43.         self.id = id
    44.         self.codename = codename
    45.         self.description = description
    46.         self.release = release
    48.         self.binary_type = "deb"
    49.         self.source_type = "deb-src"
    51.     def get_sources(self, sourceslist):
    52.         """
    53.         Find the corresponding template, main and child sources
    54.         for the distribution
    55.         """
    57.         self.sourceslist = sourceslist
    58.         # corresponding sources
    59.         self.source_template = None
    60.         self.child_sources = []
    61.         self.main_sources = []
    62.         self.disabled_sources = []
    63.         self.cdrom_sources = []
    64.         self.download_comps = []
    65.         self.enabled_comps = []
    66.         self.cdrom_comps = []
    67.         self.used_media = []
    68.         self.get_source_code = False
    69.         self.source_code_sources = []
    71.         # location of the sources
    72.         self.default_server = ""
    73.         self.main_server = ""
    74.         self.nearest_server = ""
    75.         self.used_servers = []
    77.         # find the distro template
    78.         for template in self.sourceslist.matcher.templates:
    79.             if (self.is_codename(template.name) and
    80.                 template.distribution == self.id):
    81.                 #print "yeah! found a template for %s" % self.description
    82.                 #print template.description, template.base_uri, \
    83.                 #    template.components
    84.                 self.source_template = template
    85.                 break
    86.         if self.source_template is None:
    87.             raise NoDistroTemplateException(
    88.                 "Error: could not find a distribution template for %s/%s" %
    89.                 (self.id, self.codename))
    91.         # find main and child sources
    92.         media = []
    93.         comps = []
    94.         cdrom_comps = []
    95.         enabled_comps = []
    96.         #source_code = []
    97.         for source in self.sourceslist.list:
    98.             if (source.invalid == False and
    99.                 self.is_codename(source.dist) and
    100.                 source.template and
    101.                 source.template.official == True and
    102.                 self.is_codename(source.template.name)):
    103.                 #print "yeah! found a distro repo:  %s" % source.line
    104.                 # cdroms need do be handled differently
    105.                 if (source.uri.startswith("cdrom:") and
    106.                     source.disabled == False):
    107.                     self.cdrom_sources.append(source)
    108.                     cdrom_comps.extend(source.comps)
    109.                 elif (source.uri.startswith("cdrom:") and
    110.                       source.disabled == True):
    111.                     self.cdrom_sources.append(source)
    112.                 elif (source.type == self.binary_type and
    113.                       source.disabled == False):
    114.                     self.main_sources.append(source)
    115.                     comps.extend(source.comps)
    116.                     media.append(source.uri)
    117.                 elif (source.type == self.binary_type and
    118.                       source.disabled == True):
    119.                     self.disabled_sources.append(source)
    120.                 elif (source.type == self.source_type
    121.                         and source.disabled == False):
    122.                     self.source_code_sources.append(source)
    123.                 elif (source.type == self.source_type and
    124.                       source.disabled == True):
    125.                     self.disabled_sources.append(source)
    126.             if (source.invalid == False and
    127.                 source.template in self.source_template.children):
    128.                 if (source.disabled == False
    129.                     and source.type == self.binary_type):
    130.                     self.child_sources.append(source)
    131.                 elif (source.disabled == False
    132.                       and source.type == self.source_type):
    133.                     self.source_code_sources.append(source)
    134.                 else:
    135.                     self.disabled_sources.append(source)
    136.         self.download_comps = set(comps)
    137.         self.cdrom_comps = set(cdrom_comps)
    138.         enabled_comps.extend(comps)
    139.         enabled_comps.extend(cdrom_comps)
    140.         self.enabled_comps = set(enabled_comps)
    141.         self.used_media = set(media)
    142.         self.get_mirrors()
    144.     def get_mirrors(self, mirror_template=None):
    145.         """
    146.         Provide a set of mirrors where you can get the distribution from
    147.         """
    148.         # the main server is stored in the template
    149.         self.main_server = self.source_template.base_uri
    151.         # other used servers
    152.         for medium in self.used_media:
    153.             if not medium.startswith("cdrom:"):
    154.                 # seems to be a network source
    155.                 self.used_servers.append(medium)
    157.         if len(self.main_sources) == 0:
    158.             self.default_server = self.main_server
    159.         else:
    160.             self.default_server = self.main_sources[0].uri
    162.         # get a list of country codes and real names
    163.         self.countries = {}
    164.         fname = "/usr/share/xml/iso-codes/iso_3166.xml"
    165.         if os.path.exists(fname):
    166.             et = ElementTree(file=fname)
    167.             # python2.6 compat, the next two lines can get removed
    168.             # once we do not use py2.6 anymore
    169.             if getattr(et, "iter", None) is None:
    170.                 et.iter = et.getiterator
    171.             it = et.iter('iso_3166_entry')
    172.             for elm in it:
    173.                 try:
    174.                     descr = elm.attrib["common_name"]
    175.                 except KeyError:
    176.                     descr = elm.attrib["name"]
    177.                 try:
    178.                     code = elm.attrib["alpha_2_code"]
    179.                 except KeyError:
    180.                     code = elm.attrib["alpha_3_code"]
    181.                 self.countries[code.lower()] = gettext.dgettext('iso_3166',
    182.                                                                 descr)
    184.         # try to guess the nearest mirror from the locale
    185.         self.country = None
    186.         self.country_code = None
    187.         locale = os.getenv("LANG", default="en_UK")
    188.         a = locale.find("_")
    189.         z = locale.find(".")
    190.         if z == -1:
    191.             z = len(locale)
    192.         country_code = locale[a+1:z].lower()
    194.         if mirror_template:
    195.             self.nearest_server = mirror_template % country_code
    197.         if country_code in self.countries:
    198.             self.country = self.countries[country_code]
    199.             self.country_code = country_code
    201.     def _get_mirror_name(self, server):
    202.         ''' Try to get a human readable name for the main mirror of a country
    203.             Customize for different distributions '''
    204.         country = None
    205.         i = server.find("://")
    206.         l = server.find(".archive.ubuntu.com")
    207.         if i != -1 and l != -1:
    208.             country = server[i+len("://"):l]
    209.         if country in self.countries:
    210.             # TRANSLATORS: %s is a country
    211.             return _("Server for %s") % self.countries[country]
    212.         else:
    213.             return("%s" % server.rstrip("/ "))
    215.     def get_server_list(self):
    216.         ''' Return a list of used and suggested servers '''
    218.         def compare_mirrors(mir1, mir2):
    219.             ''' Helper function that handles comaprision of mirror urls
    220.                 that could contain trailing slashes'''
    221.             return re.match(mir1.strip("/ "), mir2.rstrip("/ "))
    223.         # Store all available servers:
    224.         # Name, URI, active
    225.         mirrors = []
    226.         if (len(self.used_servers) < 1 or
    227.             (len(self.used_servers) == 1 and
    228.              compare_mirrors(self.used_servers[0], self.main_server))):
    229.             mirrors.append([_("Main server"), self.main_server, True])
    230.             if self.nearest_server:
    231.                 mirrors.append([self._get_mirror_name(self.nearest_server),
    232.                                 self.nearest_server, False])
    233.         elif (len(self.used_servers) == 1 and not
    234.               compare_mirrors(self.used_servers[0], self.main_server)):
    235.             mirrors.append([_("Main server"), self.main_server, False])
    236.             # Only one server is used
    237.             server = self.used_servers[0]
    239.             # Append the nearest server if it's not already used
    240.             if self.nearest_server:
    241.                 if not compare_mirrors(server, self.nearest_server):
    242.                     mirrors.append([self._get_mirror_name(self.nearest_server),
    243.                                     self.nearest_server, False])
    244.             if server:
    245.                 mirrors.append([self._get_mirror_name(server), server, True])
    247.         elif len(self.used_servers) > 1:
    248.             # More than one server is used. Since we don't handle this case
    249.             # in the user interface we set "custom servers" to true and
    250.             # append a list of all used servers
    251.             mirrors.append([_("Main server"), self.main_server, False])
    252.             if self.nearest_server:
    253.                 mirrors.append([self._get_mirror_name(self.nearest_server),
    254.                                 self.nearest_server, False])
    255.             mirrors.append([_("Custom servers"), None, True])
    256.             for server in self.used_servers:
    257.                 mirror_entry = [self._get_mirror_name(server), server, False]
    258.                 if (compare_mirrors(server, self.nearest_server) or
    259.                     compare_mirrors(server, self.main_server)):
    260.                     continue
    261.                 elif not mirror_entry in mirrors:
    262.                     mirrors.append(mirror_entry)
    264.         return mirrors
    266.     def add_source(self, type=None,
    267.                  uri=None, dist=None, comps=None, comment=""):
    268.         """
    269.         Add distribution specific sources
    270.         """
    271.         if uri is None:
    272.             # FIXME: Add support for the server selector
    273.             uri = self.default_server
    274.         if dist is None:
    275.             dist = self.codename
    276.         if comps is None:
    277.             comps = list(self.enabled_comps)
    278.         if type is None:
    279.             type = self.binary_type
    280.         new_source = self.sourceslist.add(type, uri, dist, comps, comment)
    281.         # if source code is enabled add a deb-src line after the new
    282.         # source
    283.         if self.get_source_code == True and type == self.binary_type:
    284.             self.sourceslist.add(self.source_type, uri, dist, comps, comment,
    285.                                  file=new_source.file,
    286.                                  pos=self.sourceslist.list.index(new_source)+1)
    288.     def enable_component(self, comp):
    289.         """
    290.         Enable a component in all main, child and source code sources
    291.         (excluding cdrom based sources)
    293.         comp:         the component that should be enabled
    294.         """
    295.         comps = set([comp])
    296.         # look for parent components that we may have to add
    297.         for source in self.main_sources:
    298.             for c in source.template.components:
    299.                 if c.name == comp and c.parent_component:
    300.                     comps.add(c.parent_component)
    301.         for c in comps:
    302.             self._enable_component(c)
    304.     def _enable_component(self, comp):
    306.         def add_component_only_once(source, comps_per_dist):
    307.             """
    308.             Check if we already added the component to the repository, since
    309.             a repository could be splitted into different apt lines. If not
    310.             add the component
    311.             """
    312.             # if we don't have that distro, just return (can happen for e.g.
    313.             # dapper-update only in deb-src
    314.             if source.dist not in comps_per_dist:
    315.                 return
    316.             # if we have seen this component already for this distro,
    317.             # return (nothing to do)
    318.             if comp in comps_per_dist[source.dist]:
    319.                 return
    320.             # add it
    321.             source.comps.append(comp)
    322.             comps_per_dist[source.dist].add(comp)
    324.         sources = []
    325.         sources.extend(self.main_sources)
    326.         sources.extend(self.child_sources)
    327.         # store what comps are enabled already per distro (where distro is
    328.         # e.g. "dapper", "dapper-updates")
    329.         comps_per_dist = {}
    330.         comps_per_sdist = {}
    331.         for s in sources:
    332.             if s.type == self.binary_type:
    333.                 if s.dist not in comps_per_dist:
    334.                     comps_per_dist[s.dist] = set()
    335.                 for c in s.comps:
    336.                     comps_per_dist[s.dist].add(c)
    337.         for s in self.source_code_sources:
    338.             if s.type == self.source_type:
    339.                 if s.dist not in comps_per_sdist:
    340.                     comps_per_sdist[s.dist] = set()
    341.                 for c in s.comps:
    342.                     comps_per_sdist[s.dist].add(c)
    344.         # check if there is a main source at all
    345.         if len(self.main_sources) < 1:
    346.             # create a new main source
    347.             self.add_source(comps=["%s"%comp])
    348.         else:
    349.             # add the comp to all main, child and source code sources
    350.             for source in sources:
    351.                 add_component_only_once(source, comps_per_dist)
    353.             for source in self.source_code_sources:
    354.                 add_component_only_once(source, comps_per_sdist)
    356.         # check if there is a main source code source at all
    357.         if self.get_source_code == True:
    358.             if len(self.source_code_sources) < 1:
    359.                 # create a new main source
    360.                 self.add_source(type=self.source_type, comps=["%s"%comp])
    361.             else:
    362.                 # add the comp to all main, child and source code sources
    363.                 for source in self.source_code_sources:
    364.                     add_component_only_once(source, comps_per_sdist)
    366.     def disable_component(self, comp):
    367.         """
    368.         Disable a component in all main, child and source code sources
    369.         (excluding cdrom based sources)
    370.         """
    371.         sources = []
    372.         sources.extend(self.main_sources)
    373.         sources.extend(self.child_sources)
    374.         sources.extend(self.source_code_sources)
    375.         if comp in self.cdrom_comps:
    376.             sources = []
    377.             sources.extend(self.main_sources)
    378.         for source in sources:
    379.             if comp in source.comps:
    380.                 source.comps.remove(comp)
    381.                 if len(source.comps) < 1:
    382.                     self.sourceslist.remove(source)
    384.     def change_server(self, uri):
    385.         ''' Change the server of all distro specific sources to
    386.             a given host '''
    388.         def change_server_of_source(source, uri, seen):
    389.             # Avoid creating duplicate entries
    390.             source.uri = uri
    391.             for comp in source.comps:
    392.                 if [source.uri, source.dist, comp] in seen:
    393.                     source.comps.remove(comp)
    394.                 else:
    395.                     seen.append([source.uri, source.dist, comp])
    396.             if len(source.comps) < 1:
    397.                 self.sourceslist.remove(source)
    399.         seen_binary = []
    400.         seen_source = []
    401.         self.default_server = uri
    402.         for source in self.main_sources:
    403.             change_server_of_source(source, uri, seen_binary)
    404.         for source in self.child_sources:
    405.             # Do not change the forces server of a child source
    406.             if (source.template.base_uri is None or
    407.                 source.template.base_uri != source.uri):
    408.                 change_server_of_source(source, uri, seen_binary)
    409.         for source in self.source_code_sources:
    410.             change_server_of_source(source, uri, seen_source)
    412.     def is_codename(self, name):
    413.         ''' Compare a given name with the release codename. '''
    414.         if name == self.codename:
    415.             return True
    416.         else:
    417.             return False
    420. class DebianDistribution(Distribution):
    421.     ''' Class to support specific Debian features '''
    423.     def is_codename(self, name):
    424.         ''' Compare a given name with the release codename and check if
    425.             if it can be used as a synonym for a development releases '''
    426.         if name == self.codename or self.release in ("testing", "unstable"):
    427.             return True
    428.         else:
    429.             return False
    431.     def _get_mirror_name(self, server):
    432.         ''' Try to get a human readable name for the main mirror of a country
    433.             Debian specific '''
    434.         country = None
    435.         i = server.find("://ftp.")
    436.         l = server.find(".debian.org")
    437.         if i != -1 and l != -1:
    438.             country = server[i+len("://ftp."):l]
    439.         if country in self.countries:
    440.             # TRANSLATORS: %s is a country
    441.             return _("Server for %s") % gettext.dgettext(
    442.                 "iso_3166", self.countries[country].rstrip()).rstrip()
    443.         else:
    444.             return("%s" % server.rstrip("/ "))
    446.     def get_mirrors(self):
    447.         Distribution.get_mirrors(
    448.             self, mirror_template="http://ftp.%s.debian.org/debian/")
    451. class UbuntuDistribution(Distribution):
    452.     ''' Class to support specific Ubuntu features '''
    454.     def get_mirrors(self):
    455.         Distribution.get_mirrors(
    456.             self, mirror_template="http://%s.archive.ubuntu.com/ubuntu/")
    459. def _lsb_release():
    460.     """Call lsb_release --idrc and return a mapping."""
    461.     from subprocess import Popen, PIPE
    462.     import errno
    463.     result = {'Codename': 'sid', 'Distributor ID': 'Debian',
    464.               'Description': 'Debian GNU/Linux unstable (sid)',
    465.               'Release': 'unstable'}
    466.     try:
    467.         out = Popen(['lsb_release', '-idrc'], stdout=PIPE).communicate()[0]
    468.         # Convert to unicode string, needed for Python 3.1
    469.         out = out.decode("utf-8")
    470.         result.update(l.split(":\t") for l in out.split("\n") if ':\t' in l)
    471.     except OSError as exc:
    472.         if exc.errno != errno.ENOENT:
    473.             logging.warning('lsb_release failed, using defaults:' % exc)
    474.     return result
    477. def get_distro(id=None, codename=None, description=None, release=None):
    478.     """
    479.     Check the currently used distribution and return the corresponding
    480.     distriubtion class that supports distro specific features.
    482.     If no paramter are given the distro will be auto detected via
    483.     a call to lsb-release
    484.     """
    485.     # make testing easier
    486.     if not (id and codename and description and release):
    487.         result = _lsb_release()
    488.         id = result['Distributor ID']
    489.         codename = result['Codename']
    490.         description = result['Description']
    491.         release = result['Release']
    492.     if id == "Ubuntu":
    493.         return UbuntuDistribution(id, codename, description, release)
    494.     elif id == "Debian":
    495.         return DebianDistribution(id, codename, description, release)
    496.     else:
    497.         return Distribution(id, codename, description, release)
    Copy the Code
    多谢 : D
    Reply View the author
    v***h@gmail.com
    deepin
    2023-06-14 17:32
    #5
    It has been deleted!