From 8b88b494c70d9afb8e6389ec6919d9a023eb1137 Mon Sep 17 00:00:00 2001
From: Michael Tremer <michael.tremer@ipfire.org>
Date: Mon, 3 Oct 2011 01:06:07 +0200
Subject: [PATCH] Implement file size of source packages.

---
 python/pakfire/packages/base.py     | 7 ++++++-
 python/pakfire/packages/make.py     | 4 ----
 python/pakfire/packages/packager.py | 3 +++
 3 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/python/pakfire/packages/base.py b/python/pakfire/packages/base.py
index 8c5ae374e..0bb4f2156 100644
--- a/python/pakfire/packages/base.py
+++ b/python/pakfire/packages/base.py
@@ -208,7 +208,12 @@ class Package(object):
 
 	@property
 	def inst_size(self):
-		raise NotImplementedError, "%s" % self
+		"""
+			The used disk space when the package is installed.
+
+			Returns None if inst_size is unknown.
+		"""
+		return None
 
 	@property
 	def local(self):
diff --git a/python/pakfire/packages/make.py b/python/pakfire/packages/make.py
index 2019f47f6..ad48cb35e 100644
--- a/python/pakfire/packages/make.py
+++ b/python/pakfire/packages/make.py
@@ -343,10 +343,6 @@ class Makefile(MakefileBase):
 
 			shutil.copy2(_filename, filename)
 
-	@property
-	def inst_size(self):
-		return 0
-
 
 class MakefilePackage(MakefileBase):
 	def __init__(self, pakfire, name, lexer):
diff --git a/python/pakfire/packages/packager.py b/python/pakfire/packages/packager.py
index 4e913cbd2..b50483b49 100644
--- a/python/pakfire/packages/packager.py
+++ b/python/pakfire/packages/packager.py
@@ -552,6 +552,9 @@ class SourcePackager(Packager):
 		info.update(self.pakfire.distro.info)
 		info.update(self.pkg.info)
 
+		# Size is the size of the (uncompressed) datafile.
+		info["inst_size"] = os.path.getsize(datafile)
+
 		# Update package information for string formatting.
 		requires = [PACKAGE_INFO_DEPENDENCY_LINE % r for r in self.pkg.requires]
 		info.update({
-- 
2.39.5