Annotation of /mga-gnome/trunk/mga-gnome

#!/usr/bin/python

import os
import os.path
import sys
import re
import subprocess
import urllib2
import urlparse
import argparse
import errno
import tempfile
import shutil
from sgmllib import SGMLParser

MEDIA="Core Release Source"
URL="http://download.gnome.org/sources/"
PKGROOT='~/pkgs'

re_version = re.compile(r'([-.]|\d+|[^-.\d]+)')

def version_cmp(a, b):
    """Compares two versions

    Returns
      -1 if a < b
      0  if a == b
      1  if a > b

    Logic from Bugzilla::Install::Util::vers_cmp"""
    A = re_version.findall(a.lstrip('0'))
    B = re_version.findall(b.lstrip('0'))

    while A and B:
        a = A.pop(0)
        b = B.pop(0)

        if a == b:
            continue
        elif a == '-':
            return -1
        elif b == '-':
            return 1
        elif a == '.':
            return -1
        elif b == '.':
            return 1
        elif a.isdigit() and b.isdigit():
            c = cmp(a, b) if (a.startswith('0') or b.startswith('0')) else cmp(int(a, 10), int(b, 10))
            if c:
                return c
        else:
            c = cmp(a.upper(), b.upper())
            if c:
                return c

    return cmp(len(A), len(B))

def get_latest_version(versions, max_version=None):
    """Gets the latest version number

    if max_version is specified, gets the latest version number before
    max_version"""
    latest = None
    for version in versions:
        if ( latest is None or version_cmp(version, latest) > 0 ) \
           and ( max_version is None or version_cmp(version, max_version) < 0 ):
            latest = version
    return latest

def line_input (file):
    for line in file:
        if line[-1] == '\n':
            yield line[:-1]
        else:
            yield line

def call_editor(filename):
    """Return a sequence of possible editor binaries for the current platform"""

    editors = []

    for varname in 'VISUAL', 'EDITOR':
        if varname in os.environ:
            editors.append(os.environ[varname])

    editors.extend(('/usr/bin/editor', 'vi', 'pico', 'nano', 'joe'))

    for editor in editors:
        try:
            ret = subprocess.call([editor, filename])
        except OSError, e:
            if e.errno == 2:
                continue
            raise

        if ret == 127:
            continue

        return True

class urllister(SGMLParser):
    def reset(self):
        SGMLParser.reset(self)
        self.urls = []

    def start_a(self, attrs):
        href = [v for k, v in attrs if k=='href']
        if href:
            self.urls.extend(href)

class SpecFile(object):
    re_update_version = re.compile(r'^(?P<pre>Version:\s*)(?P<version>.+)(?P<post>\s*)$', re.MULTILINE + re.IGNORECASE)
    re_update_release = re.compile(r'^(?P<pre>Release:\s*)(?P<release>%mkrel \d+)(?P<post>\s*)$', re.MULTILINE + re.IGNORECASE)

    def __init__(self, path):
        self.path = path
        self.cwd = os.path.dirname(path)

    @property
    def version(self):
        return subprocess.check_output(["rpm", "--specfile", self.path, "--queryformat", "%{VERSION}\n"]).splitlines()[0]
    def update(self, version):
        """Update specfile (increase version)"""
        cur_version = self.version

        if version_cmp(version, cur_version) != 1:
            print >>sys.stderr, "ERROR: Version %s is older than current version %s!" % (version, cur_version)
            return False

        with open(self.path, "rw") as f:
            data = f.read()

            if data.count("%mkrel") != 1:
                print "WARNING: Multiple %mkrel found; don't know what to do!"
                return False

            data, nr = self.re_update_version.subn(r'\g<pre>%s\g<post>' % version, data, 1)
            if nr != 1:
                print "WARNING: Could not increase version!"
                return False

            data, nr = self.re_update_release.subn(r'\g<pre>%mkrel 1\g<post>', data, 1)
            if nr != 1:
                print "WARNING: Could not reset release!"
                return False

            # Overwrite file with new version number
            write_file(self.path, data)


        # Check RPM also agrees that version number has increased
        if self.version != version:
            print "ERROR: Increased version to %s, but RPM doesn't agree!?!" % version
            return False

        try:
            # Download new tarball
            subprocess.check_call(['mgarepo', 'sync', '-d'], cwd=self.cwd)
            # Check patches still apply
            subprocess.check_call(['bm', '-p', '--nodeps'], cwd=self.cwd)
        except subprocess.CalledProcessError:
            return False

        return True

class Patch(object):
    """Do things with patches"""

    re_dep3 = re.compile(r'^(?:#\s*)?(?P<header>[-A-Za-z0-9]+?):\s*(?P<data>.*)$')
    re_dep3_cont = re.compile(r'^#?\s+(?P<data>.*)$')

    def __init__(self, path, show_path=False):
        """Path: path to patch (might not exist)"""
        self.path = path
        self.show_path = show_path

    def __str__(self):
        return self.path if self.show_path else os.path.basename(self.path)

    def add_dep3(self):
        """Add DEP-3 headers to a patch file"""
        if self.dep3['valid']:
            return False

        new_headers = (
            ('Author', self.svn_author),
            ('Subject', ''),
            ('Applied-Upstream', ''),
            ('Forwarded', ''),
            ('Bug', ''),
        )

        with tempfile.NamedTemporaryFile(dir=os.path.dirname(self.path), delete=False) as fdst:
            with open(self.path, "r") as fsrc:
                # Start with any existing DEP3 headers
                for i in range(self.dep3['last_nr']):
                    fdst.write(fsrc.read())

                # After that add the DEP3 headers
                add_line = False
                for header, data in new_headers:
                    if header in self.dep3['headers']:
                        continue

                    # XXX - wrap this at 80 chars
                    add_line = True
                    print >>fdst, "%s: %s" % (header, "" if data is None else data)

                if add_line: print >>fdst, ""
                # Now copy any other data and the patch
                shutil.copyfileobj(fsrc, fdst)

            fdst.flush()
            os.rename(fdst.name, self.path)

        call_editor(self.path)

    #Author: fwang
    #Subject: Build fix: Fix glib header inclusion
    #Applied-Upstream: commit:30602
    #Forwarded: yes
    #Bug: http://bugzilla.abisource.com/show_bug.cgi?id=13247

    def _read_dep3(self):
        """Read DEP-3 headers from an existing patch file

        This will also parse git headers"""
        dep3 = {}
        headers = {}

        last_header = None
        last_nr = 0
        nr = 0
        try:
            with open(self.path, "r") as f:
                for line in line_input(f):
                    nr += 1
                    # stop trying to parse when real patch begins
                    if line == '---':
                        break

                    r = self.re_dep3.match(line)
                    if r:
                        info = r.groupdict()

                        # Avoid matching URLS
                        if info['data'].startswith('//') and info['header'].lower () == info['header']:
                            continue

                        headers[info['header']] = info['data']
                        last_header = info['header']
                        last_nr = nr
                        continue

                    r = self.re_dep3_cont.match(line)
                    if r:
                        info = r.groupdict()
                        if last_header:
                            headers[last_header] = " ".join((headers[last_header], info['data']))
                            last_nr = nr
                        continue

                    last_header = None
        except IOError:
            pass

        dep3['valid'] = \
            (('Description' in headers and headers['Description'].strip() != '')
                or ('Subject' in headers and headers['Subject'].strip() != '')) \
            and (('Origin' in headers and headers['Origin'].strip() != '') \
                or ('Author' in headers and headers['Author'].strip() != '') \
                or ('From' in headers and headers['From'].strip() != ''))
        dep3['last_nr'] = last_nr
        dep3['headers'] = headers

        self._dep3 = dep3

    @property
    def dep3(self):
        if not hasattr(self, '_dep3'):
            self._read_dep3()

        return self._dep3

    @property
    def svn_author(self):
        if not hasattr(self, '_svn_author'):
            p = subprocess.Popen(['svn', 'log', '-q', "--", self.path], stdout=subprocess.PIPE, close_fds=True)
            contents = p.stdout.read().strip("\n").splitlines()
            ecode = p.wait()
            if ecode == 0:
                for line in contents:
                    if ' | ' not in line:
                        continue

                    fields = line.split(' | ')
                    if len(fields) >= 3:
                        self._svn_author = fields[1]

        if not hasattr(self, '_svn_author'):
            return None

        return self._svn_author

def get_upstream_names():
    urlopen = urllib2.build_opener()

    good_dir = re.compile('^[-A-Za-z0-9_+.]+/$')

    # Get the files
    usock = urlopen.open(URL)
    parser = urllister()
    parser.feed(usock.read())
    usock.close()
    parser.close()
    files = parser.urls

    tarballs = set([filename.replace('/', '') for filename in files if good_dir.search(filename)])

    return tarballs

def get_downstream_names():
    re_file = re.compile(r'^(?P<module>.*?)[_-](?:(?P<oldversion>([0-9]+[\.])*[0-9]+)-)?(?P<version>([0-9]+[\.\-])*[0-9]+)\.(?P<format>(?:tar\.|diff\.)?[a-z][a-z0-9]*)$')

    p = subprocess.Popen(['urpmf', '--files', '.', "--media", MEDIA], stdout=subprocess.PIPE, close_fds=True)
    contents = p.stdout.read().strip("\n").splitlines()
    ecode = p.wait()
    if ecode != 0:
        sys.exit(1)

    FILES = {}
    TARBALLS = {}

    for line in  contents:
        try:
            srpm, filename = line.split(":")
        except ValueError:
            print >>sys.stderr, line
            continue

        if '.tar' in filename:
            r = re_file.match(filename)
            if r:
                fileinfo = r.groupdict()
                module = fileinfo['module']

                if module not in TARBALLS:
                    TARBALLS[module] = set()
                TARBALLS[module].add(srpm)

        if srpm not in FILES:
            FILES[srpm] = set()
        FILES[srpm].add(filename)

    return TARBALLS, FILES


def write_file(path, data):
    with tempfile.NamedTemporaryFile(dir=os.path.dirname(path), delete=False) as fdst:
        fdst.write(data)
        fdst.flush()
        os.rename(fdst.name, path)

def cmd_co(options, parser):
    upstream = get_upstream_names()
    downstream, downstream_files = get_downstream_names()

    cwd = os.path.expanduser(PKGROOT)

    matches = upstream & set(downstream.keys())
    for module in matches:
        print module, "\t".join(downstream[module])
        for package in downstream[module]:
            subprocess.call(['mgarepo', 'co', package], cwd=cwd)

def cmd_ls(options, parser):
    upstream = get_upstream_names()
    downstream, downstream_files = get_downstream_names()

    matches = upstream & set(downstream.keys())
    for module in matches:
        print "\n".join(downstream[module])

def cmd_patches(options, parser):
    upstream = get_upstream_names()
    downstream, downstream_files = get_downstream_names()

    path = os.path.expanduser(PKGROOT)

    import pprint

    matches = upstream & set(downstream.keys())
    for module in sorted(matches):
        for srpm in downstream[module]:
            for filename in downstream_files[srpm]:
                if '.patch' in filename or '.diff' in filename:

                    p = Patch(os.path.join(path, srpm, "SOURCES", filename), show_path=options.path)
                    valid = ""
                    forwarded = ""
                    if p.dep3['headers']:
                        forwarded = p.dep3['headers'].get('Forwarded', "no")
                        if p.dep3['valid']:
                            valid="VALID"
                    print "\t".join((module, srpm, str(p), forwarded, valid))

def cmd_dep3(options, parser):
    p = Patch(options.patch)
    p.add_dep3()

def cmd_package_new_version(options, parser):
    cwd = os.path.expanduser(PKGROOT)
    package = options.package

    subprocess.call(['mgarepo', 'co', package], cwd=cwd)
    s = SpecFile(os.path.join(cwd, package, "SPECS", "%s.spec" % package))
    print s.version
    if not s.update(options.version):
        sys.exit(1)


def main():
    description = """Mageia GNOME commands."""
    epilog="""Report bugs to Olav Vitters"""
    parser = argparse.ArgumentParser(description=description,epilog=epilog)

    # SUBPARSERS
    subparsers = parser.add_subparsers(title='subcommands')
    #   install
    subparser = subparsers.add_parser('co', help='checkout all GNOME modules')
    subparser.set_defaults(
        func=cmd_co
    )

    subparser = subparsers.add_parser('packages', help='list all GNOME packages')
    subparser.set_defaults(
        func=cmd_ls
    )

    subparser = subparsers.add_parser('patches', help='list all GNOME patches')
    subparser.add_argument("-p", "--path", action="store_true", dest="path",
                                       help="Show full path to patch")
    subparser.set_defaults(
        func=cmd_patches, path=False
    )

    subparser = subparsers.add_parser('dep3', help='Add dep3 headers')
    subparser.add_argument("patch", help="Patch")
    subparser.set_defaults(
        func=cmd_dep3, path=False
    )

    subparser = subparsers.add_parser('increase', help='Increase version number')
    subparser.add_argument("package", help="Package name")
    subparser.add_argument("version", help="Version number")
    subparser.set_defaults(
        func=cmd_package_new_version, path=False
    )

    if len(sys.argv) == 1:
        parser.print_help()
        sys.exit(2)

    options = parser.parse_args()

    try:
        options.func(options, parser)
    except KeyboardInterrupt:
        print('Interrupted')
        sys.exit(1)
    except EOFError:
        print('EOF')
        sys.exit(1)
    except IOError, e:
        if e.errno != errno.EPIPE:
            raise
        sys.exit(0)

if __name__ == "__main__":
    main()
1	ovitters	2932	#!/usr/bin/python
2
3			import os
4			import os.path
5			import sys
6			import re
7			import subprocess
8			import urllib2
9			import urlparse
10			import argparse
11	ovitters	2936	import errno
12	ovitters	2944	import tempfile
13			import shutil
14	ovitters	2932	from sgmllib import SGMLParser
15
16			MEDIA="Core Release Source"
17			URL="http://download.gnome.org/sources/"
18			PKGROOT='~/pkgs'
19
20	ovitters	3012	re_version = re.compile(r'([-.]\|\d+\|[^-.\d]+)')
21
22			def version_cmp(a, b):
23			"""Compares two versions
24
25			Returns
26			-1 if a < b
27			0 if a == b
28			1 if a > b
29
30			Logic from Bugzilla::Install::Util::vers_cmp"""
31			A = re_version.findall(a.lstrip('0'))
32			B = re_version.findall(b.lstrip('0'))
33
34			while A and B:
35			a = A.pop(0)
36			b = B.pop(0)
37
38			if a == b:
39			continue
40			elif a == '-':
41			return -1
42			elif b == '-':
43			return 1
44			elif a == '.':
45			return -1
46			elif b == '.':
47			return 1
48			elif a.isdigit() and b.isdigit():
49			c = cmp(a, b) if (a.startswith('0') or b.startswith('0')) else cmp(int(a, 10), int(b, 10))
50			if c:
51			return c
52			else:
53			c = cmp(a.upper(), b.upper())
54			if c:
55			return c
56
57			return cmp(len(A), len(B))
58
59			def get_latest_version(versions, max_version=None):
60			"""Gets the latest version number
61
62			if max_version is specified, gets the latest version number before
63			max_version"""
64			latest = None
65			for version in versions:
66			if ( latest is None or version_cmp(version, latest) > 0 ) \
67			and ( max_version is None or version_cmp(version, max_version) < 0 ):
68			latest = version
69			return latest
70
71	ovitters	2936	def line_input (file):
72			for line in file:
73			if line[-1] == '\n':
74			yield line[:-1]
75			else:
76			yield line
77
78	ovitters	2955	def call_editor(filename):
79			"""Return a sequence of possible editor binaries for the current platform"""
80
81			editors = []
82
83			for varname in 'VISUAL', 'EDITOR':
84			if varname in os.environ:
85			editors.append(os.environ[varname])
86
87			editors.extend(('/usr/bin/editor', 'vi', 'pico', 'nano', 'joe'))
88
89			for editor in editors:
90			try:
91			ret = subprocess.call([editor, filename])
92			except OSError, e:
93			if e.errno == 2:
94			continue
95			raise
96
97			if ret == 127:
98			continue
99
100			return True
101
102	ovitters	2932	class urllister(SGMLParser):
103			def reset(self):
104			SGMLParser.reset(self)
105			self.urls = []
106
107			def start_a(self, attrs):
108			href = [v for k, v in attrs if k=='href']
109			if href:
110			self.urls.extend(href)
111
112	ovitters	3012	class SpecFile(object):
113			re_update_version = re.compile(r'^(?P<pre>Version:\s)(?P<version>.+)(?P<post>\s)$', re.MULTILINE + re.IGNORECASE)
114			re_update_release = re.compile(r'^(?P<pre>Release:\s)(?P<release>%mkrel \d+)(?P<post>\s)$', re.MULTILINE + re.IGNORECASE)
115
116			def __init__(self, path):
117			self.path = path
118			self.cwd = os.path.dirname(path)
119
120			@property
121			def version(self):
122			return subprocess.check_output(["rpm", "--specfile", self.path, "--queryformat", "%{VERSION}\n"]).splitlines()[0]
123			def update(self, version):
124			"""Update specfile (increase version)"""
125			cur_version = self.version
126
127			if version_cmp(version, cur_version) != 1:
128			print >>sys.stderr, "ERROR: Version %s is older than current version %s!" % (version, cur_version)
129			return False
130
131			with open(self.path, "rw") as f:
132			data = f.read()
133
134			if data.count("%mkrel") != 1:
135			print "WARNING: Multiple %mkrel found; don't know what to do!"
136			return False
137
138			data, nr = self.re_update_version.subn(r'\g<pre>%s\g<post>' % version, data, 1)
139			if nr != 1:
140			print "WARNING: Could not increase version!"
141			return False
142
143			data, nr = self.re_update_release.subn(r'\g<pre>%mkrel 1\g<post>', data, 1)
144			if nr != 1:
145			print "WARNING: Could not reset release!"
146			return False
147
148			# Overwrite file with new version number
149			write_file(self.path, data)
150
151
152			# Check RPM also agrees that version number has increased
153			if self.version != version:
154			print "ERROR: Increased version to %s, but RPM doesn't agree!?!" % version
155			return False
156
157	ovitters	3034	try:
158			# Download new tarball
159			subprocess.check_call(['mgarepo', 'sync', '-d'], cwd=self.cwd)
160			# Check patches still apply
161			subprocess.check_call(['bm', '-p', '--nodeps'], cwd=self.cwd)
162			except subprocess.CalledProcessError:
163			return False
164	ovitters	3012
165			return True
166
167	ovitters	2936	class Patch(object):
168			"""Do things with patches"""
169
170			re_dep3 = re.compile(r'^(?:#\s)?(?P<header>[-A-Za-z0-9]+?):\s(?P<data>.*)$')
171			re_dep3_cont = re.compile(r'^#?\s+(?P<data>.*)$')
172
173			def __init__(self, path, show_path=False):
174			"""Path: path to patch (might not exist)"""
175			self.path = path
176			self.show_path = show_path
177
178			def __str__(self):
179			return self.path if self.show_path else os.path.basename(self.path)
180
181			def add_dep3(self):
182	ovitters	2955	"""Add DEP-3 headers to a patch file"""
183	ovitters	2944	if self.dep3['valid']:
184			return False
185
186			new_headers = (
187			('Author', self.svn_author),
188			('Subject', ''),
189			('Applied-Upstream', ''),
190			('Forwarded', ''),
191			('Bug', ''),
192			)
193
194			with tempfile.NamedTemporaryFile(dir=os.path.dirname(self.path), delete=False) as fdst:
195			with open(self.path, "r") as fsrc:
196			# Start with any existing DEP3 headers
197			for i in range(self.dep3['last_nr']):
198			fdst.write(fsrc.read())
199
200			# After that add the DEP3 headers
201			add_line = False
202			for header, data in new_headers:
203			if header in self.dep3['headers']:
204			continue
205
206			# XXX - wrap this at 80 chars
207			add_line = True
208	ovitters	2955	print >>fdst, "%s: %s" % (header, "" if data is None else data)
209	ovitters	2944
210			if add_line: print >>fdst, ""
211			# Now copy any other data and the patch
212			shutil.copyfileobj(fsrc, fdst)
213
214			fdst.flush()
215			os.rename(fdst.name, self.path)
216
217	ovitters	2955	call_editor(self.path)
218
219	ovitters	2936	#Author: fwang
220			#Subject: Build fix: Fix glib header inclusion
221			#Applied-Upstream: commit:30602
222			#Forwarded: yes
223			#Bug: http://bugzilla.abisource.com/show_bug.cgi?id=13247
224
225			def _read_dep3(self):
226	ovitters	2955	"""Read DEP-3 headers from an existing patch file
227
228			This will also parse git headers"""
229	ovitters	2936	dep3 = {}
230	ovitters	2944	headers = {}
231	ovitters	2936
232			last_header = None
233	ovitters	2944	last_nr = 0
234			nr = 0
235	ovitters	2936	try:
236			with open(self.path, "r") as f:
237			for line in line_input(f):
238	ovitters	2944	nr += 1
239			# stop trying to parse when real patch begins
240	ovitters	2936	if line == '---':
241			break
242
243			r = self.re_dep3.match(line)
244			if r:
245			info = r.groupdict()
246	ovitters	3012
247			# Avoid matching URLS
248			if info['data'].startswith('//') and info['header'].lower () == info['header']:
249			continue
250
251	ovitters	2944	headers[info['header']] = info['data']
252	ovitters	2936	last_header = info['header']
253	ovitters	2944	last_nr = nr
254	ovitters	2936	continue
255
256			r = self.re_dep3_cont.match(line)
257			if r:
258			info = r.groupdict()
259			if last_header:
260	ovitters	2944	headers[last_header] = " ".join((headers[last_header], info['data']))
261			last_nr = nr
262	ovitters	2936	continue
263
264			last_header = None
265			except IOError:
266			pass
267	ovitters	2944
268			dep3['valid'] = \
269			(('Description' in headers and headers['Description'].strip() != '')
270			or ('Subject' in headers and headers['Subject'].strip() != '')) \
271			and (('Origin' in headers and headers['Origin'].strip() != '') \
272			or ('Author' in headers and headers['Author'].strip() != '') \
273			or ('From' in headers and headers['From'].strip() != ''))
274			dep3['last_nr'] = last_nr
275			dep3['headers'] = headers
276
277	ovitters	2936	self._dep3 = dep3
278
279			@property
280			def dep3(self):
281			if not hasattr(self, '_dep3'):
282			self._read_dep3()
283
284			return self._dep3
285
286	ovitters	2944	@property
287			def svn_author(self):
288			if not hasattr(self, '_svn_author'):
289			p = subprocess.Popen(['svn', 'log', '-q', "--", self.path], stdout=subprocess.PIPE, close_fds=True)
290			contents = p.stdout.read().strip("\n").splitlines()
291			ecode = p.wait()
292			if ecode == 0:
293			for line in contents:
294			if ' \| ' not in line:
295			continue
296	ovitters	2936
297	ovitters	2944	fields = line.split(' \| ')
298			if len(fields) >= 3:
299			self._svn_author = fields[1]
300
301	ovitters	2955	if not hasattr(self, '_svn_author'):
302			return None
303
304	ovitters	2944	return self._svn_author
305
306	ovitters	2932	def get_upstream_names():
307			urlopen = urllib2.build_opener()
308
309			good_dir = re.compile('^[-A-Za-z0-9_+.]+/$')
310
311			# Get the files
312			usock = urlopen.open(URL)
313			parser = urllister()
314			parser.feed(usock.read())
315			usock.close()
316			parser.close()
317			files = parser.urls
318
319			tarballs = set([filename.replace('/', '') for filename in files if good_dir.search(filename)])
320
321			return tarballs
322
323			def get_downstream_names():
324			re_file = re.compile(r'^(?P<module>.?)[_-](?:(?P<oldversion>([0-9]+[\.])[0-9]+)-)?(?P<version>([0-9]+[\.\-])[0-9]+)\.(?P<format>(?:tar\.\|diff\.)?[a-z][a-z0-9])$')
325
326			p = subprocess.Popen(['urpmf', '--files', '.', "--media", MEDIA], stdout=subprocess.PIPE, close_fds=True)
327			contents = p.stdout.read().strip("\n").splitlines()
328			ecode = p.wait()
329			if ecode != 0:
330			sys.exit(1)
331
332			FILES = {}
333			TARBALLS = {}
334
335			for line in contents:
336			try:
337			srpm, filename = line.split(":")
338			except ValueError:
339			print >>sys.stderr, line
340			continue
341
342			if '.tar' in filename:
343			r = re_file.match(filename)
344			if r:
345			fileinfo = r.groupdict()
346			module = fileinfo['module']
347
348			if module not in TARBALLS:
349			TARBALLS[module] = set()
350			TARBALLS[module].add(srpm)
351
352			if srpm not in FILES:
353			FILES[srpm] = set()
354			FILES[srpm].add(filename)
355
356			return TARBALLS, FILES
357
358	ovitters	3012
359			def write_file(path, data):
360			with tempfile.NamedTemporaryFile(dir=os.path.dirname(path), delete=False) as fdst:
361			fdst.write(data)
362			fdst.flush()
363			os.rename(fdst.name, path)
364
365	ovitters	2932	def cmd_co(options, parser):
366			upstream = get_upstream_names()
367			downstream, downstream_files = get_downstream_names()
368
369			cwd = os.path.expanduser(PKGROOT)
370
371			matches = upstream & set(downstream.keys())
372			for module in matches:
373			print module, "\t".join(downstream[module])
374			for package in downstream[module]:
375			subprocess.call(['mgarepo', 'co', package], cwd=cwd)
376
377			def cmd_ls(options, parser):
378			upstream = get_upstream_names()
379			downstream, downstream_files = get_downstream_names()
380
381			matches = upstream & set(downstream.keys())
382			for module in matches:
383			print "\n".join(downstream[module])
384
385			def cmd_patches(options, parser):
386			upstream = get_upstream_names()
387			downstream, downstream_files = get_downstream_names()
388
389			path = os.path.expanduser(PKGROOT)
390
391	ovitters	2936	import pprint
392
393	ovitters	2932	matches = upstream & set(downstream.keys())
394			for module in sorted(matches):
395			for srpm in downstream[module]:
396			for filename in downstream_files[srpm]:
397			if '.patch' in filename or '.diff' in filename:
398	ovitters	3012
399	ovitters	2936	p = Patch(os.path.join(path, srpm, "SOURCES", filename), show_path=options.path)
400	ovitters	3012	valid = ""
401			forwarded = ""
402	ovitters	2944	if p.dep3['headers']:
403	ovitters	3012	forwarded = p.dep3['headers'].get('Forwarded', "no")
404	ovitters	2944	if p.dep3['valid']:
405	ovitters	3012	valid="VALID"
406			print "\t".join((module, srpm, str(p), forwarded, valid))
407	ovitters	2932
408	ovitters	2944	def cmd_dep3(options, parser):
409			p = Patch(options.patch)
410			p.add_dep3()
411
412	ovitters	3012	def cmd_package_new_version(options, parser):
413			cwd = os.path.expanduser(PKGROOT)
414			package = options.package
415
416			subprocess.call(['mgarepo', 'co', package], cwd=cwd)
417			s = SpecFile(os.path.join(cwd, package, "SPECS", "%s.spec" % package))
418			print s.version
419			if not s.update(options.version):
420			sys.exit(1)
421
422
423	ovitters	2932	def main():
424			description = """Mageia GNOME commands."""
425			epilog="""Report bugs to Olav Vitters"""
426			parser = argparse.ArgumentParser(description=description,epilog=epilog)
427
428			# SUBPARSERS
429			subparsers = parser.add_subparsers(title='subcommands')
430			# install
431			subparser = subparsers.add_parser('co', help='checkout all GNOME modules')
432			subparser.set_defaults(
433			func=cmd_co
434			)
435
436			subparser = subparsers.add_parser('packages', help='list all GNOME packages')
437			subparser.set_defaults(
438			func=cmd_ls
439			)
440
441			subparser = subparsers.add_parser('patches', help='list all GNOME patches')
442			subparser.add_argument("-p", "--path", action="store_true", dest="path",
443	ovitters	2944	help="Show full path to patch")
444	ovitters	2932	subparser.set_defaults(
445			func=cmd_patches, path=False
446			)
447
448	ovitters	2944	subparser = subparsers.add_parser('dep3', help='Add dep3 headers')
449			subparser.add_argument("patch", help="Patch")
450			subparser.set_defaults(
451			func=cmd_dep3, path=False
452			)
453	ovitters	2932
454	ovitters	3012	subparser = subparsers.add_parser('increase', help='Increase version number')
455			subparser.add_argument("package", help="Package name")
456			subparser.add_argument("version", help="Version number")
457			subparser.set_defaults(
458			func=cmd_package_new_version, path=False
459			)
460
461	ovitters	2932	if len(sys.argv) == 1:
462			parser.print_help()
463			sys.exit(2)
464
465			options = parser.parse_args()
466
467			try:
468			options.func(options, parser)
469			except KeyboardInterrupt:
470			print('Interrupted')
471			sys.exit(1)
472			except EOFError:
473			print('EOF')
474			sys.exit(1)
475			except IOError, e:
476			if e.errno != errno.EPIPE:
477			raise
478			sys.exit(0)
479
480			if __name__ == "__main__":
481			main()