1*4882a593SmuzhiyunPRSERV_DUMPDIR ??= "${LOG_DIR}/db" 2*4882a593SmuzhiyunLOCALCOUNT_DUMPFILE ??= "${PRSERV_DUMPDIR}/prserv-localcount-exports.inc" 3*4882a593Smuzhiyun 4*4882a593Smuzhiyunpython migrate_localcount_handler () { 5*4882a593Smuzhiyun import bb.event 6*4882a593Smuzhiyun if not e.data: 7*4882a593Smuzhiyun return 8*4882a593Smuzhiyun 9*4882a593Smuzhiyun pv = e.data.getVar('PV') 10*4882a593Smuzhiyun if not 'AUTOINC' in pv: 11*4882a593Smuzhiyun return 12*4882a593Smuzhiyun 13*4882a593Smuzhiyun localcounts = bb.persist_data.persist('BB_URI_LOCALCOUNT', e.data) 14*4882a593Smuzhiyun pn = e.data.getVar('PN') 15*4882a593Smuzhiyun revs = localcounts.get_by_pattern('%%-%s_rev' % pn) 16*4882a593Smuzhiyun counts = localcounts.get_by_pattern('%%-%s_count' % pn) 17*4882a593Smuzhiyun if not revs or not counts: 18*4882a593Smuzhiyun return 19*4882a593Smuzhiyun 20*4882a593Smuzhiyun if len(revs) != len(counts): 21*4882a593Smuzhiyun bb.warn("The number of revs and localcounts don't match in %s" % pn) 22*4882a593Smuzhiyun return 23*4882a593Smuzhiyun 24*4882a593Smuzhiyun version = e.data.getVar('PRAUTOINX') 25*4882a593Smuzhiyun srcrev = bb.fetch2.get_srcrev(e.data) 26*4882a593Smuzhiyun base_ver = 'AUTOINC-%s' % version[:version.find(srcrev)] 27*4882a593Smuzhiyun pkgarch = e.data.getVar('PACKAGE_ARCH') 28*4882a593Smuzhiyun value = max(int(count) for count in counts) 29*4882a593Smuzhiyun 30*4882a593Smuzhiyun if len(revs) == 1: 31*4882a593Smuzhiyun if srcrev != ('AUTOINC+%s' % revs[0]): 32*4882a593Smuzhiyun value += 1 33*4882a593Smuzhiyun else: 34*4882a593Smuzhiyun value += 1 35*4882a593Smuzhiyun 36*4882a593Smuzhiyun bb.utils.mkdirhier(e.data.getVar('PRSERV_DUMPDIR')) 37*4882a593Smuzhiyun df = e.data.getVar('LOCALCOUNT_DUMPFILE') 38*4882a593Smuzhiyun flock = bb.utils.lockfile("%s.lock" % df) 39*4882a593Smuzhiyun with open(df, 'a') as fd: 40*4882a593Smuzhiyun fd.write('PRAUTO$%s$%s$%s = "%s"\n' % 41*4882a593Smuzhiyun (base_ver, pkgarch, srcrev, str(value))) 42*4882a593Smuzhiyun bb.utils.unlockfile(flock) 43*4882a593Smuzhiyun} 44*4882a593Smuzhiyun 45*4882a593Smuzhiyunaddhandler migrate_localcount_handler 46*4882a593Smuzhiyunmigrate_localcount_handler[eventmask] = "bb.event.RecipeParsed" 47