Advertisement
Guest User

Untitled

a guest
May 6th, 2017
161
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 87.94 KB | None | 0 0
  1. #! /usr/bin/python
  2.  
  3. import fnmatch, glob, os, re, string, sys, time, cStringIO
  4. from optparse import OptionParser
  5. from ConfigParser import SafeConfigParser
  6.  
  7. sys.path[0:0] = ['/usr/share/pycentral-data', '/usr/share/python']
  8. import pyversions
  9.  
  10. try:
  11. SetType = set
  12. except NameError:
  13. import sets
  14. SetType = sets.Set
  15. set = sets.Set
  16.  
  17. program = os.path.basename(sys.argv[0])
  18.  
  19. shared_base = '/usr/share/pycentral/'
  20. shared_base2 = '/usr/share/pyshared/'
  21. pycentral_version = '0.6.11ubuntu2'
  22. req_pycentral_version = '0.6.11'
  23.  
  24.  
  25. def get_file_overwrite_error(existing_files):
  26. """
  27. helper that returns a error string to be passed to the user when
  28. local files are detected (and querries if they are local installs
  29. or part of another package)
  30. """
  31. from subprocess import PIPE, Popen
  32. l = []
  33. # ask dpkg what it knows about the files in question
  34. (stdout, stderr) = Popen(["dpkg","-S"]+existing_files,
  35. env={"LANG" : "C"},
  36. stdout=PIPE, stderr=PIPE).communicate()
  37. # this should never happen, dpkg should always give us
  38. # something but we deal with it for robustness
  39. if not stdout and not stderr:
  40. l.append("Error, can not overwrite existing files:")
  41. l.extend(existing_files)
  42. # stdout has files that belong to packages
  43. if stdout:
  44. l.append("Not overwriting files owned by other packages:")
  45. for line in map(string.strip, stdout.split("\n")):
  46. if line:
  47. l.append(" "+line)
  48. # stderr has local (or maintainer script created) files
  49. if stderr:
  50. l.append("Not overwriting local files:")
  51. for line in map(string.strip, stderr.split("\n")):
  52. if line:
  53. l.append(" "+line)
  54. return "\n".join(l)
  55.  
  56. def samefs(path1, path2):
  57. if not (os.path.exists(path1) and os.path.exists(path2)):
  58. return False
  59. while path1 != os.path.dirname(path1):
  60. if os.path.ismount(path1):
  61. break
  62. path1 = os.path.dirname(path1)
  63. while path2 != os.path.dirname(path2):
  64. if os.path.ismount(path2):
  65. break
  66. path2 = os.path.dirname(path2)
  67. return path1 == path2
  68.  
  69. def version2depends(vinfo):
  70. if isinstance(vinfo, set):
  71. vinfo = list(vinfo)
  72. if isinstance(vinfo, list):
  73. vinfo = vinfo[:]
  74. vinfo.sort()
  75. nv = [int(s) for s in vinfo[-1].split('.')]
  76. deps = 'python (>= %s), python (<< %d.%d)' % (vinfo[0], nv[0], nv[1]+1)
  77. elif vinfo in ('all', 'current'):
  78. supported = [d[6:] for d in pyversions.supported_versions()
  79. if re.match(r'python\d\.\d', d)]
  80. supported.sort()
  81. deps = 'python (>= %s)' % supported[0]
  82. elif vinfo == 'current_ext':
  83. cv = pyversions.default_version(version_only=True)
  84. nv = [int(s) for s in cv.split('.')]
  85. deps = 'python (>= %s), python (<< %d.%d)' % (cv, nv[0], nv[1]+1)
  86. else:
  87. raise ValueError, 'unknown version info %s' % vinfo
  88. return deps + ', python-central (>= %s)' % req_pycentral_version
  89.  
  90. def third_party_dir(version):
  91. if version.startswith('python'):
  92. version = version[6:]
  93. if version in ('2.3', '2.4', '2.5'):
  94. return 'usr/lib/python' + version + '/site-packages'
  95. else:
  96. return 'usr/lib/python' + version + '/dist-packages'
  97.  
  98. def build_relative_link(tgt, link):
  99. t = tgt.split('/')
  100. l = link.split('/')
  101. while l[0] == t[0]:
  102. del l[0], t[0]
  103. return '/'.join(['..' for i in range(len(l)-1)] + t)
  104.  
  105. def read_dpkg_status(verbose=False):
  106. """Read the dpkg status file, return a list of packages depending
  107. on python-central and having a Python-Version information field."""
  108. packages = []
  109. rx = re.compile(r'\bpython-central\b')
  110. pkgname = version = None
  111. depends = ''
  112. status = []
  113. for line in file('/var/lib/dpkg/status'):
  114. if line.startswith('Package:'):
  115. if version != None and 'installed' in status:
  116. if 'python-support' in depends:
  117. pass
  118. elif rx.search(depends):
  119. packages.append((pkgname, version))
  120. if verbose:
  121. print " %s: %s (%s)" % (pkgname, version, status)
  122. version = None
  123. status = []
  124. pkgname = line.split(':', 1)[1].strip()
  125. elif line.startswith('Python-Version:'):
  126. version = line.split(':', 1)[1].strip()
  127. elif line.startswith('Depends:'):
  128. depends = line.split(':', 1)[1].strip()
  129. elif line.startswith('Status:'):
  130. status = line.split(':', 1)[1].strip().split()
  131. if version != None and 'installed' in status:
  132. if rx.search(depends):
  133. packages.append((pkgname, version))
  134. if verbose:
  135. print " %s: %s (%s)" % (pkgname, version, status)
  136. return packages
  137.  
  138.  
  139. class PyCentralConfigParser(SafeConfigParser):
  140. '''SafeConfigParser allowing mixed case, `:' and `=' in keys'''
  141. OPTCRE = re.compile(
  142. r'(?P<option>[^\s].*)'
  143. r'\s*(?P<vi>[:=])\s*'
  144. r'(?P<value>.*)$'
  145. )
  146. optionxform = str
  147.  
  148. class PyCentralError(Exception):
  149. """Python Central Exception"""
  150. pass
  151.  
  152. class PyCentralVersionMissingError(PyCentralError):
  153. """Python Central Version Missing Exception"""
  154. pass
  155.  
  156. class PythonRuntime:
  157. def __init__(self, name, version, interp, prefix):
  158. self.name = name
  159. self.version = version
  160. if name.startswith('python'):
  161. self.short_name = name[6:]
  162. else:
  163. self.short_name = name
  164. self.interp = interp
  165. if version in ('2.3', '2.4', '2.5'):
  166. sitedir = 'site-packages/'
  167. else:
  168. sitedir = 'dist-packages/'
  169. if prefix.endswith('/'):
  170. self.prefix = prefix + sitedir
  171. else:
  172. self.prefix = prefix + '/' + sitedir
  173.  
  174. def byte_compile_dirs(self, dirs, bc_option, exclude=None):
  175. """call compileall.py -x <exclude regexp> <dirs> according
  176. to bc_options"""
  177.  
  178. logging.debug('\tbyte-compile directories')
  179. errors = False
  180. cmd = [self.interp, self.prefix + '/compileall.py', '-q']
  181. if exclude:
  182. cmd.extend(['-x', exclude])
  183. cmd.extend(dirs)
  184. for opt in ('standard', 'optimize'):
  185. if not opt in bc_option:
  186. continue
  187. if opt == 'optimize':
  188. cmd[1:1] = ['-O']
  189. rv = os.spawnv(os.P_WAIT, self.interp, cmd[1:])
  190. if rv:
  191. raise PyCentralError
  192.  
  193. def byte_compile(self, files, bc_option, exclude=None, ignore_errors=False, force=False):
  194. errors = False
  195. if exclude:
  196. rx = re.compile(exclude)
  197. files2 = []
  198. for fn in files:
  199. mo = rx.search(fn)
  200. if mo:
  201. continue
  202. files2.append(fn)
  203. else:
  204. files2 = files
  205. if not files2:
  206. logging.info('\tno files to byte-compile')
  207. return
  208. logging.debug('\tbyte-compile files (%d/%d) %s' \
  209. % (len(files), len(files2), self.name))
  210. debug_files = files2[:min(2, len(files2))]
  211. if len(files2) > 2:
  212. debug_files.append('...')
  213. logging.debug('\t %s' % debug_files)
  214. cmd = [self.interp, '/usr/bin/py_compilefiles', '-q']
  215. if ignore_errors:
  216. cmd.append('-i')
  217. if force:
  218. cmd.append('-f')
  219. cmd.append('-')
  220. for opt in ('standard', 'optimize'):
  221. if not opt in bc_option:
  222. continue
  223. if opt == 'optimize':
  224. cmd[1:1] = ['-O']
  225. try:
  226. import subprocess
  227. p = subprocess.Popen(cmd, bufsize=1,
  228. shell=False, stdin=subprocess.PIPE)
  229. fd = p.stdin
  230. except ImportError:
  231. p = None
  232. fd = os.popen(' '.join(cmd), 'w')
  233. for fn in files2:
  234. fd.write(fn + '\n')
  235. rv = fd.close()
  236. if p:
  237. p.wait()
  238. errors = p.returncode != 0
  239. else:
  240. errors = rv != None
  241. if errors:
  242. raise PyCentralError, 'error byte-compiling files (%d)' % len(files2)
  243.  
  244. def remove_byte_code(self, files):
  245. errors = False
  246. logging.debug('\tremove byte-code files (%d)' % (len(files)))
  247. for ext in ('c', 'o'):
  248. for fn in files:
  249. fnc = fn + ext
  250. if os.path.exists(fnc):
  251. try:
  252. os.unlink(fnc)
  253. except OSError, e:
  254. print "Sorry", e
  255. errors = True
  256. if errors:
  257. raise PyCentralError, 'error removing the byte-code files'
  258.  
  259. def list_byte_code(self, files):
  260. logging.debug('\tlist byte-code files (%d)' % (len(files)))
  261. for ext in ('c', 'o'):
  262. for fn in files:
  263. fnc = fn + ext
  264. yield fnc
  265.  
  266. installed_runtimes = None
  267. default_runtime = None
  268.  
  269. def get_installed_runtimes(with_unsupported=True):
  270. global installed_runtimes
  271. global default_runtime
  272.  
  273. if not installed_runtimes:
  274. import glob
  275. installed_runtimes = []
  276. default_version = pyversions.default_version(version_only=True)
  277. supported = pyversions.supported_versions()
  278. old = pyversions.old_versions()
  279. if with_unsupported:
  280. unsupported = pyversions.unsupported_versions()
  281. else:
  282. unsupported = []
  283. for interp in glob.glob('/usr/bin/python[0-9].[0-9]'):
  284. if old and os.path.basename(interp) in old:
  285. pass
  286. #print "INFO: using old version '%s'" % interp
  287. elif unsupported and os.path.basename(interp) in unsupported:
  288. print "INFO: using unsupported version '%s'" % interp
  289. if not (os.path.basename(interp) in supported
  290. or (old and os.path.basename(interp) in old)
  291. or (unsupported and os.path.basename(interp) in unsupported)):
  292. # pycentral used to ignore everything it has no explicit
  293. # knowledge about. this causes install errors when
  294. # debian_defaults is not up-to-date (LP: #354228)
  295. if with_unsupported:
  296. print "INFO: using unknown version '%s' (debian_defaults not up-to-date?)" % interp
  297. else:
  298. print "INFO: ignoring unknown version '%s'" % interp
  299. continue
  300. version = interp[-3:]
  301. rt = PythonRuntime('python' + version,
  302. version,
  303. '/usr/bin/python' + version,
  304. '/usr/lib/python' + version)
  305. installed_runtimes.append(rt)
  306. if version == default_version:
  307. default_runtime = rt
  308. return installed_runtimes
  309.  
  310. def get_default_runtime():
  311. get_installed_runtimes()
  312. return default_runtime
  313.  
  314. def get_runtime_for_version(version):
  315. if version == 'current':
  316. return get_default_runtime()
  317. for rt in get_installed_runtimes():
  318. if rt.version == version:
  319. return rt
  320. return None
  321.  
  322. debian_config = None
  323. def get_debian_config():
  324. global debian_config
  325. if debian_config is not None:
  326. return debian_config
  327.  
  328. config = SafeConfigParser()
  329. fn = '/etc/python/debian_config'
  330. if os.path.exists(fn):
  331. try:
  332. config.readfp(open(fn))
  333. except Error:
  334. logging.error("error reading config file `%s'" % fn)
  335. sys.exit(1)
  336. # checks
  337. if not config.has_option('DEFAULT', 'byte-compile'):
  338. config.set('DEFAULT', 'byte-compile', 'standard')
  339. bc_option = config.get('DEFAULT', 'byte-compile')
  340. bc_values = set([v.strip() for v in bc_option.split(',')])
  341. bc_unknown = bc_values - set(['standard', 'optimize'])
  342. if bc_unknown:
  343. sys.stderr.write("%s: `%s': unknown values `%s'"
  344. " in `byte-compile option'\n"
  345. % (program, fn, ', '.join(list(bc_unknown))))
  346. sys.exit(1)
  347. config.set('DEFAULT', 'byte-compile', ', '.join(bc_values))
  348. if config.has_option('DEFAULT', 'overwrite-local'):
  349. val = config.get('DEFAULT', 'overwrite-local').strip().lower()
  350. overwrite_local = val in ('yes', '1', 'true')
  351. else:
  352. overwrite_local = False
  353. config.set('DEFAULT', 'overwrite-local', overwrite_local and '1' or '0')
  354. debian_config = config
  355. return debian_config
  356.  
  357. class DebPackage:
  358. def __init__(self, kind, name,
  359. oldstyle=False, default_runtime=None,
  360. pkgdir=None, pkgconfig=None, parse_versions=True):
  361. self.kind = kind
  362. self.name = name
  363. self.version_field = None
  364. self.oldstyle = oldstyle
  365. self.parse_versions = parse_versions
  366. self.default_runtime = default_runtime
  367. self.shared_prefix = None
  368. self.pkgdir = pkgdir
  369. self.pkgconfig = pkgconfig
  370. self.has_shared_extension = {}
  371. self.has_private_extension = False
  372. self.has_shared_module = {}
  373. self.has_private_module = False
  374. if pkgdir:
  375. self.read_control()
  376. else:
  377. self.read_pyfiles()
  378. #self.print_info()
  379.  
  380. def read_pyfiles(self):
  381. self.shared_files = []
  382. self.pylib_files = {}
  383. self.other_pylib_files = []
  384. self.private_files = []
  385. self.omitted_files = []
  386. self.pysupport_files = []
  387. if self.pkgdir:
  388. lines = []
  389. for root, dirs, files in os.walk(self.pkgdir):
  390. if root.endswith('DEBIAN'):
  391. continue
  392. if root != self.pkgdir:
  393. d = root[len(self.pkgdir):]
  394. lines.append(d)
  395. for name in files:
  396. lines.append(os.path.join(d, name))
  397. else:
  398. config_file = '/usr/share/pyshared-data/%s' % self.name
  399. if self.pkgconfig:
  400. lines = [fn for fn, t in self.pkgconfig.items('files')]
  401. lines.sort()
  402. elif os.path.isfile(config_file):
  403. logging.debug("reading %s" % config_file)
  404. self.pkgconfig = PyCentralConfigParser()
  405. self.pkgconfig.optionxform = str
  406. self.pkgconfig.readfp(file(config_file))
  407. if not self.pkgconfig.has_option('pycentral', 'include-links'):
  408. self.pkgconfig.set('pycentral', 'include-links', '0')
  409. elif self.pkgconfig.get('pycentral', 'include-links') == '':
  410. # LP #332532, empty value in file
  411. self.pkgconfig.set('pycentral', 'include-links', '0')
  412. lines = [fn for fn, t in self.pkgconfig.items('files')]
  413. lines.sort()
  414. else:
  415. lines = self.read_preinst_pkgconfig()
  416. if lines:
  417. pass
  418. elif os.environ.has_key("PYCENTRAL_NO_DPKG_QUERY"):
  419. logging.debug("Not using dpkg-query as requested")
  420. lines = map(string.strip, open('/var/lib/dpkg/info/%s.list' % self.name).readlines())
  421. else:
  422. cmd = ['/usr/bin/dpkg-query', '-L', self.name]
  423. try:
  424. import subprocess
  425. p = subprocess.Popen(cmd, bufsize=1,
  426. shell=False, stdout=subprocess.PIPE)
  427. fd = p.stdout
  428. except ImportError:
  429. fd = os.popen(' '.join(cmd))
  430. lines = [s[:-1] for s in fd.readlines()]
  431. if not self.pkgconfig:
  432. pc = PyCentralConfigParser()
  433. pc.optionxform = str
  434. pc.add_section('python-package')
  435. pc.set('python-package', 'format', '1')
  436. pc.add_section('pycentral')
  437. pc.set('pycentral', 'version', req_pycentral_version)
  438. pc.set('pycentral', 'include-links', '0')
  439. pc.add_section('files')
  440. for line in lines:
  441. if os.path.isdir(line) and not os.path.islink(line):
  442. pc.set('files', line, 'd')
  443. elif os.path.exists(line):
  444. pc.set('files', line, 'f')
  445. else:
  446. pass # should not happen
  447. self.pkgconfig = pc
  448.  
  449. old_shared_base = shared_base + self.name + '/site-packages/'
  450. found_old_base = found_base2 = False
  451. for line in lines:
  452. fn = line
  453. if fn.startswith(shared_base2):
  454. # keep _all_ files and directories
  455. self.shared_files.append(fn)
  456. if fn.endswith('.py'):
  457. self.has_shared_module['all'] = True
  458. found_base2 = True
  459. self.shared_prefix = shared_base2
  460. continue
  461. elif fn.startswith(old_shared_base):
  462. # keep _all_ files and directories
  463. self.shared_files.append(fn)
  464. if fn.endswith('.py'):
  465. self.has_shared_module['all'] = True
  466. found_old_base = True
  467. self.shared_prefix = old_shared_base
  468. continue
  469. elif fn.startswith('/usr/share/python-support') \
  470. or fn.startswith('/usr/lib/python-support'):
  471. self.pysupport_files.append(fn)
  472. continue
  473. elif not fn.endswith('.py') and not fn.endswith('.so'):
  474. if re.match(r'/usr/lib/python\d\.\d/', fn):
  475. self.other_pylib_files.append(fn)
  476. continue
  477. elif fn.startswith('/etc/') or fn.startswith('/usr/share/doc/'):
  478. # omit files in /etc and /usr/share/doc
  479. self.omitted_files.append(fn)
  480. continue
  481. elif re.search(r'/s?bin/', fn):
  482. # omit files located in directories
  483. self.omitted_files.append(fn)
  484. continue
  485. elif fn.startswith('/usr/lib/site-python/'):
  486. version = pyversions.default_version(version_only=True)
  487. self.pylib_files.setdefault(version, []).append(fn)
  488. continue
  489. elif re.match(r'/usr/lib/python\d\.\d/', fn):
  490. version = fn[15:18]
  491. if fn.endswith('.so'):
  492. self.has_shared_extension[version] = True
  493. if fn.endswith('.py'):
  494. self.has_shared_module[version] = True
  495. self.pylib_files.setdefault(version, []).append(fn)
  496. continue
  497. else:
  498. self.private_files.append(fn)
  499. if fn.endswith('.py'):
  500. self.has_private_module = True
  501.  
  502. if found_old_base and found_base2:
  503. raise PyCentralError, \
  504. 'shared files found in old (%s) and new locations (%s)' % (old_shared_base, shared_base2)
  505.  
  506. def read_preinst_pkgconfig(self):
  507. try:
  508. fd = open('/var/lib/dpkg/info/%s.preinst' % self.name, 'r')
  509. except:
  510. return None
  511. buffer = None
  512. for line in fd.readlines():
  513. if line == '[python-package]\n':
  514. buffer = cStringIO.StringIO()
  515. if line in ('PYEOF\n', 'EOF\n'):
  516. break
  517. if buffer:
  518. buffer.write(line)
  519. if buffer is None:
  520. return None
  521. self.pkgconfig = PyCentralConfigParser()
  522. self.pkgconfig.optionxform = str
  523. self.pkgconfig.readfp(cStringIO.StringIO(buffer.getvalue()))
  524. # never set in the preinst copy of the config file
  525. self.pkgconfig.set('pycentral', 'include-links', '0')
  526. files = [fn for fn, t in self.pkgconfig.items('files')]
  527. files.sort()
  528. return files
  529.  
  530. def print_info(self, fd=sys.stdout):
  531. fd.write('Package: %s\n' % self.name)
  532. fd.write(' shared files :%4d\n' % len(self.shared_files))
  533. fd.write(' private files :%4d\n' % len(self.private_files))
  534. for ver, files in self.pylib_files.items():
  535. fd.write(' pylib%s files:%4d\n' % (ver, len(files)))
  536.  
  537. def read_control(self):
  538. """read the debian/control file, extract the XS-Python-Version
  539. field; check that XB-Python-Version exists for the package."""
  540. if not os.path.exists('debian/control'):
  541. raise PyCentralError, "debian/control not found"
  542. self.version_field = None
  543. self.sversion_field = None
  544. try:
  545. section = None
  546. for line in file('debian/control'):
  547. line = line.strip()
  548. if line == '':
  549. section = None
  550. elif line.startswith('Source:'):
  551. section = 'Source'
  552. elif line.startswith('Package: ' + self.name):
  553. section = self.name
  554. elif line.startswith('XS-Python-Version:'):
  555. if section != 'Source':
  556. raise PyCentralError, \
  557. 'attribute XS-Python-Version not in Source section'
  558. self.sversion_field = line.split(':', 1)[1].strip()
  559. elif line.startswith('XB-Python-Version:'):
  560. if section == self.name:
  561. self.version_field = line.split(':', 1)[1].strip()
  562. except:
  563. pass
  564. if self.version_field == None:
  565. raise PyCentralVersionMissingError, \
  566. 'missing XB-Python-Version attribute in package %s' % self.name
  567. if self.sversion_field == None:
  568. raise PyCentralError, 'missing XS-Python-Version attribute'
  569. if self.parse_versions:
  570. self.sversion_info = pyversions.parse_versions(self.sversion_field)
  571. else:
  572. self.sversion_info = 'all' # dummy
  573. self.has_private_extension = self.sversion_info == 'current_ext'
  574.  
  575. def move_files(self):
  576. """move files from the installation directory to the pycentral location"""
  577. import shutil
  578.  
  579. dsttop = self.pkgdir + shared_base2
  580. try:
  581. os.makedirs(dsttop)
  582. except OSError:
  583. pass
  584. pversions = pyversions.supported_versions() \
  585. + pyversions.unsupported_versions() + pyversions.old_versions()
  586. pversions = list(set(pversions))
  587.  
  588. # rename .egg-info files and dirs, remove *.py[co] files
  589. rename = 'norename' not in os.environ.get('DH_PYCENTRAL', '')
  590. vrx = re.compile(r'(.*)(-py\d\.\d)(.*)(\.egg-info|\.pth)$')
  591. for pversion in pversions:
  592. if pversion in ('python2.3', 'python2.4', 'python2.5'):
  593. srctop = os.path.join(self.pkgdir, 'usr/lib', pversion, 'site-packages')
  594. else:
  595. # srctop and srctop3 are the old/wrong locations, move these.
  596. srctop = os.path.join(self.pkgdir, 'usr/lib', pversion, 'site-packages')
  597. srctop2 = os.path.join(self.pkgdir, 'usr/lib', pversion, 'dist-packages')
  598. srctop3 = os.path.join(self.pkgdir, 'usr/local/lib', pversion, 'dist-packages')
  599. if os.path.isdir(srctop2) and (os.path.isdir(srctop) or os.path.isdir(srctop3)):
  600. raise PyCentralError, 'both directories site-packages and dist-packages exist.'
  601. if os.path.isdir(srctop):
  602. print 'renaming %s to %s' % (srctop, srctop2)
  603. os.rename(srctop, srctop2)
  604. elif os.path.isdir(srctop3):
  605. print 'renaming %s to %s' % (srctop3, srctop2)
  606. try:
  607. os.makedirs(os.path.dirname(srctop2))
  608. except OSError:
  609. pass
  610. os.rename(srctop3, srctop2)
  611. while srctop3:
  612. srctop3 = os.path.dirname(srctop3)
  613. try:
  614. os.rmdir(srctop3)
  615. except OSError:
  616. break
  617. srctop = srctop2
  618. # do the same with -dbg packages (which don't have an Python-Version attribute)
  619. pkgdbgdir = self.pkgdir + '-dbg'
  620. if os.path.isdir(pkgdbgdir):
  621. srcdtop = os.path.join(pkgdbgdir, 'usr/lib', pversion, 'site-packages')
  622. srcdtop2 = os.path.join(pkgdbgdir, 'usr/lib', pversion, 'dist-packages')
  623. srcdtop3 = os.path.join(pkgdbgdir, 'usr/local/lib', pversion, 'dist-packages')
  624. if os.path.isdir(srcdtop2) and (os.path.isdir(srcdtop) or os.path.isdir(srcdtop3)):
  625. print 'both directories site-packages and dist-packages exist.'
  626. if os.path.isdir(srcdtop):
  627. print 'renaming %s to %s' % (srcdtop, srcdtop2)
  628. os.rename(srcdtop, srcdtop2)
  629. elif os.path.isdir(srcdtop3):
  630. print 'renaming %s to %s' % (srcdtop3, srcdtop2)
  631. try:
  632. os.makedirs(os.path.dirname(srcdtop2))
  633. except OSError:
  634. pass
  635. os.rename(srcdtop3, srcdtop2)
  636. while srcdtop3:
  637. srcdtop3 = os.path.dirname(srcdtop3)
  638. try:
  639. os.rmdir(srcdtop3)
  640. except OSError:
  641. break
  642. srcdtop = os.path.join(pkgdbgdir, 'usr/lib/debug/usr/lib', pversion, 'site-packages')
  643. srcdtop2 = os.path.join(pkgdbgdir, 'usr/lib/debug/usr/lib', pversion, 'dist-packages')
  644. srcdtop3 = os.path.join(pkgdbgdir, 'usr/lib/debug/usr/local/lib', pversion, 'dist-packages')
  645. if os.path.isdir(srcdtop2) and (os.path.isdir(srcdtop) or os.path.isdir(srcdtop3)):
  646. print 'both directories site-packages and dist-packages exist.'
  647. if os.path.isdir(srcdtop):
  648. print 'renaming %s to %s' % (srcdtop, srcdtop2)
  649. os.rename(srcdtop, srcdtop2)
  650. elif os.path.isdir(srcdtop3):
  651. print 'renaming %s to %s' % (srcdtop3, srcdtop2)
  652. try:
  653. os.makedirs(os.path.dirname(srcdtop2))
  654. except OSError:
  655. pass
  656. os.rename(srcdtop3, srcdtop2)
  657. while srcdtop3:
  658. srcdtop3 = os.path.dirname(srcdtop3)
  659. try:
  660. os.rmdir(srcdtop3)
  661. except OSError:
  662. break
  663. for root, dirs, files in os.walk(srctop, topdown=False):
  664. for name in files:
  665. m = vrx.match(name)
  666. if m and rename:
  667. name2 = ''.join(m.group(1, 3, 4))
  668. print "rename: %s -> %s" % (name, name2)
  669. os.rename(os.path.join(root, name), os.path.join (root, name2))
  670. elif name.endswith('.pyc') or name.endswith('.pyo'):
  671. os.unlink(os.path.join(root, name))
  672. for name in dirs:
  673. m = vrx.match(name)
  674. if m and rename:
  675. name2 = ''.join(m.group(1, 3, 4))
  676. print "rename: %s -> %s" % (name, name2)
  677. os.rename(os.path.join(root, name), os.path.join (root, name2))
  678.  
  679. # search for differences
  680. import filecmp
  681. class MyDircmp(filecmp.dircmp):
  682. def report(self):
  683. if self.left_only or self.right_only or self.diff_files or self.funny_files:
  684. self.differs = True
  685. print 'diff', self.left, self.right
  686. if self.left_only:
  687. self.left_only.sort()
  688. print 'Only in', self.left, ':', self.left_only
  689. if self.right_only:
  690. self.right_only.sort()
  691. print 'Only in', self.right, ':', self.right_only
  692. if self.diff_files:
  693. self.diff_files.sort()
  694. print 'Differing files :', self.diff_files
  695. if self.funny_files:
  696. self.funny_files.sort()
  697. print 'Trouble with common files :', self.funny_files
  698.  
  699. for pv1 in pversions:
  700. for pv2 in pversions[pversions.index(pv1)+1:]:
  701. site1 = os.path.join(self.pkgdir, third_party_dir(pv1))
  702. site2 = os.path.join(self.pkgdir, third_party_dir(pv2))
  703. if not (os.path.isdir(site1) and os.path.isdir(site2)):
  704. continue
  705. dco = MyDircmp(site1, site2)
  706. dco.differs = False
  707. dco.report_full_closure()
  708.  
  709. # move around
  710. for pversion in pversions:
  711. srctop = os.path.join(self.pkgdir, third_party_dir(pversion))
  712. for root, dirs, files in os.walk(srctop):
  713. if root == srctop:
  714. d = '.'
  715. else:
  716. d = root[len(srctop)+1:]
  717. for name in dirs:
  718. src = os.path.join(root, name)
  719. dst = os.path.join(dsttop, d, name)
  720. try:
  721. os.mkdir(dst)
  722. shutil.copymode(src, dst)
  723. except OSError:
  724. pass
  725. for name in files:
  726. src = os.path.join(root, name)
  727. dst = os.path.join(dsttop, d, name)
  728. if re.search(r'\.so(\.\d+)*?$', name):
  729. continue
  730. # TODO: if dst already exists, make sure, src == dst
  731. os.rename(src, dst)
  732. # remove empty dirs in /usr/lib/pythonX.Y
  733. for root, dirs, files in os.walk(self.pkgdir + '/usr/lib', topdown=False):
  734. try:
  735. if re.match("/usr/lib/python\d\.\d($|/)", root.replace(self.pkgdir, "")):
  736. os.rmdir(root)
  737. except OSError:
  738. pass
  739. try:
  740. os.rmdir(self.pkgdir + '/usr/lib')
  741. except OSError:
  742. pass
  743. # remove empty dirs in /usr/share/pyshared
  744. for root, dirs, files in os.walk(self.pkgdir + shared_base2, topdown=False):
  745. try:
  746. os.rmdir(root)
  747. except OSError:
  748. pass
  749. try:
  750. os.rmdir(self.pkgdir + shared_base2)
  751. except OSError:
  752. pass
  753.  
  754. def add_shared_links(self):
  755. versions = pyversions.requested_versions(self.sversion_field, True)
  756. tgttop = os.path.join(self.pkgdir, shared_base2.strip('/'))
  757. existing_links = []
  758. for version in versions:
  759. linktop = os.path.join(self.pkgdir, third_party_dir(version))
  760. try:
  761. os.makedirs(linktop)
  762. except OSError:
  763. pass
  764. for root, dirs, files in os.walk(tgttop, topdown=True):
  765. linkroot = root.replace(tgttop, linktop, 1)
  766. for name in files:
  767. tgt = os.path.join(root, name)
  768. link = os.path.join(linkroot, name)
  769. if os.path.exists(link):
  770. existing_links.append(link)
  771. else:
  772. rel_tgt = build_relative_link(tgt, link)
  773. os.symlink(rel_tgt, link)
  774. for name in dirs:
  775. try:
  776. os.makedirs(os.path.join(linkroot, name))
  777. except OSError:
  778. pass
  779. if existing_links:
  780. print 'Found existing files while including symlinks:'
  781. for link in existing_links:
  782. print ' ', link
  783. raise PyCentralError, 'Found existing files while including symlinks'
  784.  
  785. def gen_substvars(self):
  786. supported = [d[6:] for d in pyversions.supported_versions()
  787. if re.match(r'python\d\.\d', d)]
  788. versions = ''
  789. prversions = ''
  790. self.depends = None
  791. if len(self.has_shared_module) or len(self.has_shared_extension):
  792. # shared modules / extensions
  793. if len(self.has_shared_extension):
  794. versions = self.has_shared_extension.keys()
  795. else:
  796. if self.sversion_info in ('current', 'current_ext'):
  797. versions = 'current'
  798. elif self.sversion_info == 'all':
  799. versions = 'all'
  800. prversions = supported
  801. else:
  802. versions = self.sversion_field
  803. prversions = list(self.sversion_info.intersection(supported))
  804. self.depends = version2depends(self.sversion_info)
  805. elif self.has_private_module or self.has_private_extension:
  806. if self.sversion_info == 'all':
  807. versions = 'current'
  808. elif self.sversion_info == 'current':
  809. versions = 'current'
  810. elif self.sversion_info == 'current_ext':
  811. versions = [pyversions.default_version(version_only=True)]
  812. elif isinstance(self.sversion_info, list) or isinstance(self.sversion_info, set):
  813. # exact version info required, no enumeration, no relops
  814. if len(self.sversion_info) != 1 or not re.match(r'\d\.\d', self.sversion_info[0]):
  815. raise PyCentralError, 'no exact version for package with private modules'
  816. versions = [list(self.sversion_info)[0]]
  817. else:
  818. raise PyCentralError, 'version error for package with private modules'
  819. else:
  820. # just "copy" it from the source field
  821. if self.sversion_info == 'current':
  822. versions = 'current'
  823. elif self.sversion_info == 'current_ext':
  824. versions = [pyversions.default_version(version_only=True)]
  825. elif self.sversion_info == 'all':
  826. versions = 'all'
  827. prversions = supported
  828. else:
  829. versions = self.sversion_field
  830. prversions = list(self.sversion_info.intersection(supported))
  831. self.depends = version2depends(self.sversion_info)
  832.  
  833. if (len(self.has_shared_module) or len(self.has_shared_extension)) \
  834. and self.has_private_module or self.has_private_extension:
  835. # use case? use the information for the shared stuff
  836. pass
  837. if versions == '':
  838. raise PyCentralError, 'unable to determine Python-Version attribute'
  839. if isinstance(versions, list) or isinstance(versions, set):
  840. self.version_field = ', '.join(versions)
  841. else:
  842. self.version_field = versions
  843. if not self.depends:
  844. self.depends = version2depends(versions)
  845. if self.name.startswith('python-'):
  846. if prversions == '':
  847. prversions = versions
  848. self.provides = ', '.join([self.name.replace('python-', 'python%s-' % ver)
  849. for ver in prversions])
  850.  
  851. def set_version_field(self, version_field):
  852. self.version_field = version_field
  853. if self.parse_versions:
  854. self.version_info = pyversions.parse_versions(version_field)
  855.  
  856. def read_version_info(self, use_default_if_missing=False):
  857. """Read the Python-Version information field"""
  858. if self.version_field:
  859. return
  860. if self.pkgconfig and self.pkgconfig.has_option('python-package', 'python-version'):
  861. self.version_field = self.pkgconfig.get('python-package', 'python-version')
  862. logging.debug("Using python-version from pkgconfig: %s" % self.version_field)
  863. elif os.environ.has_key("PYCENTRAL_NO_DPKG_QUERY"):
  864. logging.debug("Not using dpkg-query as requested")
  865. needle = "Package: %s\n" % self.name
  866. for block in open("/var/lib/dpkg/status").read().split("\n\n"):
  867. if needle in block:
  868. for line in block.split("\n"):
  869. if line.startswith('Python-Version:'):
  870. self.version_field = line.split(':', 1)[1].strip()
  871. break
  872. else:
  873. logging.debug("dpkg-query -s %s" % self.name)
  874. cmd = ['/usr/bin/dpkg-query', '-s', self.name]
  875. try:
  876. import subprocess
  877. p = subprocess.Popen(cmd, bufsize=1,
  878. shell=False, stdout=subprocess.PIPE)
  879. fd = p.stdout
  880. except ImportError:
  881. fd = os.popen(' '.join(cmd))
  882.  
  883. for line in fd:
  884. if line.startswith('Python-Version:'):
  885. self.version_field = line.split(':', 1)[1].strip()
  886. break
  887. fd.close()
  888. # now verify/parse it
  889. if not self.version_field and use_default_if_missing:
  890. logging.warn("%s: has no Python-Version field, assuming default runtime" % self.name)
  891. self.version_field = get_default_runtime().version
  892. if not self.version_field:
  893.  
  894. raise PyCentralError, "package has no field Python-Version"
  895. if self.parse_versions:
  896. self.version_info = pyversions.parse_versions(self.version_field)
  897.  
  898. def set_default_runtime_from_version_info(self):
  899. versions = list(pyversions.requested_versions(self.version_field, version_only=True))
  900. if not versions:
  901. #raise PyCentralError, "no matching runtime for `%s'" % self.version_field
  902. logging.warn("%s: no matching runtime for `%s', using default"
  903. % (self.name, self.version_field))
  904. self.default_runtime = get_default_runtime()
  905. if len(versions) == 1:
  906. self.default_runtime = get_runtime_for_version(versions[0])
  907. elif pyversions.default_version(version_only=True) in versions:
  908. self.default_runtime = get_default_runtime()
  909. else:
  910. self.default_runtime = get_runtime_for_version(versions[0])
  911.  
  912. def byte_compile(self, runtimes, bc_option, exclude_regex, ignore_errors=False):
  913. """byte compiles all files not handled by pycentral"""
  914.  
  915. logging.debug(" byte-compile %s" % self.name)
  916. if self.shared_files:
  917. ppos = len(self.shared_prefix)
  918. for rt in runtimes:
  919. linked_files = [ rt.prefix + fn[ppos:]
  920. for fn in self.shared_files
  921. if fn[-3:] == '.py']
  922. rt.byte_compile(linked_files, bc_option, exclude_regex, ignore_errors)
  923.  
  924. for pyver, files in self.pylib_files.items():
  925. logging.debug("bc for v%s (%d files)" % (pyver, len(files)))
  926. rt = get_runtime_for_version(pyver)
  927. if rt in runtimes:
  928. rt.byte_compile(files, bc_option, exclude_regex)
  929.  
  930. if self.private_files:
  931. logging.debug("bc %s private (%d files)" %
  932. (self.default_runtime.version, len(self.private_files)))
  933. rt = self.default_runtime
  934. rt.byte_compile(self.private_files, bc_option, exclude_regex)
  935.  
  936. def remove_bytecode(self):
  937. """remove all byte-compiled files not handled by pycentral"""
  938.  
  939. assert self.oldstyle
  940.  
  941. logging.debug(" remove byte-code for %s" % self.name)
  942. pyfiles = []
  943. for files in self.pylib_files.values():
  944. pyfiles.extend(files)
  945. pyfiles.extend(self.private_files)
  946.  
  947. errors = False
  948. for ext in ('c', 'o'):
  949. for fn in pyfiles:
  950. fnc = fn + ext
  951. if os.path.exists(fnc):
  952. try:
  953. os.unlink(fnc)
  954. except OSError, e:
  955. print "Sorry", e
  956. errors = True
  957. if errors:
  958. raise PyCentralError
  959.  
  960. def link_shared_files(self, rt):
  961. """link shared files at runtime"""
  962. #if samefs(rt.prefix, self.shared_files[0]):
  963. # link_cmd = os.link
  964. #else:
  965. # link_cmd = os.symlink
  966. logging.debug("\tlink shared files %s/%s" % (rt.name, self.name))
  967. if not self.shared_files:
  968. return []
  969. link_cmd = os.symlink
  970. ppos = len(self.shared_prefix)
  971. existing_files = []
  972. for fn in self.shared_files:
  973. fn2 = rt.prefix + fn[ppos:]
  974. if os.path.isdir(fn) and not os.path.islink(fn):
  975. continue
  976. if os.path.exists(fn2):
  977. link = abs_link = None
  978. if os.path.islink(fn2):
  979. link = abs_link = os.readlink(fn2)
  980. if link.startswith('../'):
  981. abs_link = os.path.normpath(os.path.join(os.path.dirname(fn2), link))
  982. if abs_link == fn or link == fn:
  983. continue
  984. if not link or not (abs_link.startswith(shared_base2) or abs_link.startswith(shared_base)):
  985. existing_files.append(fn2)
  986.  
  987. if existing_files:
  988. conf = get_debian_config()
  989. overwrite_local = conf.get('DEFAULT', 'overwrite-local') == '1'
  990. if overwrite_local:
  991. print '\n '.join(["overwriting local files:"] + existing_files)
  992.  
  993. linked_files = []
  994. try:
  995. for fn in self.shared_files:
  996. fn2 = rt.prefix + fn[ppos:]
  997. if os.path.isdir(fn) and not os.path.islink(fn):
  998. if os.path.isdir(fn2):
  999. continue
  1000. os.makedirs(fn2)
  1001. linked_files.append(fn2)
  1002. else:
  1003. if os.path.exists(fn2):
  1004. msg = "already exists: %s" % fn2
  1005. link = abs_link = None
  1006. if os.path.islink(fn2):
  1007. link = abs_link = os.readlink(fn2)
  1008. if link.startswith('../'):
  1009. abs_link = os.path.normpath(os.path.join(os.path.dirname(fn2), link))
  1010. if abs_link == fn or link == fn:
  1011. linked_files.append(fn2)
  1012. continue
  1013. if not link or not (abs_link.startswith(shared_base2) or abs_link.startswith(shared_base)):
  1014. msg = msg + " -> %s" % link
  1015. if overwrite_local:
  1016. print "warning:", msg
  1017. os.unlink(fn2)
  1018. else:
  1019. continue # raise PyCentralError, msg at end of method
  1020. # make sure that fn2 really does not exist; this is a
  1021. # special hack to make pycentral work with fakechroot,
  1022. # which has a slightly weird treatment of symlinks
  1023. # now needed to switch between old and new prefix
  1024. try:
  1025. os.unlink(fn2)
  1026. except OSError:
  1027. pass
  1028. link_cmd(fn, fn2)
  1029. linked_files.append(fn2)
  1030. except PyCentralError, msg:
  1031. raise
  1032. except Exception, msg:
  1033. print msg
  1034. # FIXME: undo
  1035. linked_files.reverse()
  1036. return []
  1037. else:
  1038. if existing_files and not overwrite_local:
  1039. s = get_file_overwrite_error(existing_files)
  1040. raise PyCentralError, s
  1041. return linked_files
  1042.  
  1043. def unlink_shared_files(self, rt):
  1044. logging.debug('\tunlink_shared_files %s/%s' % (rt.name, self.name))
  1045. if not self.shared_files:
  1046. return
  1047. ppos = len(self.shared_prefix)
  1048. shared_files = self.shared_files[:]
  1049. shared_files.reverse()
  1050. for fn in shared_files:
  1051. fn2 = rt.prefix + fn[ppos:]
  1052. if os.path.isdir(fn2) and not os.path.islink(fn2):
  1053. try:
  1054. os.removedirs(fn2)
  1055. except OSError:
  1056. pass
  1057. else:
  1058. if os.path.exists(fn2):
  1059. os.unlink(fn2)
  1060.  
  1061. def list_shared_files(self, rt):
  1062. logging.debug('\tlist_shared_files %s/%s' % (rt.name, self.name))
  1063. if not self.shared_files:
  1064. return
  1065. ppos = len(self.shared_prefix)
  1066. for fn in self.shared_files:
  1067. fn2 = rt.prefix + fn[ppos:]
  1068. yield fn2
  1069.  
  1070.  
  1071. def install(self, runtimes, bc_option, exclude_regex,
  1072. byte_compile_default=True, ignore_errors=False):
  1073. logging.debug('\tinstall package %s' % self.name)
  1074. # install shared .py files
  1075. symlinks_included = self.pkgconfig.getboolean('pycentral', 'include-links')
  1076. if symlinks_included:
  1077. pass
  1078. else:
  1079. if self.shared_files:
  1080. for rt in runtimes:
  1081. linked_files = self.link_shared_files(rt)
  1082. rt.byte_compile(linked_files, bc_option, exclude_regex, ignore_errors)
  1083. # byte compile files inside prefix
  1084. if self.pylib_files:
  1085. for pyver, files in self.pylib_files.items():
  1086. rt = get_runtime_for_version(pyver)
  1087. if rt in runtimes:
  1088. rt.byte_compile(files, bc_option, exclude_regex, ignore_errors)
  1089. # byte compile with the default runtime for the package
  1090. if byte_compile_default:
  1091. if self.private_files:
  1092. self.default_runtime.byte_compile(self.private_files, bc_option,
  1093. exclude_regex, ignore_errors)
  1094.  
  1095. def prepare(self, runtimes, old_runtimes, old_pkg, ignore_errors=False):
  1096. logging.debug('\tprepare package %s' % self.name)
  1097.  
  1098. if old_pkg and old_pkg.private_files:
  1099. fs_in_old = set([fn for fn in old_pkg.private_files if fn[-3:] == '.py'])
  1100. fs_in_new = set([fn for fn in self.private_files if fn[-3:] == '.py'])
  1101. removed_fs = list(fs_in_old.difference(fs_in_new))
  1102. if removed_fs:
  1103. logging.debug_list('\t', 'removed private', removed_fs)
  1104. default_runtime.remove_byte_code(removed_fs)
  1105.  
  1106. old_pylib_fs = []
  1107. if old_pkg and old_pkg.pylib_files:
  1108. for pyver, files in old_pkg.pylib_files.items():
  1109. fs_in_old = set([fn for fn in files if fn[-3:] == '.py'])
  1110. fs_in_new = set([fn for fn in self.pylib_files.get(pyver, []) if fn[-3:] == '.py'])
  1111. removed_fs = list(fs_in_old.difference(fs_in_new))
  1112. if removed_fs:
  1113. logging.debug_list('\t', 'removed pylib', removed_fs)
  1114. default_runtime.remove_byte_code(removed_fs)
  1115. old_pylib_fs += files
  1116. old_pylib_fs += old_pkg.other_pylib_files
  1117.  
  1118. if old_pkg and old_pkg.shared_files:
  1119. for rt in old_runtimes:
  1120. if rt in runtimes:
  1121. continue
  1122. linked_files = [ rt.prefix + fn[ppos:]
  1123. for fn in old_pkg.shared_files if fn[-3:] == '.py']
  1124. if linked_files:
  1125. logging.debug_list('\t', 'removed runtimes', linked_files)
  1126. default_runtime.remove_byte_code(linked_files)
  1127. self.unlink_files(rt)
  1128.  
  1129. if not self.shared_files:
  1130. return
  1131.  
  1132. dirs_in_new = set([fn for fn, t in self.pkgconfig.items('files')
  1133. if t == 'd' if fn.startswith(self.shared_prefix)])
  1134. dirs_in_old = set()
  1135. if old_pkg:
  1136. dirs_in_old = set([fn for fn, t in old_pkg.pkgconfig.items('files')
  1137. if t == 'd' if fn.startswith(self.shared_prefix)])
  1138. new_dirs = list(dirs_in_new.difference(dirs_in_old))
  1139. new_dirs.sort()
  1140. removed_dirs = list(dirs_in_old.difference(dirs_in_new))
  1141. removed_dirs.sort()
  1142.  
  1143. fs_in_new = set([fn for fn, t in self.pkgconfig.items('files')
  1144. if t == 'f' if fn.startswith(self.shared_prefix)])
  1145. fs_in_old = set()
  1146. if old_pkg:
  1147. fs_in_old = set([fn for fn, t in old_pkg.pkgconfig.items('files')
  1148. if t == 'f' if fn.startswith(self.shared_prefix)])
  1149. new_fs = list(fs_in_new.difference(fs_in_old))
  1150. new_fs.sort()
  1151. removed_fs = list(fs_in_old.difference(fs_in_new))
  1152. removed_fs.sort()
  1153.  
  1154. logging.debug_list('\t', 'new dirs', new_dirs)
  1155. logging.debug_list('\t', 'removed dirs', removed_dirs)
  1156. logging.debug_list('\t', 'new files', new_fs)
  1157. logging.debug_list('\t', 'removed files', removed_fs)
  1158.  
  1159. link_cmd = os.symlink
  1160. ppos = len(self.shared_prefix)
  1161. existing_files = []
  1162. for rt in runtimes:
  1163. for f1 in new_fs:
  1164. f2 = rt.prefix + f1[ppos:]
  1165. if os.path.exists(f2):
  1166. link = abs_link = None
  1167. if os.path.islink(f2):
  1168. link = abs_link = os.readlink(f2)
  1169. if link.startswith('../'):
  1170. abs_link = os.path.normpath(os.path.join(os.path.dirname(f2), link))
  1171. if abs_link == f1 or link == f1:
  1172. continue
  1173. if not link or not (abs_link.startswith(shared_base2) or abs_link.startswith(shared_base)):
  1174. existing_files.append(f2)
  1175. if existing_files:
  1176. # if the current installed version does not use python-central
  1177. # then having a file here is expected and harmless
  1178. if not self.name in [p for (p,v) in read_dpkg_status()]:
  1179. logging.info("%s: upgrade from package version not using python-central" % self.name)
  1180. return
  1181. # if all existing files are found in the old package in
  1182. # /usr/lib/pythonX.Y/site-packages, and moved to the shared area,
  1183. # do nothing.
  1184. not_in_same_pkg = set(existing_files)
  1185. if old_pkg:
  1186. not_in_same_pkg.difference_update(old_pylib_fs)
  1187. if not_in_same_pkg == set():
  1188. logging.info("%s: upgrade from package version with unmoved files" % self.name)
  1189. return
  1190. conf = get_debian_config()
  1191. overwrite_local = conf.get('DEFAULT', 'overwrite-local') == '1'
  1192. if overwrite_local:
  1193. print "overwriting local files"
  1194.  
  1195. for rt in runtimes:
  1196. for d1 in new_dirs:
  1197. d2 = rt.prefix + d1[ppos:]
  1198. if os.path.isdir(d2):
  1199. continue
  1200. os.makedirs(d2)
  1201. for f1 in new_fs:
  1202. f2 = rt.prefix + f1[ppos:]
  1203. if os.path.exists(f2):
  1204. msg = "already exists: %s" % f2
  1205. link = abs_link = None
  1206. if os.path.islink(f2):
  1207. link = abs_link = os.readlink(f2)
  1208. if link.startswith('../'):
  1209. abs_link = os.path.normpath(os.path.join(os.path.dirname(f2), link))
  1210. if abs_link == f1 or link == f1:
  1211. continue
  1212. if not link or not (abs_link.startswith(shared_base2) or abs_link.startswith(shared_base)):
  1213. msg = msg + " -> %s" % link
  1214. if overwrite_local:
  1215. print "warning:", msg
  1216. os.unlink(f2)
  1217. else:
  1218. continue # raise PyCentralError, msg at end of loop
  1219. # hack to make pycentral work with fakechroot
  1220. # now needed to switch between old and new prefix
  1221. try:
  1222. os.unlink(f2)
  1223. except OSError:
  1224. pass
  1225. try:
  1226. d = os.path.dirname(f2)
  1227. if not os.path.isdir(d):
  1228. print "create directory %s" % d
  1229. os.makedirs(d)
  1230. link_cmd(f1, f2)
  1231. except OSError:
  1232. print "unable to create symlink %s" % f2
  1233. raise
  1234. if existing_files and not overwrite_local:
  1235. raise PyCentralError, "not overwriting local files"
  1236.  
  1237. for rt in runtimes:
  1238. for f1 in removed_fs:
  1239. f2 = rt.prefix + f1[ppos:]
  1240. try:
  1241. os.unlink(f2)
  1242. os.unlink(f2 + 'c')
  1243. os.unlink(f2 + 'o')
  1244. except OSError:
  1245. pass
  1246. for d1 in removed_dirs:
  1247. d2 = rt.prefix + d1[ppos:]
  1248. try:
  1249. os.rmdir(d2)
  1250. except OSError:
  1251. pass
  1252.  
  1253. return
  1254.  
  1255. def remove(self, runtimes, remove_script_files=True):
  1256. logging.debug('\tremove package %s' % self.name)
  1257. # remove shared .py files
  1258. symlinks_included = self.pkgconfig.getboolean('pycentral', 'include-links')
  1259. if symlinks_included:
  1260. pass
  1261. else:
  1262. if self.shared_files:
  1263. ppos = len(self.shared_prefix)
  1264. for rt in runtimes:
  1265. linked_files = [ rt.prefix + fn[ppos:]
  1266. for fn in self.shared_files
  1267. if fn[-3:] == '.py']
  1268. #print self.shared_files
  1269. #print linked_files
  1270. default_runtime.remove_byte_code(linked_files)
  1271. self.unlink_shared_files(rt)
  1272. # remove byte compiled files inside prefix
  1273. if self.pylib_files:
  1274. for pyver, files in self.pylib_files.items():
  1275. rt = get_runtime_for_version(pyver)
  1276. if rt in runtimes:
  1277. default_runtime.remove_byte_code(files)
  1278. # remove byte code for script files
  1279. if remove_script_files:
  1280. if self.private_files:
  1281. default_runtime.remove_byte_code(self.private_files)
  1282.  
  1283. def list(self, runtimes, list_script_files=True):
  1284. logging.debug('\tlist package %s' % self.name)
  1285. # list shared .py files
  1286. symlinks_included = self.pkgconfig.getboolean('pycentral', 'include-links')
  1287. if not symlinks_included and self.shared_files:
  1288. ppos = len(self.shared_prefix)
  1289. for rt in runtimes:
  1290. linked_files = [ rt.prefix + fn[ppos:]
  1291. for fn in self.shared_files
  1292. if fn[-3:] == '.py']
  1293. for f in default_runtime.list_byte_code(linked_files):
  1294. yield f
  1295. for f in self.list_shared_files(rt):
  1296. yield f
  1297. # list byte compiled files inside prefix
  1298. if self.pylib_files:
  1299. for pyver, files in self.pylib_files.items():
  1300. rt = get_runtime_for_version(pyver)
  1301. if rt in runtimes:
  1302. for f in default_runtime.list_byte_code(files):
  1303. yield f
  1304. # list byte code for script files
  1305. if list_script_files:
  1306. if self.private_files:
  1307. for f in default_runtime.list_byte_code(self.private_files):
  1308. yield f
  1309.  
  1310. def update_bytecode_files(self, runtimes, rt_default, bc_option):
  1311. # byte-compile with default python version
  1312. logging.debug('\tupdate byte-code for %s' % self.name)
  1313. exclude_regex = None
  1314. # update shared .py files
  1315. if self.shared_files:
  1316. ppos = len(self.shared_prefix)
  1317. for rt in runtimes:
  1318. if rt == rt_default:
  1319. linked_files = self.link_shared_files(rt)
  1320. rt.byte_compile(linked_files, bc_option, exclude_regex)
  1321. else:
  1322. linked_files = [ rt.prefix + fn[ppos:]
  1323. for fn in self.shared_files
  1324. if fn[-3:] == '.py']
  1325. rt.remove_byte_code(linked_files)
  1326. self.unlink_shared_files(rt)
  1327. # byte compile with the default runtime for the package
  1328. if self.private_files:
  1329. self.default_runtime.byte_compile(self.private_files,
  1330. bc_option, exclude_regex, force=True)
  1331.  
  1332. known_actions = {}
  1333. def register_action(action_class):
  1334. known_actions[action_class.name] = action_class
  1335.  
  1336. class Action:
  1337. _option_parser = None
  1338. name = None
  1339. help = ""
  1340. usage = "<options>"
  1341. def __init__(self):
  1342. self.errors_occured = 0
  1343. parser = self.get_option_parser()
  1344. parser.set_usage(
  1345. 'usage: %s [<options> ...] %s %s' % (program, self.name, self.usage))
  1346.  
  1347. def get_option_parser(self):
  1348. if not self._option_parser:
  1349. p = OptionParser()
  1350. self._option_parser = p
  1351. return self._option_parser
  1352.  
  1353. def info(self, msg, stream=sys.stderr):
  1354. logging.info('%s %s: %s' % (program, self.name, msg))
  1355.  
  1356. def warn(self, msg, stream=sys.stderr):
  1357. logging.warn('%s %s: %s' % (program, self.name, msg))
  1358.  
  1359. def error(self, msg, stream=sys.stderr, go_on=False):
  1360. logging.error('%s %s: %s' % (program, self.name, msg))
  1361. self.errors_occured += 1
  1362. if not go_on:
  1363. sys.exit(1)
  1364.  
  1365. def parse_args(self, arguments):
  1366. self.options, self.args = self._option_parser.parse_args(arguments)
  1367. return self.options, self.args
  1368.  
  1369. def check_args(self, global_options):
  1370. return self.errors_occured
  1371.  
  1372. def run(self, global_opts):
  1373. pass
  1374.  
  1375.  
  1376. class ActionByteCompile(Action):
  1377. """byte compile the *.py files in <package> using the the
  1378. default python version (or use the version specified with -v.
  1379. Any additional directory arguments are ignored (only files
  1380. found in the package are byte compiled. Files in
  1381. /usr/lib/pythonX.Y are compiled with the matching python version.
  1382.  
  1383. bccompile is a replacement for the current byte compilation
  1384. generated by the dh_python debhelper script.
  1385. """
  1386. name = 'bccompile'
  1387. help = 'byte compile .py files in a package'
  1388. usage = '[<options>] <package> [<dir> ...]'
  1389.  
  1390. def get_option_parser(self):
  1391. if not self._option_parser:
  1392. p = OptionParser()
  1393. p.add_option('-x', '--exclude',
  1394. help="skip files matching the regular expression",
  1395. default=None, action='store', dest='exclude')
  1396. p.add_option('-V', '--version',
  1397. help="byte compile using this python version",
  1398. default='current', action='store', dest='version')
  1399. self._option_parser = p
  1400. return self._option_parser
  1401.  
  1402. def check_args(self, global_options):
  1403. if len(self.args) < 1:
  1404. self._option_parser.print_help()
  1405. sys.exit(1)
  1406. self.pkgname = self.args[0]
  1407. self.runtime = get_runtime_for_version(self.options.version)
  1408. if not self.runtime:
  1409. self.error("unknown runtime version %s" % self.options.version)
  1410.  
  1411. if not os.path.exists('/var/lib/dpkg/info/%s.list' % self.pkgname):
  1412. self.error("package %s is not installed" % self.pkgname)
  1413. self.pkg = DebPackage('package', self.pkgname, oldstyle=False,
  1414. default_runtime=self.runtime)
  1415. self.pkg.read_version_info()
  1416. return self.errors_occured
  1417.  
  1418. def run(self, global_options):
  1419. logging.debug('bccompile %s' % self.pkgname)
  1420. runtimes = get_installed_runtimes()
  1421. config = get_debian_config()
  1422. bc_option = config.get('DEFAULT', 'byte-compile')
  1423. requested = pyversions.requested_versions_for_runtime(self.pkg.version_field, version_only=True)
  1424. used_runtimes = [rt for rt in runtimes if rt.short_name in requested]
  1425. # called with directories as arguments
  1426. if 0 and self.directories:
  1427. try:
  1428. for version, dirs in self.pylib_dirs.items():
  1429. rt = get_runtime_for_version(version)
  1430. rt.byte_compile_dirs(dirs, bc_option, self.options.exclude)
  1431. if self.private_dirs:
  1432. version = pkg.version_field
  1433. if version == 'current':
  1434. version = pyversions.default_version(version_only=True)
  1435. rt = get_runtime_for_version(version)
  1436. rt.byte_compile_dirs(private_dirs, bc_option, self.options.exclude)
  1437. except PyCentralError:
  1438. self.error("error byte-compiling package `%s'" % self.pkgname)
  1439. return
  1440.  
  1441. try:
  1442. self.pkg.byte_compile(used_runtimes, bc_option, self.options.exclude)
  1443. except PyCentralError:
  1444. self.error("error byte-compiling package `%s'" % self.pkgname)
  1445.  
  1446. register_action(ActionByteCompile)
  1447.  
  1448. class ActionPkgInstall(Action):
  1449. name = 'pkginstall'
  1450. help = 'make a package available for all supported runtimes'
  1451. usage = '[<options>] <package>'
  1452.  
  1453. def get_option_parser(self):
  1454. if not self._option_parser:
  1455. p = OptionParser()
  1456. p.add_option('-x', '--exclude',
  1457. help="skip files matching the regular expression",
  1458. default=None, action='store', dest='exclude')
  1459. self._option_parser = p
  1460. return self._option_parser
  1461.  
  1462. def check_args(self, global_options):
  1463. if len(self.args) != 1:
  1464. self._option_parser.print_help()
  1465. sys.exit(1)
  1466. self.pkgname = self.args[0]
  1467. if not os.path.exists('/var/lib/dpkg/info/%s.list' % self.pkgname):
  1468. self.error("package %s is not installed" % self.pkgname)
  1469. return self.errors_occured
  1470.  
  1471. def run(self, global_options):
  1472. runtimes = get_installed_runtimes()
  1473. config = get_debian_config()
  1474. bc_option = config.get('DEFAULT', 'byte-compile')
  1475. pkg = DebPackage('package', self.args[0], oldstyle=False)
  1476. pkg.read_version_info()
  1477. requested = pyversions.requested_versions_for_runtime(pkg.version_field, version_only=True)
  1478. used_runtimes = [rt for rt in runtimes if rt.short_name in requested]
  1479. try:
  1480. pkg.set_default_runtime_from_version_info()
  1481. except ValueError:
  1482. # Package doesn't provide support for any supported runtime
  1483. if len(used_runtimes) == 0:
  1484. self.error('%s needs unavailable runtime (%s)'
  1485. % (self.pkgname, pkg.version_field))
  1486. else:
  1487. # Still byte compile for the available runtimes (with the
  1488. # first matching runtime)
  1489. pkg.default_runtime = get_runtime_for_version(used_runtimes[0])
  1490. logging.debug('\tavail=%s, pkg=%s, install=%s'
  1491. % ([rt.short_name for rt in runtimes],
  1492. pkg.version_field,
  1493. [rt.short_name for rt in used_runtimes]))
  1494. try:
  1495. pkg.install(used_runtimes, bc_option,
  1496. self.options.exclude, byte_compile_default=True)
  1497. except PyCentralError, msg:
  1498. self.error(msg)
  1499.  
  1500.  
  1501. register_action(ActionPkgInstall)
  1502.  
  1503.  
  1504. class ActionPkgPrepare(Action):
  1505. name = 'pkgprepare'
  1506. help = 'prepare a package for all supported runtimes'
  1507. usage = '[<options>] <package>'
  1508.  
  1509. def get_option_parser(self):
  1510. if not self._option_parser:
  1511. p = OptionParser()
  1512. p.add_option('-x', '--exclude',
  1513. help="skip files matching the regular expression",
  1514. default=None, action='store', dest='exclude')
  1515. self._option_parser = p
  1516. return self._option_parser
  1517.  
  1518. def check_args(self, global_options):
  1519. if len(self.args) != 1:
  1520. self._option_parser.print_help()
  1521. sys.exit(1)
  1522. self.pkgname = self.args[0]
  1523. # FIXME: run from the preinst, package may not exist
  1524. #if not os.path.exists('/var/lib/dpkg/info/%s.list' % self.pkgname):
  1525. # self.error("package %s is not installed" % self.pkgname)
  1526. return self.errors_occured
  1527.  
  1528. def run(self, global_options):
  1529. # functionality disabled starting with 0.6.9
  1530. return
  1531.  
  1532. runtimes = get_installed_runtimes()
  1533. config = get_debian_config()
  1534. pkgconfig = PyCentralConfigParser()
  1535. pkgconfig.optionxform = str
  1536. pkgconfig.readfp(sys.stdin)
  1537. version_field = pkgconfig.get('python-package', 'python-version')
  1538. try:
  1539. requested = pyversions.requested_versions_for_runtime(version_field, version_only=True)
  1540. except pyversions.PyCentralEmptyValueError, msg:
  1541. # cannot install yet; remove the symlinked files and byte code files from the old
  1542. # version, rely on the pkginstall in the postinst.
  1543. print "pycentral: required runtimes not yet installed, skip pkgprepare, call pkgremove"
  1544. runtimes = get_installed_runtimes(with_unsupported=True)
  1545. pkg = DebPackage('package', self.args[0], oldstyle=False)
  1546. pkg.read_version_info()
  1547. pkg.default_runtime = get_default_runtime()
  1548. try:
  1549. pkg.remove(runtimes, remove_script_files=True)
  1550. except PyCentralError, msg:
  1551. self.warn(msg)
  1552. return
  1553. used_runtimes = [rt for rt in runtimes if rt.short_name in requested]
  1554. pkg = DebPackage('package', self.args[0], oldstyle=False, pkgconfig=pkgconfig)
  1555. pkg.set_version_field(version_field)
  1556. try:
  1557. pkg.set_default_runtime_from_version_info()
  1558. except ValueError:
  1559. # Package doesn't provide support for any supported runtime
  1560. if len(used_runtimes) == 0:
  1561. self.error('%s needs unavailable runtime (%s)'
  1562. % (self.pkgname, pkg.version_field))
  1563. else:
  1564. # Still byte compile for the available runtimes (with the
  1565. # first matching runtime)
  1566. pkg.default_runtime = get_runtime_for_version(used_runtimes[0])
  1567.  
  1568. if os.path.exists('/var/lib/dpkg/info/%s.list' % self.pkgname):
  1569. old_pkg = DebPackage('package', self.args[0], oldstyle=False)
  1570. try:
  1571. old_pkg.read_version_info()
  1572. except PyCentralError:
  1573. old_pkg.set_version_field(version_field)
  1574.  
  1575. old_requested = pyversions.requested_versions_for_runtime(old_pkg.version_field, version_only=True)
  1576. old_used_runtimes = [rt for rt in runtimes if rt.short_name in requested]
  1577. else:
  1578. old_pkg = None
  1579. old_used_runtimes = []
  1580. logging.debug('\tavail=%s, pkg=%s, prepare=%s'
  1581. % ([rt.short_name for rt in runtimes],
  1582. version_field,
  1583. [rt.short_name for rt in used_runtimes]))
  1584. try:
  1585. pkg.prepare(used_runtimes, old_used_runtimes, old_pkg)
  1586. except PyCentralError, msg:
  1587. self.error(msg)
  1588.  
  1589. register_action(ActionPkgPrepare)
  1590.  
  1591.  
  1592. class ActionBCRemove(Action):
  1593. """remove the byte-compiled files in <package>.
  1594.  
  1595. bccompile is a replacement for the current byte compilation
  1596. generated by the dh_python debhelper script.
  1597. """
  1598. name = 'bcremove'
  1599. help = 'remove the byte compiled .py files'
  1600. usage = '<package>'
  1601.  
  1602. def check_args(self, global_options):
  1603. if len(self.args) != 1:
  1604. self._option_parser.print_help()
  1605. sys.exit(1)
  1606. self.pkgname = self.args[0]
  1607. if not os.path.exists('/var/lib/dpkg/info/%s.list' % self.pkgname):
  1608. self.error("package %s is not installed" % self.pkgname)
  1609. return self.errors_occured
  1610.  
  1611. def run(self, global_options):
  1612. pkg = DebPackage('package', self.args[0], oldstyle=True)
  1613. try:
  1614. pkg.remove_bytecode()
  1615. except PyCentralError, msg:
  1616. self.error(msg)
  1617.  
  1618. register_action(ActionBCRemove)
  1619.  
  1620.  
  1621. class ActionPkgRemove(Action):
  1622. """
  1623. """
  1624. name = 'pkgremove'
  1625. help = 'remove a package installed for all supported runtimes'
  1626. usage = '<package>'
  1627.  
  1628. def check_args(self, global_options):
  1629. if len(self.args) != 1:
  1630. self._option_parser.print_help()
  1631. sys.exit(1)
  1632. self.pkgname = self.args[0]
  1633. if not os.path.exists('/var/lib/dpkg/info/%s.list' % self.pkgname):
  1634. self.error("package %s is not installed" % self.pkgname)
  1635. return self.errors_occured
  1636.  
  1637. def run(self, global_options):
  1638. runtimes = get_installed_runtimes(with_unsupported=True)
  1639. pkg = DebPackage('package', self.args[0], oldstyle=False)
  1640. pkg.read_version_info(use_default_if_missing=True)
  1641. try:
  1642. pkg.set_default_runtime_from_version_info()
  1643. except ValueError:
  1644. # original runtime is already removed, use the default for removal
  1645. pkg.default_runtime = get_default_runtime()
  1646. try:
  1647. pkg.remove(runtimes, remove_script_files=True)
  1648. except PyCentralError, msg:
  1649. self.error(msg)
  1650.  
  1651. register_action(ActionPkgRemove)
  1652.  
  1653. class ActionPkgList(Action):
  1654. name = 'pkglist'
  1655. help = 'list all pycentral-managed files for <package>'
  1656. usage = '<package>'
  1657.  
  1658. def check_args(self, global_options):
  1659. if len(self.args) != 1:
  1660. self._option_parser.print_help()
  1661. sys.exit(1)
  1662. self.pkgname = self.args[0]
  1663. if not os.path.exists('/var/lib/dpkg/info/%s.list' % self.pkgname):
  1664. self.error("package %s is not installed" % self.pkgname)
  1665. return self.errors_occured
  1666.  
  1667. def run(self, global_options):
  1668. runtimes = get_installed_runtimes(with_unsupported=True)
  1669. pkg = DebPackage('package', self.args[0], oldstyle=False)
  1670. pkg.read_version_info()
  1671. try:
  1672. pkg.set_default_runtime_from_version_info()
  1673. except ValueError:
  1674. # original runtime may be removed, use the default
  1675. pkg.default_runtime = get_default_runtime()
  1676. try:
  1677. for f in pkg.list(runtimes, list_script_files=True):
  1678. print f
  1679. except PyCentralError, msg:
  1680. self.error(msg)
  1681.  
  1682. register_action(ActionPkgList)
  1683.  
  1684.  
  1685. class ActionRuntimeInstall(Action):
  1686. name = 'rtinstall'
  1687. help = 'make installed packages available for this runtime'
  1688.  
  1689. def get_option_parser(self):
  1690. if not self._option_parser:
  1691. p = OptionParser()
  1692. p.add_option('-i', '--ignore-errors',
  1693. help="ignore errors during byte compilations",
  1694. default=None, action='store', dest='ignore')
  1695. self._option_parser = p
  1696. return self._option_parser
  1697.  
  1698. def check_args(self, global_options):
  1699. if len(self.args) != 1:
  1700. self._option_parser.print_help()
  1701. sys.exit(1)
  1702. self.rtname = self.args[0]
  1703. if self.rtname[-8:] == '-minimal':
  1704. self.rtname = self.rtname[:-8]
  1705. self.runtime = None
  1706. for rt in get_installed_runtimes():
  1707. if rt.name == self.rtname:
  1708. self.runtime = rt
  1709. break
  1710. if not self.runtime:
  1711. self.error('installed runtime %s not found' % self.rtname)
  1712. self.rtname_short = self.rtname[6:]
  1713. return self.errors_occured
  1714.  
  1715. def run(self, global_options):
  1716. packages = [(p, v) for p, v in read_dpkg_status()
  1717. if not p in (self.rtname, self.rtname+'-minimal')]
  1718. needed_packages = []
  1719. for pkgname, vstring in packages:
  1720. try:
  1721. requested = list(pyversions.requested_versions(vstring, version_only=True))
  1722. if not self.rtname_short in requested:
  1723. logging.info("%s not in what requested_versions() returned, adding it anyway" % self.rtname_short)
  1724. requested.append(self.rtname_short)
  1725. except ValueError:
  1726. logging.info('\tunsupported for %s: %s (%s)' % (self.rtname, pkgname, vstring))
  1727. continue
  1728. if self.runtime.short_name in requested:
  1729. needed_packages.append((pkgname, vstring, requested))
  1730. logging.info('\t%d packages with Python-Version info installed, %d for %s'
  1731. % (len(packages), len(needed_packages), self.rtname))
  1732.  
  1733. # XXX not needed for an upgrade of a runtime
  1734. byte_compile_for_default = (self.runtime == default_runtime)
  1735.  
  1736. bc_option = get_debian_config().get('DEFAULT', 'byte-compile')
  1737. for pkgname, vstring, vinfo in needed_packages:
  1738. try:
  1739. logging.info('\tsupport %s for %s' % (pkgname, self.rtname))
  1740. pkg = DebPackage('package', pkgname, oldstyle=False)
  1741. pkg.read_version_info()
  1742. try:
  1743. pkg.set_default_runtime_from_version_info()
  1744. except ValueError:
  1745. logging.warn('\t%s not available for %s (%s)'
  1746. % (pkgname, self.rtname, pkg.version_field))
  1747. pkg.install([self.runtime], bc_option, None,
  1748. byte_compile_for_default,
  1749. ignore_errors = self.options.ignore != None)
  1750. except PyCentralError, msg:
  1751. self.error('package %s: %s' % (pkgname, msg))
  1752.  
  1753. register_action(ActionRuntimeInstall)
  1754.  
  1755. class ActionRuntimeRemove(Action):
  1756. name = 'rtremove'
  1757. help = 'remove packages installed for this runtime'
  1758.  
  1759. def check_args(self, global_options):
  1760. if len(self.args) != 1:
  1761. self._option_parser.print_help()
  1762. sys.exit(1)
  1763. self.rtname = self.args[0]
  1764. if self.rtname[-8:] == '-minimal':
  1765. self.rtname = self.rtname[:-8]
  1766. self.runtime = None
  1767. for rt in get_installed_runtimes(with_unsupported=True):
  1768. if rt.name == self.rtname:
  1769. self.runtime = rt
  1770. break
  1771. if not self.runtime:
  1772. self.error('installed runtime %s not found' % self.rtname)
  1773. return self.errors_occured
  1774.  
  1775. def run(self, global_options):
  1776. packages = [(p, v) for p, v in read_dpkg_status(verbose=True)
  1777. if not p in (self.rtname, self.rtname+'-minimal')]
  1778. needed_packages = []
  1779. import subprocess
  1780. for pkgname, vstring in packages:
  1781. if not os.path.exists('/var/lib/dpkg/info/%s.list' % pkgname):
  1782. # already removed, but /var/lib/dpkg/status not yet updated
  1783. continue
  1784. cmd = ['/usr/bin/dpkg-query', '-W', '-f', '${Status}\n', pkgname]
  1785. p = subprocess.Popen(cmd, bufsize=1,
  1786. shell=False, stdout=subprocess.PIPE)
  1787. fd = p.stdout
  1788. status = fd.readline().strip().split()
  1789. fd.close()
  1790. if not 'installed' in status:
  1791. # already removed, but /var/lib/dpkg/status not yet updated
  1792. continue
  1793. try:
  1794. requested = list(pyversions.requested_versions_for_runtime(vstring, version_only=True))
  1795. except ValueError:
  1796. logging.info('\tunsupported for %s: %s (%s)' % (self.rtname, pkgname, vstring))
  1797. continue
  1798. if self.runtime.short_name in requested:
  1799. needed_packages.append((pkgname, vstring, requested))
  1800. logging.info('\t%d pycentral supported packages installed, %d for %s'
  1801. % (len(packages), len(needed_packages), self.rtname))
  1802. failed = []
  1803. for pkgname, vstring, vinfo in needed_packages:
  1804. logging.info('\trtremove: remove package %s for %s' % (pkgname, self.rtname))
  1805. pkg = DebPackage('package', pkgname)
  1806. pkg.set_version_field(vstring)
  1807. try:
  1808. pkg.set_default_runtime_from_version_info()
  1809. except ValueError:
  1810. # original runtime is already removed, use the default for removal
  1811. pkg.default_runtime = get_default_runtime()
  1812. try:
  1813. pkg.remove([self.runtime], remove_script_files=False)
  1814. except PyCentralError, msg:
  1815. self.error('failed to remove %s support for package %s' % (self.rtname, pkgname), go_on=True)
  1816. failed.append(pkgname)
  1817. if failed:
  1818. self.error('failed to remove %s support for %d packages' % len(failed))
  1819.  
  1820. register_action(ActionRuntimeRemove)
  1821.  
  1822.  
  1823. class ActionUpdateDefault(Action):
  1824. name = 'updatedefault'
  1825. help = 'update the default python version'
  1826. usage = '<old runtime> <new runtime>'
  1827.  
  1828. def check_args(self, global_options):
  1829. if len(self.args) != 2:
  1830. self._option_parser.print_help()
  1831. sys.exit(1)
  1832. self.oldrtname = self.args[0]
  1833. self.rtname = self.args[1]
  1834. packages = read_dpkg_status()
  1835. self.needed_packages = []
  1836. for pkgname, vstring in packages:
  1837. if vstring.find('current') == -1:
  1838. continue
  1839. try:
  1840. versions = pyversions.requested_versions(vstring, version_only=True)
  1841. except ValueError:
  1842. self.error("package %s is not ready to be updated for %s"
  1843. % (pkgname, self.rtname))
  1844. continue
  1845. pkg = DebPackage('package', pkgname)
  1846. self.needed_packages.append(pkg)
  1847. return self.errors_occured
  1848.  
  1849. def run(self, global_options):
  1850. logging.info('\tupdate default: update %d packages for %s'
  1851. % (len(self.needed_packages), self.rtname))
  1852. runtimes = get_installed_runtimes()
  1853. default_rt = get_default_runtime()
  1854. bc_option = get_debian_config().get('DEFAULT', 'byte-compile')
  1855. try:
  1856. for pkg in self.needed_packages:
  1857. pkg.read_version_info()
  1858. pkg.set_default_runtime_from_version_info()
  1859. if pkg.shared_files or pkg.private_files:
  1860. pkg.update_bytecode_files(runtimes, default_rt, bc_option)
  1861. except PyCentralError, msg:
  1862. self.error(msg)
  1863.  
  1864. register_action(ActionUpdateDefault)
  1865.  
  1866.  
  1867. class ActionList(Action):
  1868. name = 'list'
  1869. help = 'List all pycentral-managed files'
  1870.  
  1871. def check_args(self, global_options):
  1872. if len(self.args) != 0:
  1873. self._option_parser.print_help()
  1874. sys.exit(1)
  1875. return self.errors_occured
  1876.  
  1877. def run(self, global_options):
  1878. runtimes = get_installed_runtimes(with_unsupported=True)
  1879. for (p, v) in read_dpkg_status():
  1880. pkg = DebPackage('package', p)
  1881. pkg.read_version_info()
  1882. for f in pkg.list(runtimes, list_script_files=True):
  1883. print f
  1884.  
  1885. register_action(ActionList)
  1886.  
  1887.  
  1888. class ActionShowDefault(Action):
  1889. name = 'showdefault'
  1890. help = 'Show default python version number'
  1891.  
  1892. def check_args(self, global_options):
  1893. if len(self.args) != 0:
  1894. self._option_parser.print_help()
  1895. sys.exit(1)
  1896. return self.errors_occured
  1897.  
  1898. def run(self, global_options):
  1899. print pyversions.default_version(version_only=True)
  1900. sys.stderr.write("pycentral showdefault is deprecated, use `pyversions -vd'\n")
  1901.  
  1902. register_action(ActionShowDefault)
  1903.  
  1904.  
  1905. class ActionShowVersions(Action):
  1906. name = 'showversions'
  1907. help = 'Show version numbers of supported python versions'
  1908.  
  1909. def check_args(self, global_options):
  1910. if len(self.args) != 0:
  1911. self._option_parser.print_help()
  1912. sys.exit(1)
  1913.  
  1914. return self.errors_occured
  1915.  
  1916. def run(self, global_options):
  1917. supported = pyversions.supported_versions()
  1918. versions = [d[6:] for d in supported if re.match(r'python\d\.\d', d)]
  1919. print ' '.join(versions)
  1920. sys.stderr.write("pycentral showversions is deprecated, use `pyversions -vs'\n")
  1921.  
  1922. register_action(ActionShowVersions)
  1923.  
  1924. class ActionShowSupported(Action):
  1925. name = 'showsupported'
  1926. help = 'Show the supported python versions'
  1927.  
  1928. def check_args(self, global_options):
  1929. if len(self.args) != 0:
  1930. self._option_parser.print_help()
  1931. sys.exit(1)
  1932. return self.errors_occured
  1933.  
  1934. def run(self, global_options):
  1935. supported = pyversions.supported_versions()
  1936. print ' '.join(supported)
  1937. sys.stderr.write("pycentral showsupported is deprecated, use `pyversions -s'\n")
  1938.  
  1939. register_action(ActionShowSupported)
  1940.  
  1941.  
  1942. class ActionPyCentralDir(Action):
  1943. name = 'pycentraldir'
  1944. help = 'Show the pycentral installation directory for the package'
  1945. usage = '<package>'
  1946.  
  1947. def check_args(self, global_options):
  1948. if len(self.args) != 1:
  1949. self._option_parser.print_help()
  1950. sys.exit(1)
  1951. self.pkgname = self.args[0]
  1952. return self.errors_occured
  1953.  
  1954. def run(self, global_options):
  1955. if shared_base2[-1] == '/':
  1956. print shared_base2[:-1]
  1957. else:
  1958. print shared_base2
  1959.  
  1960. register_action(ActionPyCentralDir)
  1961.  
  1962.  
  1963. class ActionVersion(Action):
  1964. name = 'version'
  1965. help = 'Show the pycentral version'
  1966.  
  1967. def check_args(self, global_options):
  1968. if len(self.args) != 0:
  1969. self._option_parser.print_help()
  1970. sys.exit(1)
  1971.  
  1972. return self.errors_occured
  1973.  
  1974. def run(self, global_options):
  1975. sys.stdout.write("%s\n" % pycentral_version)
  1976.  
  1977. register_action(ActionVersion)
  1978.  
  1979.  
  1980. class ActionDebhelper(Action):
  1981. name = 'debhelper'
  1982. help = 'move files to pycentral location, variable substitutions'
  1983. usage = '[-p|--provides] [--no-move|--include-links] <package> [<package directory>]'
  1984.  
  1985. def get_option_parser(self):
  1986. if not self._option_parser:
  1987. envvar = os.environ.get('DH_PYCENTRAL', '')
  1988. substvars_default = 'no'
  1989. if 'substvars=file' in envvar:
  1990. substvars_default = 'file'
  1991. if 'substvars=stdout' in envvar:
  1992. substvars_default = 'stdout'
  1993.  
  1994. p = OptionParser()
  1995. p.add_option('-p', '--provides',
  1996. help="generate substitution for python:Provides",
  1997. default='add-provides' in envvar, action='store_true', dest='provides')
  1998. p.add_option('--no-move', '--nomove',
  1999. help="do not move files to pycentral location",
  2000. default='no-move' in envvar or 'nomove' in envvar, action='store_true', dest='nomove')
  2001. p.add_option('--include-links',
  2002. help="include links to the pycentral location",
  2003. default='include-links' in envvar, action='store_true', dest='includelinks')
  2004. p.add_option('--stdout',
  2005. help="just print substitution variables to stdout",
  2006. default='stdout' in envvar, action='store_true', dest='stdout')
  2007. p.add_option('--substvars',
  2008. help="where to print substitution vars (no, file, stdout)",
  2009. default=substvars_default, dest='substvars')
  2010. p.add_option('--no-act', '--dry-run',
  2011. help="dry run",
  2012. default=('dry-run' in envvar) or ('no-act' in envvar),
  2013. action='store_true', dest='dryrun')
  2014. self._option_parser = p
  2015. return self._option_parser
  2016.  
  2017. def check_args(self, global_options):
  2018. if not len(self.args) in (1, 2):
  2019. self._option_parser.print_help()
  2020. sys.exit(1)
  2021. if 'file' in self.options.substvars:
  2022. self.options.substvars = 'file'
  2023. if 'stdout' in self.options.substvars:
  2024. self.options.substvars = 'stdout'
  2025. if self.options.nomove and self.options.includelinks:
  2026. self.error("options `--no-move' and `--include-links' conflict")
  2027. return self.errors_occured
  2028.  
  2029. def run(self, global_options):
  2030. if len(self.args) < 2:
  2031. pkgdir = 'debian/' + self.args[0]
  2032. else:
  2033. pkgdir = self.args[1]
  2034. try:
  2035. pkg = DebPackage('package', self.args[0], pkgdir=pkgdir,
  2036. parse_versions=self.options.substvars!='no')
  2037. if not self.options.nomove:
  2038. pkg.move_files()
  2039. if self.options.includelinks:
  2040. pkg.add_shared_links()
  2041. pkg.read_pyfiles()
  2042. if self.options.substvars!='no':
  2043. pkg.gen_substvars()
  2044. except PyCentralVersionMissingError, msg:
  2045. self.warn(msg)
  2046. return
  2047. except PyCentralError, msg:
  2048. self.error(msg)
  2049.  
  2050. out = None
  2051. if self.options.stdout or self.options.substvars == 'stdout':
  2052. out = sys.stdout
  2053. elif self.options.substvars == 'file':
  2054. out = file('debian/%s.substvars' % pkg.name, 'a+')
  2055. if out:
  2056. out.write('python:Versions=%s\n' % pkg.version_field)
  2057. out.write('python:Depends=%s\n' % pkg.depends)
  2058. out.write('python:Provides=%s\n' % pkg.provides)
  2059.  
  2060.  
  2061. register_action(ActionDebhelper)
  2062.  
  2063. # match a string with the list of available actions
  2064. def action_matches(action, actions):
  2065. prog = re.compile('[^-]*?-'.join(action.split('-')))
  2066. return [a for a in actions if prog.match(a)]
  2067.  
  2068. def usage(stream, msg=None):
  2069. print >>stream, msg
  2070. print >>stream, "use `%s help' for help on actions and arguments" % program
  2071. print >>stream
  2072. sys.exit(1)
  2073.  
  2074. # parse command line arguments
  2075. def parse_options(args):
  2076. shortusage = 'usage: %s [<option> ...] <action> <pkgname>' % program
  2077. parser = OptionParser(usage=shortusage)
  2078. parser.disable_interspersed_args()
  2079.  
  2080. # setup the parsers object
  2081. parser.remove_option('-h')
  2082. parser.add_option('-h', '--help',
  2083. help='help screen',
  2084. action='store_true', dest='help')
  2085. parser.add_option('-v', '--verbose',
  2086. help='verbose mode',
  2087. action='store_true', dest='verbose')
  2088.  
  2089. global_options, args = parser.parse_args()
  2090. # Print the help screen and exit
  2091. if len(args) == 0 or global_options.help:
  2092. parser.print_help()
  2093. print "\nactions:"
  2094. action_names = known_actions.keys()
  2095. action_names.sort()
  2096. for n in action_names:
  2097. print " %-21s %s" % (n, known_actions[n].help)
  2098. print ""
  2099. sys.exit(1)
  2100.  
  2101. # check if the specified action really exists
  2102. action_name = args[0]
  2103. del args[0]
  2104. matching_actions = action_matches(action_name, known_actions.keys())
  2105. if len(matching_actions) == 0:
  2106. usage(sys.stderr, "unknown action `%s'" % action_name)
  2107. elif len(matching_actions) > 1:
  2108. usage(sys.stderr,
  2109. "ambiguous action `%s', matching actions: %s"
  2110. % (action_name, strlist(matching_actions)))
  2111. else:
  2112. action_name = matching_actions[0]
  2113.  
  2114. # instantiate an object for the action and parse the remaining arguments
  2115. action = known_actions[action_name]()
  2116. action_options, action_names = action.parse_args(args)
  2117.  
  2118. return global_options, action
  2119.  
  2120. class Logging:
  2121. DEBUG, INFO, WARN, ERROR = range(4)
  2122. def __init__(self, level=WARN):
  2123. self.fd = None
  2124. self.level = level
  2125. try:
  2126. self.fd = file('/var/log/pycentral.log', 'a+')
  2127. except IOError:
  2128. self.fd = None
  2129. def msg(self, level, s):
  2130. if level < self.level:
  2131. return
  2132. d = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
  2133. if self.fd:
  2134. self.fd.write('%s %s %s\n' % (d, level, s))
  2135. self.fd.flush()
  2136. sys.stdout.write('pycentral: %s\n' % (s))
  2137. sys.stdout.flush()
  2138. def info(self, s):
  2139. self.msg(self.INFO, s)
  2140. def warn(self, s):
  2141. self.msg(self.WARN, s)
  2142. def error(self, s):
  2143. self.msg(self.ERROR, s)
  2144. sys.stderr.write('%s\n' % s)
  2145. def debug(self, s):
  2146. self.msg(self.DEBUG, s)
  2147.  
  2148. def debug_list(self, tab, s, l, n=4):
  2149. l2 = l[:min(n, len(l))]
  2150. if len(l) > n:
  2151. l2.append('...')
  2152. self.msg(self.DEBUG, "%s%s (%s/%s)" % (tab, s, len(l2), len(l)))
  2153. if len(l2) > 0:
  2154. logging.debug('%s %s' % (tab, l2))
  2155.  
  2156. def setup_logging(loglevel=Logging.WARN, verbose=False):
  2157. levels = ['debug', 'info', 'warn', 'error']
  2158. env_level = os.environ.get('PYCENTRAL', 'warn').lower()
  2159. for i in range(len(levels)):
  2160. if env_level.find(levels[i]) != -1:
  2161. loglevel = i
  2162. if verbose:
  2163. loglevel = Logging.DEBUG
  2164. global logging
  2165. logging = Logging(loglevel)
  2166.  
  2167. def main():
  2168. global_options, action = parse_options(sys.argv[1:])
  2169.  
  2170. os.umask(0022)
  2171.  
  2172. # setup logging stuff
  2173. setup_logging(Logging.WARN, global_options.verbose)
  2174. if action.name == 'debhelper' or action.name.startswith('show'):
  2175. pass
  2176. else:
  2177. logging.debug('pycentral ' + ' '.join(sys.argv[1:]))
  2178.  
  2179. # check the arguments according to the action called
  2180. if action.check_args(global_options):
  2181. sys.exit(1)
  2182.  
  2183. # run the action and exit
  2184. rv = action.run(global_options)
  2185. sys.exit(rv)
  2186.  
  2187.  
  2188. # call the main routine
  2189. if __name__ == '__main__':
  2190. main()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement