pactest/pmpkg.py

   1 #! /usr/bin/python
   2 #
   3 #  Copyright (c) 2006 by Aurelien Foret <orelien@chez.com>
   4 #
   5 #  This program is free software; you can redistribute it and/or modify
   6 #  it under the terms of the GNU General Public License as published by
   7 #  the Free Software Foundation; either version 2 of the License, or
   8 #  (at your option) any later version.
   9 #
  10 #  This program is distributed in the hope that it will be useful,
  11 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 #  GNU General Public License for more details.
  14 #
  15 #  You should have received a copy of the GNU General Public License
  16 #  along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17
  18
  19 import os
  20 import tempfile
  21 import stat
  22 import shutil
  23 import tarfile
  24
  25 from util import *
  26
  27
  28 class pmpkg:
  29     """Package object.
  30
  31     Object holding data from an ArchLinux package.
  32     """
  33
  34     def __init__(self, name, version = "1.0-1"):
  35         self.path = "" #the path of the generated package
  36         # desc
  37         self.name = name
  38         self.version = version
  39         self.desc = ""
  40         self.groups = []
  41         self.url = ""
  42         self.license = []
  43         self.arch = ""
  44         self.builddate = ""
  45         self.installdate = ""
  46         self.packager = ""
  47         self.size = 0
  48         self.csize = 0
  49         self.reason = 0
  50         self.md5sum = ""      # sync only
  51         self.replaces = []    # sync only (will be moved to depends)
  52         self.force = 0        # sync only (will be moved to depends)
  53         # depends
  54         self.depends = []
  55         self.optdepends = []
  56         self.conflicts = []
  57         self.provides = []
  58         # files
  59         self.files = []
  60         self.backup = []
  61         # install
  62         self.install = {
  63             "pre_install": "",
  64             "post_install": "",
  65             "pre_remove": "",
  66             "post_remove": "",
  67             "pre_upgrade": "",
  68             "post_upgrade": ""
  69         }
  70         self.checksum = {
  71             "desc": "",
  72             "depends": "",
  73             "files": "",
  74             "install": ""
  75         }
  76         self.mtime = {
  77             "desc": (0, 0, 0),
  78             "depends": (0, 0, 0),
  79             "files": (0, 0, 0),
  80             "install": (0, 0, 0)
  81         }
  82
  83     def __str__(self):
  84         s = ["%s" % self.fullname()]
  85         s.append("description: %s" % self.desc)
  86         s.append("url: %s" % self.url)
  87         s.append("depends: %s" % " ".join(self.depends))
  88         s.append("files: %s" % " ".join(self.files))
  89         s.append("reason: %d" % self.reason)
  90         return "\n".join(s)
  91
  92     def fullname(self):
  93         """Long name of a package.
  94
  95         Returns a string formatted as follows: "pkgname-pkgver".
  96         """
  97         return "%s-%s" % (self.name, self.version)
  98
  99     def filename(self):
 100         """File name of a package, including its extension.
 101
 102         Returns a string formatted as follows: "pkgname-pkgver.PKG_EXT_PKG".
 103         """
 104         return "%s%s" % (self.fullname(), PM_EXT_PKG)
 105
 106     def install_files(self, root):
 107         """Install files in the filesystem located under "root".
 108
 109         Files are created with content generated automatically.
 110         """
 111         [mkfile(os.path.join(root, f), f) for f in self.files]
 112
 113     def makepkg(self, path):
 114         """Creates an ArchLinux package archive.
 115
 116         A package archive is generated in the location 'path', based on the data
 117         from the object.
 118         """
 119         self.path = os.path.join(path, self.filename())
 120
 121         curdir = os.getcwd()
 122         tmpdir = tempfile.mkdtemp()
 123         os.chdir(tmpdir)
 124
 125         # Generate package file system
 126         for f in self.files:
 127             mkfile(f, f)
 128             self.size += os.stat(getfilename(f))[stat.ST_SIZE]
 129
 130         # .PKGINFO
 131         data = ["pkgname = %s" % self.name]
 132         data.append("pkgver = %s" % self.version)
 133         data.append("pkgdesc = %s" % self.desc)
 134         data.append("url = %s" % self.url)
 135         data.append("builddate = %s" % self.builddate)
 136         data.append("packager = %s" % self.packager)
 137         data.append("size = %s" % self.size)
 138         if self.arch:
 139             data.append("arch = %s" % self.arch)
 140         for i in self.license:
 141             data.append("license = %s" % i)
 142         for i in self.replaces:
 143             data.append("replaces = %s" % i)
 144         for i in self.groups:
 145             data.append("group = %s" % i)
 146         for i in self.depends:
 147             data.append("depend = %s" % i)
 148         for i in self.optdepends:
 149             data.append("optdepend = %s" % i)
 150         for i in self.conflicts:
 151             data.append("conflict = %s" % i)
 152         for i in self.provides:
 153             data.append("provides = %s" % i)
 154         for i in self.backup:
 155             data.append("backup = %s" % i)
 156         mkfile(".PKGINFO", "\n".join(data))
 157
 158         # .INSTALL
 159         if len(self.install.values()) > 0:
 160             mkinstallfile(".INSTALL", self.install)
 161
 162         # safely create the dir
 163         mkdir(os.path.dirname(self.path))
 164
 165         # Generate package archive
 166         tar = tarfile.open(self.path, "w:gz")
 167
 168         # package files
 169         for root, dirs, files in os.walk('.'):
 170             for d in dirs:
 171                 tar.add(os.path.join(root, d), recursive=False)
 172             for f in files:
 173                 tar.add(os.path.join(root, f))
 174
 175         tar.close()
 176
 177         os.chdir(curdir)
 178         shutil.rmtree(tmpdir)
 179
 180
 181 if __name__ == "__main__":
 182     pkg = pmpkg("dummy")
 183     print pkg
 184 # vim: set ts=4 sw=4 et: