pymanager/src/manage/uninstall_command.py at main · python/pymanager · GitHub

Repository navigation

History

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

193

194

195

196

197

198

199

200

201

202

203

204

205

206

207

208

209

210

211

212

213

214

215

216

217

218

219

220

221

222

223

224

225

226

227

228

229

230

231

232

233

234

235

236

237

238

239

240

241

242

243

244

245

246

247

248

249

250

251

252

253

254

255

256

257

258

259

260

from .exceptions import ArgumentError, FilesInUseError

from .fsutils import rmtree, unlink

from .installs import get_matching_install_tags

from .install_command import SHORTCUT_HANDLERS, update_all_shortcuts

from .logging import LOGGER

from .pathutils import Path, PurePath

from .tagutils import tag_or_range

def _iterdir(p, only_files=False):

try:

if only_files:

return [f for f in Path(p).iterdir() if f.is_file()]

return list(Path(p).iterdir())

except FileNotFoundError:

LOGGER.debug("Skipping %s because it does not exist", p)

return []

def _is_relative_to(path, root):

path = Path(path)

root = Path(root)

try:

relative = path.relative_to(root)

except ValueError:

return False

return root / relative == path

def _remove_unrecognized(root, preserve, warn_msg):

to_remove = []

for path in _iterdir(root):

if path in preserve:

continue

nested_preserve = {p for p in preserve if _is_relative_to(p, path)}

if nested_preserve:

_remove_unrecognized(path, nested_preserve, warn_msg)

else:

to_remove.append(path)

if to_remove:

LOGGER.info("Removing cached and unrecognized files from %s", root)

for path in to_remove:

rmtree(path, after_5s_warning=warn_msg.format(

"cached and unrecognized files"

))

def _do_purge_global_dir(global_dir, warn_msg, *, hive=None, subkey="Environment"):

import winreg

if hive is None:

hive = winreg.HKEY_CURRENT_USER

try:

with winreg.OpenKeyEx(hive, subkey) as key:

path, kind = winreg.QueryValueEx(key, "Path")

if kind not in (winreg.REG_SZ, winreg.REG_EXPAND_SZ):

raise ValueError("Value kind is not a string")

except (OSError, ValueError):

LOGGER.debug("Not removing global commands directory from PATH", exc_info=True)

else:

LOGGER.debug("Current PATH contains %s", path)

paths = path.split(";")

newpaths = []

for p in paths:

# We should expand entries here, but we only want to remove those

# that we added ourselves (during firstrun), and we never use

# environment variables. So even if the kind is REG_EXPAND_SZ, we

# don't need to expand to find our own entry.

#ep = os.path.expandvars(p) if kind == winreg.REG_EXPAND_SZ else p

ep = p

if PurePath(ep).match(global_dir):

LOGGER.debug("Removing from PATH: %s", p)

else:

newpaths.append(p)

if len(newpaths) < len(paths):

newpath = ";".join(newpaths)

with winreg.CreateKeyEx(hive, subkey, access=winreg.KEY_READ|winreg.KEY_WRITE) as key:

path2, kind2 = winreg.QueryValueEx(key, "Path")

if path2 == path and kind2 == kind:

LOGGER.info("Removing global commands directory from PATH")

LOGGER.debug("New PATH contains %s", newpath)

winreg.SetValueEx(key, "Path", 0, kind, newpath)

else:

LOGGER.debug("Not removing global commands directory from PATH "

"because the registry changed while processing.")

try:

from _native import broadcast_settings_change

broadcast_settings_change()

except (ImportError, OSError):

LOGGER.debug("Did not broadcast settings change notification",

exc_info=True)

if not global_dir.is_dir():

return

LOGGER.info("Purging global commands from %s", global_dir)

rmtree(global_dir, after_5s_warning=warn_msg)

def execute(cmd):

LOGGER.debug("BEGIN uninstall_command.execute: %r", cmd.args)

warn_msg = ("Attempting to remove {} is taking longer than expected. " +

"Ensure no Python interpreters are running, and continue to wait " +

"or press Ctrl+C to abort.")

if cmd.purge and cmd.cleanup:

raise ArgumentError("--purge and --cleanup cannot be combined.")

if (cmd.purge or cmd.cleanup) and cmd.args:

option = "--purge" if cmd.purge else "--cleanup"

raise ArgumentError(f"{option} does not accept runtime tags.")

# Clear any active venv so we don't try to delete it

cmd.virtual_env = None

installed = list(cmd.get_installs())

cmd.tags = []

if cmd.cleanup:

prompt = ("Clean up cached files and unrecognized content? "

"This will preserve recognized runtimes.")

if not cmd.ask_yn(prompt):

LOGGER.debug("END uninstall_command.execute")

return

recognized = {Path(i["prefix"]) for i in installed}

roots = {Path(cmd.install_dir), Path(cmd.download_dir)}

for root in roots:

if any(_is_relative_to(root, other)

for other in roots if other != root):

continue

if any(_is_relative_to(path, root) for path in recognized):

_remove_unrecognized(root, recognized, warn_msg)

else:

LOGGER.info(

"Removing cached and unrecognized files from %s", root

)

rmtree(root, after_5s_warning=warn_msg.format(

"cached and unrecognized files"

))

update_all_shortcuts(cmd)

LOGGER.debug("END uninstall_command.execute")

return

if cmd.purge:

if not cmd.ask_yn("Uninstall all runtimes?"):

LOGGER.debug("END uninstall_command.execute")

return

installs_in_use = set()

for i in installed:

LOGGER.info("Purging %s from %s", i["display-name"], i["prefix"])

try:

rmtree(

i["prefix"],

after_5s_warning=warn_msg.format(i["display-name"]),

remove_ext_first=("exe", "dll", "json")

)

except FilesInUseError:

LOGGER.warn("Unable to purge %s because it is still in use.",

i["display-name"])

installs_in_use.add(Path(i["prefix"]))

continue

LOGGER.info("Purging saved downloads from %s", cmd.download_dir)

rmtree(cmd.download_dir, after_5s_warning=warn_msg.format("cached downloads"))

# Purge global commands directory

_do_purge_global_dir(cmd.global_dir, warn_msg.format("global commands"))

LOGGER.info("Purging all shortcuts")

for _, cleanup in SHORTCUT_HANDLERS.values():

if cleanup:

cleanup(cmd, [])

unknown = [p for p in _iterdir(cmd.install_dir)

if p not in installs_in_use]

if unknown:

LOGGER.info("Purging unrecognized files from %s", cmd.install_dir)

for p in unknown:

rmtree(p, after_5s_warning=warn_msg.format(

"unrecognized files"

))

LOGGER.debug("END uninstall_command.execute")

return

if not cmd.args:

raise ArgumentError("Please specify one or more runtimes to uninstall.")

to_uninstall = []

if not cmd.by_id:

for tag in cmd.args:

try:

if tag.casefold() == "default".casefold():

cmd.tags.append(tag_or_range(cmd.default_tag))

else:

cmd.tags.append(tag_or_range(tag))

except ValueError as ex:

LOGGER.warn("%s", ex)

for tag in cmd.tags:

candidates = get_matching_install_tags(

installed,

tag,

default_platform=cmd.default_platform,

)

if not candidates:

LOGGER.warn("No install found matching '%s'", tag)

continue

i, _ = candidates[0]

LOGGER.debug("Selected %s (%s) to uninstall", i["display-name"], i["id"])

to_uninstall.append(i)

installed.remove(i)

else:

ids = {tag.casefold() for tag in cmd.args}

for i in installed:

if i["id"].casefold() in ids:

LOGGER.debug("Selected %s (%s) to uninstall", i["display-name"], i["id"])

to_uninstall.append(i)

for i in to_uninstall:

installed.remove(i)

if not to_uninstall:

LOGGER.info("No runtimes selected to uninstall.")

return

elif cmd.confirm:

if len(to_uninstall) == 1:

if not cmd.ask_yn("Uninstall %s?", to_uninstall[0]["display-name"]):

return

else:

msg = ", ".join(i["display-name"] for i in to_uninstall)

if not cmd.ask_yn("Uninstall these runtimes: %s?", msg):

return

for i in to_uninstall:

LOGGER.debug("Uninstalling %s from %s", i["display-name"], i["prefix"])

try:

rmtree(

i["prefix"],

after_5s_warning=warn_msg.format(i["display-name"]),

remove_ext_first=("exe", "dll", "json"),

)

except FilesInUseError as ex:

LOGGER.error("Could not uninstall %s because it is still in use.",

i["display-name"])

raise SystemExit(1) from ex

LOGGER.info("Removed %s", i["display-name"])

try:

for target in cmd.global_dir.glob("*.__target__"):

alias = target.with_suffix("")

entry = target.read_text(encoding="utf-8-sig", errors="strict")

if PurePath(entry).match(i["executable"]):

LOGGER.debug("Unlink %s", alias)

unlink(alias, after_5s_warning=warn_msg.format(alias))

unlink(target, after_5s_warning=warn_msg.format(target))

except OSError as ex:

LOGGER.warn("Failed to remove alias: %s", ex)

LOGGER.debug("TRACEBACK:", exc_info=True)

if to_uninstall:

update_all_shortcuts(cmd)

LOGGER.debug("END uninstall_command.execute")

pymanager/src/manage/uninstall_command.py at main · python/pymanager · GitHub

Repository navigation

Expand file tree

Breadcrumbs

uninstall_command.py

Latest commit

History

Breadcrumbs

uninstall_command.py

File metadata and controls