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
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
|
# -*- coding: utf-8 -*-
"""
reporead command
Parses a repo.db.tar.gz file and updates the Arch database with the relevant
changes.
Usage: ./manage.py reporead ARCH PATH
ARCH: architecture to update; must be available in the database
PATH: full path to the repo.db.tar.gz file.
Example:
./manage.py reporead i686 /tmp/core.db.tar.gz
"""
from collections import defaultdict
from copy import copy
import io
import os
import re
import sys
import tarfile
import logging
from datetime import datetime
from optparse import make_option
from pytz import utc
from django.core.management.base import BaseCommand, CommandError
from django.db import connections, router, transaction
from django.db.utils import IntegrityError
from devel.utils import UserFinder
from main.models import Arch, Package, PackageFile, Repo
from main.utils import utc_now, database_vendor
from packages.models import Depend, Conflict, Provision, Replacement, Update
logging.basicConfig(
level=logging.WARNING,
format='%(asctime)s -> %(levelname)s: %(message)s',
datefmt='%Y-%m-%d %H:%M:%S',
stream=sys.stderr)
TRACE = 5
logging.addLevelName(TRACE, 'TRACE')
logger = logging.getLogger()
class Command(BaseCommand):
option_list = BaseCommand.option_list + (
make_option('-f', '--force', action='store_true', dest='force', default=False,
help='Force a re-import of data for all packages instead of only new ones. Will not touch the \'last updated\' value.'),
make_option('--filesonly', action='store_true', dest='filesonly', default=False,
help='Load filelists if they are outdated, but will not add or remove any packages. Will not touch the \'last updated\' value.'),
)
help = "Runs a package repository import for the given arch and file."
args = "<arch> <filename>"
def handle(self, arch=None, filename=None, **options):
if not arch:
raise CommandError('Architecture is required.')
if not filename:
raise CommandError('Package database file is required.')
filename = os.path.normpath(filename)
if not os.path.exists(filename) or not os.path.isfile(filename):
raise CommandError('Specified package database file does not exist.')
v = int(options.get('verbosity', 0))
if v == 0:
logger.level = logging.ERROR
elif v == 1:
logger.level = logging.INFO
elif v == 2:
logger.level = logging.DEBUG
return read_repo(arch, filename, options)
class RepoPackage(object):
"""An interim 'container' object for holding Arch package data."""
bare = ( 'name', 'base', 'arch', 'filename',
'md5sum', 'sha256sum', 'url', 'packager' )
number = ( 'csize', 'isize' )
collections = ( 'depends', 'optdepends', 'conflicts',
'provides', 'replaces', 'groups', 'license', 'files' )
version_re = re.compile(r'^((\d+):)?(.+)-([^-]+)$')
def __init__(self, repo):
self.repo = repo
self.ver = None
self.rel = None
self.epoch = 0
self.desc = None
self.pgpsig = None
for k in self.bare + self.number:
setattr(self, k, None)
for k in self.collections:
setattr(self, k, ())
self.builddate = None
self.files = None
self.has_files = False
def populate(self, values):
for k, v in values.iteritems():
# ensure we stay under our DB character limit
if k in self.bare:
setattr(self, k, v[0][:254])
elif k in self.number:
setattr(self, k, long(v[0]))
elif k in ('desc', 'pgpsig'):
# do NOT prune these values at all
setattr(self, k, v[0])
elif k == 'version':
match = self.version_re.match(v[0])
self.ver = match.group(3)
self.rel = match.group(4)
if match.group(2):
self.epoch = int(match.group(2))
elif k == 'builddate':
try:
builddate = datetime.utcfromtimestamp(int(v[0]))
self.builddate = builddate.replace(tzinfo=utc)
except ValueError:
try:
self.builddate = datetime.strptime(v[0],
'%a %b %d %H:%M:%S %Y')
except ValueError:
logger.warning(
'Package %s had unparsable build date %s',
self.name, v[0])
elif k == 'files':
self.files = tuple(v)
self.has_files = True
else:
# anything left in collections
setattr(self, k, tuple(v))
@property
def full_version(self):
'''Very similar to the main.models.Package method.'''
if self.epoch > 0:
return u'%d:%s-%s' % (self.epoch, self.ver, self.rel)
return u'%s-%s' % (self.ver, self.rel)
DEPEND_RE = re.compile(r"^(.+?)((>=|<=|=|>|<)(.+))?$")
def create_depend(package, dep_str, optional=False):
depend = Depend(pkg=package, optional=optional)
# lop off any description first
parts = dep_str.split(':', 1)
if len(parts) > 1:
depend.description = parts[1].strip()
match = DEPEND_RE.match(parts[0].strip())
if match:
depend.name = match.group(1)
if match.group(3):
depend.comparison = match.group(3)
if match.group(4):
depend.version = match.group(4)
else:
logger.warning('Package %s had unparsable depend string %s',
package.pkgname, dep_str)
return None
return depend
def create_related(model, package, rel_str, equals_only=False):
related = model(pkg=package)
match = DEPEND_RE.match(rel_str)
if match:
related.name = match.group(1)
if match.group(3):
comp = match.group(3)
if not equals_only:
related.comparison = comp
elif comp != '=':
logger.warning(
'Package %s had unexpected comparison operator %s for %s in %s',
package.pkgname, comp, model.__name__, rel_str)
if match.group(4):
related.version = match.group(4)
else:
logger.warning('Package %s had unparsable %s string %s',
package.pkgname, model.___name__, rel_str)
return None
return related
def batched_bulk_create(model, all_objects):
# for short lists, just bulk_create as we should be fine
if len(all_objects) < 20:
return model.objects.bulk_create(all_objects)
if database_vendor(model, mode='write') == 'sqlite':
# 999 max variables in each SQL statement
incr = 999 // len(model._meta.fields)
else:
incr = 1000
def chunks():
offset = 0
while offset < len(all_objects):
yield all_objects[offset:offset + incr]
offset += incr
for items in chunks():
model.objects.bulk_create(items)
def create_multivalued(dbpkg, repopkg, db_attr, repo_attr):
'''Populate the simplest of multivalued attributes. These are those that
only deal with a 'name' attribute, such as licenses, groups, etc. The input
and output objects and attribute names are specified, and everything is
done via getattr().'''
collection = getattr(dbpkg, db_attr)
collection.all().delete()
model = collection.model
new_items = []
for name in getattr(repopkg, repo_attr):
new_items.append(model(pkg=dbpkg, name=name))
if new_items:
model.objects.bulk_create(new_items)
finder = UserFinder()
def populate_pkg(dbpkg, repopkg, force=False, timestamp=None):
# we reset the flag date only if the upstream version components change;
# e.g. epoch or pkgver, but not pkgrel
if dbpkg.epoch is None or dbpkg.epoch != repopkg.epoch:
dbpkg.flag_date = None
elif dbpkg.pkgver is None or dbpkg.pkgver != repopkg.ver:
dbpkg.flag_date = None
if repopkg.base:
dbpkg.pkgbase = repopkg.base
else:
dbpkg.pkgbase = repopkg.name
dbpkg.pkgver = repopkg.ver
dbpkg.pkgrel = repopkg.rel
dbpkg.epoch = repopkg.epoch
dbpkg.pkgdesc = repopkg.desc
dbpkg.url = repopkg.url
dbpkg.filename = repopkg.filename
dbpkg.compressed_size = repopkg.csize
dbpkg.installed_size = repopkg.isize
dbpkg.build_date = repopkg.builddate
dbpkg.packager_str = repopkg.packager
# attempt to find the corresponding django user for this string
dbpkg.packager = finder.find(repopkg.packager)
dbpkg.pgp_signature = repopkg.pgpsig
if timestamp:
dbpkg.last_update = timestamp
dbpkg.save()
populate_files(dbpkg, repopkg, force=force)
dbpkg.depends.all().delete()
deps = [create_depend(dbpkg, y) for y in repopkg.depends]
deps += [create_depend(dbpkg, y, True) for y in repopkg.optdepends]
batched_bulk_create(Depend, deps)
dbpkg.conflicts.all().delete()
conflicts = [create_related(Conflict, dbpkg, y) for y in repopkg.conflicts]
batched_bulk_create(Conflict, conflicts)
dbpkg.provides.all().delete()
provides = [create_related(Provision, dbpkg, y, equals_only=True)
for y in repopkg.provides]
batched_bulk_create(Provision, provides)
dbpkg.replaces.all().delete()
replaces = [create_related(Replacement, dbpkg, y) for y in repopkg.replaces]
batched_bulk_create(Replacement, replaces)
create_multivalued(dbpkg, repopkg, 'groups', 'groups')
create_multivalued(dbpkg, repopkg, 'licenses', 'license')
pkg_same_version = lambda pkg, dbpkg: pkg.ver == dbpkg.pkgver \
and pkg.rel == dbpkg.pkgrel and pkg.epoch == dbpkg.epoch
def delete_pkg_files(dbpkg):
database = router.db_for_write(Package, instance=dbpkg)
cursor = connections[database].cursor()
cursor.execute('DELETE FROM package_files WHERE pkg_id = %s', [dbpkg.id])
def populate_files(dbpkg, repopkg, force=False):
if not force:
if not pkg_same_version(repopkg, dbpkg):
logger.info("DB version (%s) didn't match repo version "
"(%s) for package %s, skipping file list addition",
dbpkg.full_version, repopkg.full_version, dbpkg.pkgname)
return
if not dbpkg.files_last_update or not dbpkg.last_update:
pass
elif dbpkg.files_last_update >= dbpkg.last_update:
return
# only delete files if we are reading a DB that contains them
if repopkg.has_files:
delete_pkg_files(dbpkg)
logger.info("adding %d files for package %s",
len(repopkg.files), dbpkg.pkgname)
pkg_files = []
for f in repopkg.files:
if '/' in f:
dirname, filename = f.rsplit('/', 1)
dirname += '/'
else:
dirname, filename = '', f
if filename == '':
filename = None
pkgfile = PackageFile(pkg=dbpkg,
is_directory=(filename is None),
directory=dirname,
filename=filename)
pkg_files.append(pkgfile)
batched_bulk_create(PackageFile, pkg_files)
dbpkg.files_last_update = utc_now()
dbpkg.save()
def update_common(archname, reponame, pkgs, sanity_check=True):
# If isolation level is repeatable-read, we need to ensure each package
# update starts a new transaction and re-queries the database as
# necessary to guard against simultaneous updates.
with transaction.commit_on_success():
# force the transaction dirty, even though we will only do reads
transaction.set_dirty()
repository = Repo.objects.get(name__iexact=reponame)
architecture = Arch.objects.get(name__iexact=archname)
# no-arg order_by() removes even the default ordering; we don't need it
dbpkgs = Package.objects.filter(
arch=architecture, repo=repository).order_by()
logger.info("%d packages in current web DB", len(dbpkgs))
logger.info("%d packages in new updating DB", len(pkgs))
if len(dbpkgs):
dbpercent = 100.0 * len(pkgs) / len(dbpkgs)
else:
dbpercent = 0.0
logger.info("DB package ratio: %.1f%%", dbpercent)
# Fewer than 20 packages makes the percentage check unreliable, but it
# also means we expect the repo to fluctuate a lot.
msg = "Package database %s (%s) has %.1f%% the number of packages " \
"the web database"
if not sanity_check:
pass
elif repository.testing or repository.staging:
pass
elif len(dbpkgs) == 0 and len(pkgs) == 0:
pass
elif len(dbpkgs) > 20 and dbpercent < 50.0:
logger.error(msg, reponame, archname, dbpercent)
raise Exception(msg % (reponame, archname, dbpercent))
elif dbpercent < 75.0:
logger.warning(msg, reponame, archname, dbpercent)
return dbpkgs
def db_update(archname, reponame, pkgs, force=False):
"""
Parses a list of packages and updates the packages database accordingly.
"""
logger.info('Updating %s (%s)', reponame, archname)
dbpkgs = update_common(archname, reponame, pkgs, True)
repository = Repo.objects.get(name__iexact=reponame)
architecture = Arch.objects.get(name__iexact=archname)
# This makes our inner loop where we find packages by name *way* more
# efficient by not having to go to the database for each package to
# SELECT them by name.
dbdict = dict((dbpkg.pkgname, dbpkg) for dbpkg in dbpkgs)
dbset = set(dbdict.keys())
syncset = set([pkg.name for pkg in pkgs])
in_sync_not_db = syncset - dbset
logger.info("%d packages in sync not db", len(in_sync_not_db))
# packages in syncdb and not in database (add to database)
for pkg in (pkg for pkg in pkgs if pkg.name in in_sync_not_db):
logger.info("Adding package %s", pkg.name)
dbpkg = Package(pkgname=pkg.name, arch=architecture, repo=repository)
try:
with transaction.commit_on_success():
populate_pkg(dbpkg, pkg, timestamp=utc_now())
Update.objects.log_update(None, dbpkg)
except IntegrityError:
logger.warning("Could not add package %s; "
"not fatal if another thread beat us to it.",
pkg.name, exc_info=True)
# packages in database and not in syncdb (remove from database)
for pkgname in (dbset - syncset):
logger.info("Removing package %s", pkgname)
dbpkg = dbdict[pkgname]
with transaction.commit_on_success():
Update.objects.log_update(dbpkg, None)
# no race condition here as long as simultaneous threads both
# issue deletes; second delete will be a no-op
delete_pkg_files(dbpkg)
dbpkg.delete()
# packages in both database and in syncdb (update in database)
pkg_in_both = syncset & dbset
for pkg in (x for x in pkgs if x.name in pkg_in_both):
logger.debug("Checking package %s", pkg.name)
dbpkg = dbdict[pkg.name]
timestamp = None
# for a force, we don't want to update the timestamp.
# for a non-force, we don't want to do anything at all.
if not force and pkg_same_version(pkg, dbpkg):
continue
elif not force:
timestamp = utc_now()
# The odd select_for_update song and dance here are to ensure
# simultaneous updates don't happen on a package, causing
# files/depends/all related items to be double-imported.
with transaction.commit_on_success():
dbpkg = Package.objects.select_for_update().get(id=dbpkg.id)
if not force and pkg_same_version(pkg, dbpkg):
logger.debug("Package %s was already updated", pkg.name)
continue
logger.info("Updating package %s", pkg.name)
prevpkg = copy(dbpkg)
populate_pkg(dbpkg, pkg, force=force, timestamp=timestamp)
Update.objects.log_update(prevpkg, dbpkg)
logger.info('Finished updating arch: %s', archname)
def filesonly_update(archname, reponame, pkgs, force=False):
"""
Parses a list of packages and updates the packages database accordingly.
"""
logger.info('Updating files for %s (%s)', reponame, archname)
dbpkgs = update_common(archname, reponame, pkgs, False)
dbdict = dict((dbpkg.pkgname, dbpkg) for dbpkg in dbpkgs)
dbset = set(dbdict.keys())
for pkg in (pkg for pkg in pkgs if pkg.name in dbset):
dbpkg = dbdict[pkg.name]
# The odd select_for_update song and dance here are to ensure
# simultaneous updates don't happen on a package, causing
# files to be double-imported.
with transaction.commit_on_success():
if not dbpkg.files_last_update or not dbpkg.last_update:
pass
elif not force and dbpkg.files_last_update >= dbpkg.last_update:
logger.debug("Files for %s are up to date", pkg.name)
continue
dbpkg = Package.objects.select_for_update().get(id=dbpkg.id)
logger.debug("Checking files for package %s", pkg.name)
populate_files(dbpkg, pkg, force=force)
logger.info('Finished updating arch: %s', archname)
def parse_info(iofile):
"""
Parses an Arch repo db information file, and returns variables as a list.
"""
store = {}
blockname = None
for line in iofile:
line = line.strip()
if len(line) == 0:
continue
elif line.startswith('%') and line.endswith('%'):
blockname = line[1:-1].lower()
logger.log(TRACE, "Parsing package block %s", blockname)
store[blockname] = []
elif blockname:
store[blockname].append(line)
else:
raise Exception("Read package info outside a block: %s" % line)
return store
def parse_repo(repopath):
"""
Parses an Arch repo db file, and returns a list of RepoPackage objects.
Arguments:
repopath -- The path of a repository db file.
"""
logger.info("Starting repo parsing")
if not os.path.exists(repopath):
logger.error("Could not read file %s", repopath)
logger.info("Reading repo tarfile %s", repopath)
filename = os.path.split(repopath)[1]
m = re.match(r"^(.*)\.(db|files)\.tar(\..*)?$", filename)
if m:
reponame = m.group(1)
else:
logger.error("File does not have the proper extension")
raise Exception("File does not have the proper extension")
repodb = tarfile.open(repopath, "r")
logger.debug("Starting package parsing")
dbfiles = ('desc', 'depends', 'files')
newpkg = lambda: RepoPackage(reponame)
pkgs = defaultdict(newpkg)
for tarinfo in repodb.getmembers():
if tarinfo.isreg():
pkgid, fname = os.path.split(tarinfo.name)
if fname not in dbfiles:
continue
data_file = repodb.extractfile(tarinfo)
data_file = io.TextIOWrapper(io.BytesIO(data_file.read()),
encoding='UTF-8')
try:
pkgs[pkgid].populate(parse_info(data_file))
except UnicodeDecodeError:
logger.warn("Could not correctly decode %s, skipping file",
tarinfo.name)
data_file.close()
del data_file
logger.debug("Done parsing file %s/%s", pkgid, fname)
repodb.close()
logger.info("Finished repo parsing, %d total packages", len(pkgs))
return (reponame, pkgs.values())
def locate_arch(arch):
"Check if arch is valid."
if isinstance(arch, Arch):
return arch
try:
return Arch.objects.get(name__iexact=arch)
except Arch.DoesNotExist:
raise CommandError(
'Specified architecture %s is not currently known.' % arch)
def read_repo(primary_arch, repo_file, options):
"""
Parses repo.db.tar.gz file and returns exit status.
"""
# always returns an Arch object, regardless of what is passed in
primary_arch = locate_arch(primary_arch)
force = options.get('force', False)
filesonly = options.get('filesonly', False)
repo, packages = parse_repo(repo_file)
# group packages by arch -- to handle noarch stuff
packages_arches = {}
for arch in Arch.objects.filter(agnostic=True):
packages_arches[arch.name] = []
packages_arches[primary_arch.name] = []
for package in packages:
if package.arch in packages_arches:
packages_arches[package.arch].append(package)
else:
raise Exception(
"Package %s in database %s had wrong architecture %s" % (
package.name, repo_file, package.arch))
del packages
database = router.db_for_write(Package)
connection = connections[database]
if connection.vendor == 'sqlite':
cursor = connection.cursor()
cursor.execute('PRAGMA synchronous = NORMAL')
logger.info('Starting database updates for %s.', repo_file)
for arch in sorted(packages_arches.keys()):
if filesonly:
filesonly_update(arch, repo, packages_arches[arch], force)
else:
db_update(arch, repo, packages_arches[arch], force)
logger.info('Finished database updates for %s.', repo_file)
return 0
# vim: set ts=4 sw=4 et:
|