EVOLUTION-MANAGER

Edit File: sources.py

"""
Our so-called sources file is simple text-based line-oriented file format.

Each line represents one source file and is in the same format as the output
of commands like `md5sum --tag filename`:

hashtype (filename) = hash

To preserve backwards compatibility, lines can also be in the older format,
which corresponds to the output of commands like `md5sum filename`:

hash  filename

This module implements a simple API to read these files, parse lines into
entries, and write these entries to the file in the proper format.
"""

import os
import re

from .errors import HashtypeMixingError, MalformedLineError, rpkgError

LINE_PATTERN = re.compile(
    r'^(?P<hashtype>[^ ]+?) $(?P<file>[^ )]+?)$ = (?P<hash>[^ ]+?)$')

class SourcesFile(object):
    def __init__(self, sourcesfile, entry_type, replace=False):
        self.sourcesfile = sourcesfile
        self.entry_type = {'old': SourceFileEntry,
                           'bsd': BSDSourceFileEntry}[entry_type]
        self.entries = []

if not replace:
            if not os.path.exists(sourcesfile):
                return

with open(sourcesfile) as f:
                for line in f:
                    entry = self.parse_line(line)

if entry and entry not in self.entries:
                        self.entries.append(entry)

def __contains__(self, filename):
        for entry in self.entries:
            if entry.file == filename:
                return True
        return False

def parse_line(self, line):
        stripped = line.strip()

if not stripped:
            return

m = LINE_PATTERN.match(stripped)
        if m is not None:
            return self.entry_type(m.group('hashtype'), m.group('file'),
                                   m.group('hash'))

# Try falling back on the old format
        try:
            hash, file = stripped.split('  ', 1)

except ValueError:
            raise MalformedLineError(
                'sources has invalid content: {0}\n'
                'Please note that sources file must not be modified manually.'
                .format(stripped))

return self.entry_type('md5', file, hash)

def add_entry(self, hashtype, file, hash):
        entry = self.entry_type(hashtype, file, hash)

for e in self.entries:
            if entry.hashtype != e.hashtype:
                raise HashtypeMixingError(e.hashtype, entry.hashtype)

if entry == e:
                return

if e.file == entry.file and e.hash != entry.hash:
                raise rpkgError(
                    'Uploading file {0} which has different checksum.\n'
                    'If this is the only file in sources, please use '
                    'new-sources.\n'
                    'If mutilple files are in sources, please remove this '
                    'file from sources manually.'.format(e.file))

self.entries.append(entry)

def write(self):
        with open(self.sourcesfile, 'w') as f:
            for entry in self.entries:
                f.write(str(entry))

class SourceFileEntry(object):
    def __init__(self, hashtype, file, hash):
        self.hashtype = hashtype.lower()
        self.hash = hash
        self.file = file

def __str__(self):
        return '%s  %s\n' % (self.hash, self.file)

def __eq__(self, other):
        return ((self.hashtype, self.hash, self.file) ==
                (other.hashtype, other.hash, other.file))

class BSDSourceFileEntry(SourceFileEntry):
    def __str__(self):
        return '%s (%s) = %s\n' % (self.hashtype.upper(), self.file,
                                   self.hash)