aboutsummaryrefslogtreecommitdiff
blob: f390e4546dc80593a66cf61178fb717f0c70167f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
#!/usr/bin/python
#
# Copyright 2009-2010 Gentoo Foundation
#
# Licensed under the GNU General Public License, v2
#
# $Header$

"""Provides attributes and methods for a category/package-version string."""

__all__ = ('CPV',)

# =======
# Imports
# =======

from portage.versions import catpkgsplit, vercmp

from gentoolkit import errors

# =======
# Classes
# =======

class CPV(object):
	"""Provides methods on a category/package-version string.

	Will also correctly split just a package or package-version string.

	Example usage:
		>>> from gentoolkit.cpv import CPV
		>>> cpv = CPV('sys-apps/portage-2.2-r1')
		>>> cpv.category, cpv.name, cpv.fullversion
		('sys-apps', 'portage', '2.2-r1')
		>>> str(cpv)
		'sys-apps/portage-2.2-r1'
		>>> # An 'rc' (release candidate) version is less than non 'rc' version:
		... CPV('sys-apps/portage-2') > CPV('sys-apps/portage-2_rc10')
		True
	"""

	def __init__(self, cpv):
		self.cpv = cpv

		values = split_cpv(cpv)
		self.category = values[0]
		self.name = values[1]
		self.version = values[2]
		self.revision = values[3]
		del values

		if not self.name:
			raise errors.GentoolkitInvalidCPV(cpv)

		sep = '/' if self.category else ''
		self.cp = sep.join((self.category, self.name))

		sep = '-' if self.revision else ''
		self.fullversion = sep.join((self.version, self.revision))
		del sep

	def __eq__(self, other):
		if not isinstance(other, self.__class__):
			return False
		return self.cpv == other.cpv

	def __ne__(self, other):
		return not self == other

	def __lt__(self, other):
		if not isinstance(other, self.__class__):
			raise TypeError("other isn't of %s type, is %s" % (
				self.__class__, other.__class__)
			)

		if self.category != other.category:
			return self.category < other.category
		elif self.name != other.name:
			return self.name < other.name
		else:
			# FIXME: this cmp() hack is for vercmp not using -1,0,1
			# See bug 266493; this was fixed in portage-2.2_rc31
			#return vercmp(self.fullversion, other.fullversion)
			result = cmp(vercmp(self.fullversion, other.fullversion), 0)
			if result == -1:
				return True
			else:
				return False

	def __gt__(self, other):
		if not isinstance(other, self.__class__):
			raise TypeError("other isn't of %s type, is %s" % (
				self.__class__, other.__class__)
			)
		return not self <= other

	def __le__(self, other):
		if not isinstance(other, self.__class__):
			raise TypeError("other isn't of %s type, is %s" % (
				self.__class__, other.__class__)
			)
		return self < other or self == other

	def __ge__(self, other):
		if not isinstance(other, self.__class__):
			raise TypeError("other isn't of %s type, is %s" % (
				self.__class__, other.__class__)
			)
		return self > other or self == other

	def __repr__(self):
		return "<%s %r>" % (self.__class__.__name__, str(self))

	def __str__(self):
		return self.cpv


# =========
# Functions
# =========

def split_cpv(cpv):
	"""Split a cpv into category, name, version and revision.

	Inlined from helpers because of circular imports.

	@todo: this function is slow and accepts some crazy things for cpv
	@type cpv: str
	@param cpv: pkg, cat/pkg, pkg-ver, cat/pkg-ver, atom or regex
	@rtype: tuple
	@return: (category, pkg_name, version, revision)
		Each tuple element is a string or empty string ("").
	"""

	result = catpkgsplit(cpv)

	if result:
		result = list(result)
		if result[0] == 'null':
			result[0] = ''
		if result[3] == 'r0':
			result[3] = ''
	else:
		result = cpv.split("/")
		if len(result) == 1:
			result = ['', cpv, '', '']
		else:
			result = result + ['', '']

	return tuple(result)

# vim: set ts=4 sw=4 tw=79: