1 files changed, 91 insertions, 0 deletions
diff --git a/apt/cdrom.py b/apt/cdrom.py
new file mode 100644
index 0000000..dc15c5b
--- /dev/null
+++ b/apt/cdrom.py
@@ -0,0 +1,91 @@
+# cdrom.py - CDROM handling
+#
+#  Copyright (c) 2005-2009 Canonical
+#  Copyright (c) 2009 Julian Andres Klode <jak@debian.org>
+#
+#  Author: Michael Vogt <michael.vogt@ubuntu.com>
+#
+#  This program is free software; you can redistribute it and/or
+#  modify it under the terms of the GNU General Public License as
+#  published by the Free Software Foundation; either version 2 of the
+#  License, or (at your option) any later version.
+#
+#  This program is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU General Public License for more details.
+#
+#  You should have received a copy of the GNU General Public License
+#  along with this program; if not, write to the Free Software
+#  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
+#  USA
+"""Classes related to cdrom handling."""
+import glob
+
+import apt_pkg
+
+from apt.progress.base import CdromProgress
+
+
+class Cdrom(apt_pkg.Cdrom):
+    """Support for apt-cdrom like features.
+
+    This class has several optional parameters for initialisation, which may
+    be used to influence the behaviour of the object:
+
+    The optional parameter `progress` is a CdromProgress() subclass, which will
+    ask for the correct cdrom, etc. If not specified or None, a CdromProgress()
+    object will be used.
+
+    The optional parameter `mountpoint` may be used to specify an alternative
+    mountpoint.
+
+    If the optional parameter `nomount` is True, the cdroms will not be
+    mounted. This is the default behaviour.
+    """
+
+    def __init__(
+        self,
+        progress: CdromProgress | None = None,
+        mountpoint: str | None = None,
+        nomount: bool = True,
+    ) -> None:
+        apt_pkg.Cdrom.__init__(self)
+        if progress is None:
+            self._progress = CdromProgress()
+        else:
+            self._progress = progress
+        # see if we have a alternative mountpoint
+        if mountpoint is not None:
+            apt_pkg.config.set("Acquire::cdrom::mount", mountpoint)
+        # do not mess with mount points by default
+        if nomount:
+            apt_pkg.config.set("APT::CDROM::NoMount", "true")
+        else:
+            apt_pkg.config.set("APT::CDROM::NoMount", "false")
+
+    def add(self, progress: CdromProgress | None = None) -> bool:
+        """Add cdrom to the sources.list."""
+        return apt_pkg.Cdrom.add(self, progress or self._progress)
+
+    def ident(self, progress: CdromProgress | None = None) -> str:
+        """Identify the cdrom."""
+        return apt_pkg.Cdrom.ident(self, progress or self._progress)
+
+    @property
+    def in_sources_list(self) -> bool:
+        """Check if the cdrom is already in the current sources.list."""
+        cd_id = self.ident()
+        if cd_id is None:
+            # FIXME: throw exception instead
+            return False
+        # Get a list of files
+        src = glob.glob(apt_pkg.config.find_dir("Dir::Etc::sourceparts") + "*")
+        src.append(apt_pkg.config.find_file("Dir::Etc::sourcelist"))
+        # Check each file
+        for fname in src:
+            with open(fname) as fobj:
+                for line in fobj:
+                    if not line.lstrip().startswith("#") and cd_id in line:
+                        return True
+        return False