First pass at a function to import an srpm

Right now the library just copies in the files, and makes the call to
upload new sources, but does not commit anything.  Leaving that up to
the library caller.

1 files changed, 70 insertions, 0 deletions

diff --git a/src/pyfedpkg/__init__.py b/src/pyfedpkg/__init__.py
index e03a102..b4d1a91 100644
--- a/src/pyfedpkg/__init__.py
+++ b/src/pyfedpkg/__init__.py
@@ -548,6 +548,76 @@ class PackageModule:
         self.kojisession.uploadWrapper(file, path, callback = callback)
         return
 
+    def import_srpm(self, srpm):
+        """Import the contents of an srpm into a repo.
+
+        srpm: File to import contents from
+
+        This function will add/remove content to match the srpm,
+
+        upload new files to the lookaside, and stage the changes.
+
+        Returns nothing or raises.
+
+        """
+
+
+        # see if the srpm even exists
+        srpm = os.path.abspath(srpm)
+        if not os.path.exists(srpm):
+            raise FedpkgError('File not found.')
+        # bail if we're dirty
+        if self.repo.is_dirty():
+            raise FedpkgError('There are uncommitted changes in your repo')
+        # Get the details of the srpm
+        name, files, uploadfiles = _srpmdetails(srpm)
+
+        # See if our module matches the srpm we're trying to imoprt
+        if name != self.module:
+            raise FedpkgError('Srpm does not match module')
+
+        # Get a list of files we're currently tracking
+        ourfiles = self.repo.git.ls_files().split()
+        # Trim out sources and .gitignore
+        ourfiles.remove('.gitignore')
+        ourfiles.remove('sources')
+
+        # Things work better if we're in our module directory
+        oldpath = os.getcwd()
+        os.chdir(self.path)
+
+        # Look through our files and if it isn't in the new files, remove it.
+        for file in ourfiles:
+            if file not in files:
+                log.info("Removing no longer used file: %s" % file)
+                rv = self.repo.index.remove([file])
+                os.remove(file)
+
+        # Extract new files
+        cmd = ['rpm2cpio', srpm]
+        # We have to force cpio to copy out (u) because git messes with
+        # timestamps
+        cmd2 = ['cpio', '-iud', '--quiet']
+
+        rpmcall = subprocess.Popen(cmd, stdout=subprocess.PIPE)
+        cpiocall = subprocess.Popen(cmd2, stdin=rpmcall.stdout)
+        output, err = cpiocall.communicate()
+        if output:
+            log.debug(output)
+        if err:
+            os.chdir(oldpath)
+            raise FedpkgError("Got an error from rpm2cpio: %s" % err)
+
+        # now process the upload files
+        self.new_sources(uploadfiles)
+        # And finally add all the files we know about (and our stock files)
+        files.append('.gitignore')
+        files.append('sources')
+        rv = self.repo.index.add(files)
+        # Return to the caller and let them take it from there.
+        os.chdir(oldpath)
+        return
+
     def init_koji(self, user, kojiconfig=None, url=None):
         """Initiate a koji session.  Available options are: