logo

oasis-root

Compiled tree of Oasis Linux based on own branch at <https://hacktivis.me/git/oasis/> git clone https://anongit.hacktivis.me/git/oasis-root.git

netrc.py (5747B)


  1. """An object-oriented interface to .netrc files."""
  2. # Module and documentation by Eric S. Raymond, 21 Dec 1998
  3. import os, shlex, stat
  4. __all__ = ["netrc", "NetrcParseError"]
  5. class NetrcParseError(Exception):
  6. """Exception raised on syntax errors in the .netrc file."""
  7. def __init__(self, msg, filename=None, lineno=None):
  8. self.filename = filename
  9. self.lineno = lineno
  10. self.msg = msg
  11. Exception.__init__(self, msg)
  12. def __str__(self):
  13. return "%s (%s, line %s)" % (self.msg, self.filename, self.lineno)
  14. class netrc:
  15. def __init__(self, file=None):
  16. default_netrc = file is None
  17. if file is None:
  18. file = os.path.join(os.path.expanduser("~"), ".netrc")
  19. self.hosts = {}
  20. self.macros = {}
  21. try:
  22. with open(file, encoding="utf-8") as fp:
  23. self._parse(file, fp, default_netrc)
  24. except UnicodeDecodeError:
  25. with open(file, encoding="locale") as fp:
  26. self._parse(file, fp, default_netrc)
  27. def _parse(self, file, fp, default_netrc):
  28. lexer = shlex.shlex(fp)
  29. lexer.wordchars += r"""!"#$%&'()*+,-./:;<=>?@[\]^_`{|}~"""
  30. lexer.commenters = lexer.commenters.replace('#', '')
  31. while 1:
  32. # Look for a machine, default, or macdef top-level keyword
  33. saved_lineno = lexer.lineno
  34. toplevel = tt = lexer.get_token()
  35. if not tt:
  36. break
  37. elif tt[0] == '#':
  38. if lexer.lineno == saved_lineno and len(tt) == 1:
  39. lexer.instream.readline()
  40. continue
  41. elif tt == 'machine':
  42. entryname = lexer.get_token()
  43. elif tt == 'default':
  44. entryname = 'default'
  45. elif tt == 'macdef': # Just skip to end of macdefs
  46. entryname = lexer.get_token()
  47. self.macros[entryname] = []
  48. lexer.whitespace = ' \t'
  49. while 1:
  50. line = lexer.instream.readline()
  51. if not line or line == '\012':
  52. lexer.whitespace = ' \t\r\n'
  53. break
  54. self.macros[entryname].append(line)
  55. continue
  56. else:
  57. raise NetrcParseError(
  58. "bad toplevel token %r" % tt, file, lexer.lineno)
  59. # We're looking at start of an entry for a named machine or default.
  60. login = ''
  61. account = password = None
  62. self.hosts[entryname] = {}
  63. while 1:
  64. tt = lexer.get_token()
  65. if (tt.startswith('#') or
  66. tt in {'', 'machine', 'default', 'macdef'}):
  67. if password:
  68. self.hosts[entryname] = (login, account, password)
  69. lexer.push_token(tt)
  70. break
  71. else:
  72. raise NetrcParseError(
  73. "malformed %s entry %s terminated by %s"
  74. % (toplevel, entryname, repr(tt)),
  75. file, lexer.lineno)
  76. elif tt == 'login' or tt == 'user':
  77. login = lexer.get_token()
  78. elif tt == 'account':
  79. account = lexer.get_token()
  80. elif tt == 'password':
  81. if os.name == 'posix' and default_netrc:
  82. prop = os.fstat(fp.fileno())
  83. if prop.st_uid != os.getuid():
  84. import pwd
  85. try:
  86. fowner = pwd.getpwuid(prop.st_uid)[0]
  87. except KeyError:
  88. fowner = 'uid %s' % prop.st_uid
  89. try:
  90. user = pwd.getpwuid(os.getuid())[0]
  91. except KeyError:
  92. user = 'uid %s' % os.getuid()
  93. raise NetrcParseError(
  94. ("~/.netrc file owner (%s) does not match"
  95. " current user (%s)") % (fowner, user),
  96. file, lexer.lineno)
  97. if (prop.st_mode & (stat.S_IRWXG | stat.S_IRWXO)):
  98. raise NetrcParseError(
  99. "~/.netrc access too permissive: access"
  100. " permissions must restrict access to only"
  101. " the owner", file, lexer.lineno)
  102. password = lexer.get_token()
  103. else:
  104. raise NetrcParseError("bad follower token %r" % tt,
  105. file, lexer.lineno)
  106. def authenticators(self, host):
  107. """Return a (user, account, password) tuple for given host."""
  108. if host in self.hosts:
  109. return self.hosts[host]
  110. elif 'default' in self.hosts:
  111. return self.hosts['default']
  112. else:
  113. return None
  114. def __repr__(self):
  115. """Dump the class data in the format of a .netrc file."""
  116. rep = ""
  117. for host in self.hosts.keys():
  118. attrs = self.hosts[host]
  119. rep += f"machine {host}\n\tlogin {attrs[0]}\n"
  120. if attrs[1]:
  121. rep += f"\taccount {attrs[1]}\n"
  122. rep += f"\tpassword {attrs[2]}\n"
  123. for macro in self.macros.keys():
  124. rep += f"macdef {macro}\n"
  125. for line in self.macros[macro]:
  126. rep += line
  127. rep += "\n"
  128. return rep
  129. if __name__ == '__main__':
  130. print(netrc())