Coverage for sources/appcore/distribution.py: 100%

1# vim: set filetype=python fileencoding=utf-8: 

2# -*- coding: utf-8 -*- 

3 

4#============================================================================# 

5# # 

6# Licensed under the Apache License, Version 2.0 (the "License"); # 

7# you may not use this file except in compliance with the License. # 

8# You may obtain a copy of the License at # 

9# # 

10# http://www.apache.org/licenses/LICENSE-2.0 # 

11# # 

12# Unless required by applicable law or agreed to in writing, software # 

13# distributed under the License is distributed on an "AS IS" BASIS, # 

14# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # 

15# See the License for the specific language governing permissions and # 

16# limitations under the License. # 

17# # 

18#============================================================================# 

19 

20 

21''' Information about package distribution. ''' 

22 

23 

24from . import __ 

25from . import exceptions as _exceptions 

26from . import io as _io 

27 

28 

29class Information( __.immut.DataclassObject ): 

30 ''' Information about a package distribution. ''' 

31 

32 name: str 

33 location: __.Path 

34 editable: bool 

35 

36 @classmethod 

37 async def prepare( 

38 selfclass, 

39 exits: __.ctxl.AsyncExitStack, 

40 anchor: __.Absential[ __.Path ] = __.absent, 

41 package: __.Absential[ str ] = __.absent, 

42 ) -> __.typx.Self: 

43 ''' Acquires information about package distribution. ''' 

44 if ( __.is_absent( anchor ) 

45 and getattr( __.sys, 'frozen', False ) 

46 and hasattr( __.sys, '_MEIPASS' ) 

47 ): # pragma: no cover 

48 location, name = await _acquire_pyinstaller_information( ) 

49 return selfclass( 

50 editable = False, location = location, name = name ) 

51 if __.is_absent( package ): 

52 package, _ = _discover_invoker_location( ) 

53 if not __.is_absent( package ): 

54 # TODO: Python 3.12: importlib.metadata 

55 from importlib_metadata import packages_distributions 

56 name = packages_distributions( ).get( package ) 

57 if name: 

58 location = ( 

59 await _acquire_production_location( package, exits ) ) 

60 return selfclass( 

61 editable = False, location = location, name = name[ 0 ] ) 

62 # https://github.com/pypa/packaging-problems/issues/609 

63 # Development sources rather than distribution. 

64 # Implies no use of importlib.resources. 

65 if __.is_absent( anchor ): 

66 _, anchor = _discover_invoker_location( ) 

67 location, name = ( 

68 await _acquire_development_information( anchor = anchor ) ) 

69 return selfclass( 

70 editable = True, location = location, name = name ) 

71 

72 def provide_data_location( self, *appendages: str ) -> __.Path: 

73 ''' Provides location of distribution data. ''' 

74 base = self.location / 'data' 

75 if appendages: return base.joinpath( *appendages ) 

76 return base 

77 

78 

79async def _acquire_development_information( 

80 anchor: __.Path 

81) -> tuple[ __.Path, str ]: 

82 location = _locate_pyproject( anchor ) 

83 pyproject = await _io.acquire_text_file_async( 

84 location / 'pyproject.toml', deserializer = __.tomli.loads ) 

85 name = pyproject[ 'project' ][ 'name' ] 

86 return location, name 

87 

88 

89async def _acquire_production_location( 

90 package: str, exits: __.ctxl.AsyncExitStack 

91) -> __.Path: 

92 # TODO: Python 3.12: importlib.resources 

93 from importlib_resources import files, as_file # pyright: ignore 

94 # Extract package contents to temporary directory, if necessary. 

95 return exits.enter_context( 

96 as_file( files( package ) ) ) # pyright: ignore 

97 

98 

99async def _acquire_pyinstaller_information( # pragma: no cover 

100) -> tuple[ __.Path, str ]: 

101 anchor_ = __.Path( 

102 getattr( __.sys, '_MEIPASS' ) ) 

103 # TODO: More rigorously determine package name. 

104 # Currently assumes 'pyproject.toml' is present in distribution. 

105 return await _acquire_development_information( anchor = anchor_ ) 

106 

107 

108def _discover_invoker_location( ) -> tuple[ __.Absential[ str ], __.Path ]: 

109 ''' Discovers file path of caller for project root detection. ''' 

110 import inspect 

111 import site 

112 import sysconfig 

113 package_location = __.Path( __file__ ).parent.resolve( ) 

114 stdlib_locations = { 

115 __.Path( sysconfig.get_path( 'stdlib' ) ).resolve( ), 

116 __.Path( sysconfig.get_path( 'platstdlib' ) ).resolve( ), 

117 } 

118 sp_locations: set[ __.Path ] = set( ) 

119 for path in site.getsitepackages( ): 

120 sp_locations.add( __.Path( path ).resolve( ) ) 

121 with __.ctxl.suppress( AttributeError ): 

122 sp_locations.add( __.Path( site.getusersitepackages( ) ).resolve( ) ) 

123 frame = inspect.currentframe( ) 

124 if frame is None: return __.absent, __.Path.cwd( ) 

125 # Walk up the call stack to find frame outside of this package. 

126 while True: 

127 frame = frame.f_back 

128 if frame is None: break # pragma: no cover 

129 location = __.Path( frame.f_code.co_filename).resolve( ) 

130 # Skip frames within this package 

131 if location.is_relative_to( package_location ): # pragma: no cover 

132 continue 

133 # Allow site-packages even if they're under stdlib paths 

134 if any( 

135 location.is_relative_to( sp_location ) 

136 for sp_location in sp_locations 

137 ): pass 

138 # Skip standard library paths 

139 elif any( 

140 location.is_relative_to( stdlib_location ) 

141 for stdlib_location in stdlib_locations 

142 ): continue 

143 mname = frame.f_globals.get( '__module__' ) 

144 if not mname: continue 

145 pname = mname.split( '.', maxsplit = 1 )[ 0 ] 

146 return pname, location.parent 

147 # Fallback location is current working directory. 

148 return __.absent, __.Path.cwd( ) # pragma: no cover 

149 

150 

151def _locate_pyproject( project_anchor: __.Path ) -> __.Path: 

152 ''' Finds project manifest, if it exists. Errors otherwise. ''' 

153 initial = project_anchor.resolve( ) 

154 current = initial if initial.is_dir( ) else initial.parent 

155 limits: set[ __.Path ] = set( ) 

156 for limits_variable in ( 'GIT_CEILING_DIRECTORIES', ): 

157 limits_value = __.os.environ.get( limits_variable ) 

158 if not limits_value: continue # pragma: no cover 

159 limits.update( # pragma: no cover 

160 __.Path( limit ).resolve( ) 

161 for limit in limits_value.split( ':' ) if limit.strip( ) ) 

162 while current != current.parent: # Not at filesystem root 

163 if ( current / 'pyproject.toml' ).exists( ): 

164 return current 

165 if current in limits: 

166 raise _exceptions.FileLocateFailure( # noqa: TRY003 # pragma: no cover 

167 'project root discovery', 'pyproject.toml' ) 

168 current = current.parent 

169 raise _exceptions.FileLocateFailure( # noqa: TRY003 # pragma: no cover 

170 'project root discovery', 'pyproject.toml' )