pre-commit/pre_commit/main.py at wip_progress_dots · pre-commit/pre-commit · 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

261

262

263

264

265

266

267

268

269

270

271

272

273

274

275

276

277

278

279

280

281

282

283

284

285

286

287

288

289

290

291

292

293

294

295

296

297

298

299

300

301

302

303

304

305

306

307

308

309

310

311

312

313

314

315

316

317

318

319

320

321

322

323

324

325

326

327

328

329

330

331

332

333

334

335

336

337

338

339

340

341

342

343

344

345

346

347

348

from __future__ import unicode_literals

import argparse

import logging

import os

import sys

import pre_commit.constants as C

from pre_commit import color

from pre_commit import five

from pre_commit import git

from pre_commit.commands.autoupdate import autoupdate

from pre_commit.commands.clean import clean

from pre_commit.commands.gc import gc

from pre_commit.commands.init_templatedir import init_templatedir

from pre_commit.commands.install_uninstall import install

from pre_commit.commands.install_uninstall import install_hooks

from pre_commit.commands.install_uninstall import uninstall

from pre_commit.commands.migrate_config import migrate_config

from pre_commit.commands.run import run

from pre_commit.commands.sample_config import sample_config

from pre_commit.commands.try_repo import try_repo

from pre_commit.error_handler import error_handler

from pre_commit.error_handler import FatalError

from pre_commit.logging_handler import logging_handler

from pre_commit.store import Store

from pre_commit.util import CalledProcessError

logger = logging.getLogger('pre_commit')

# https://github.com/pre-commit/pre-commit/issues/217

# On OSX, making a virtualenv using pyvenv at . causes `virtualenv` and `pip`

# to install packages to the wrong place. We don't want anything to deal with

# pyvenv

os.environ.pop('__PYVENV_LAUNCHER__', None)

COMMANDS_NO_GIT = {'clean', 'gc', 'init-templatedir', 'sample-config'}

def _add_color_option(parser):

parser.add_argument(

'--color', default=os.environ.get('PRE_COMMIT_COLOR', 'auto'),

type=color.use_color,

metavar='{' + ','.join(color.COLOR_CHOICES) + '}',

help='Whether to use color in output. Defaults to `%(default)s`.',

)

def _add_config_option(parser):

parser.add_argument(

'-c', '--config', default=C.CONFIG_FILE,

help='Path to alternate config file',

)

class AppendReplaceDefault(argparse.Action):

def __init__(self, *args, **kwargs):

super(AppendReplaceDefault, self).__init__(*args, **kwargs)

self.appended = False

def __call__(self, parser, namespace, values, option_string=None):

if not self.appended:

setattr(namespace, self.dest, [])

self.appended = True

getattr(namespace, self.dest).append(values)

def _add_hook_type_option(parser):

parser.add_argument(

'-t', '--hook-type', choices=(

'pre-commit', 'pre-push', 'prepare-commit-msg', 'commit-msg',

action=AppendReplaceDefault,

default=['pre-commit'],

dest='hook_types',

)

def _add_run_options(parser):

parser.add_argument('hook', nargs='?', help='A single hook-id to run')

parser.add_argument('--verbose', '-v', action='store_true', default=False)

parser.add_argument(

'--origin', '-o',

help="The origin branch's commit_id when using `git push`.",

)

parser.add_argument(

'--source', '-s',

help="The remote branch's commit_id when using `git push`.",

)

parser.add_argument(

'--commit-msg-filename',

help='Filename to check when running during `commit-msg`',

)

parser.add_argument(

'--hook-stage', choices=C.STAGES, default='commit',

help='The stage during which the hook is fired. One of %(choices)s',

)

parser.add_argument(

'--show-diff-on-failure', action='store_true',

help='When hooks fail, run `git diff` directly afterward.',

)

mutex_group = parser.add_mutually_exclusive_group(required=False)

mutex_group.add_argument(

'--all-files', '-a', action='store_true', default=False,

help='Run on all the files in the repo.',

)

mutex_group.add_argument(

'--files', nargs='*', default=[],

help='Specific filenames to run hooks on.',

)

def _adjust_args_and_chdir(args):

# `--config` was specified relative to the non-root working directory

if os.path.exists(args.config):

args.config = os.path.abspath(args.config)

if args.command in {'run', 'try-repo'}:

args.files = [os.path.abspath(filename) for filename in args.files]

if args.command == 'try-repo' and os.path.exists(args.repo):

args.repo = os.path.abspath(args.repo)

try:

os.chdir(git.get_root())

except CalledProcessError:

raise FatalError(

'git failed. Is it installed, and are you in a Git repository '

'directory?',

)

args.config = os.path.relpath(args.config)

if args.command in {'run', 'try-repo'}:

args.files = [os.path.relpath(filename) for filename in args.files]

if args.command == 'try-repo' and os.path.exists(args.repo):

args.repo = os.path.relpath(args.repo)

def main(argv=None):

argv = argv if argv is not None else sys.argv[1:]

argv = [five.to_text(arg) for arg in argv]

parser = argparse.ArgumentParser()

# https://stackoverflow.com/a/8521644/812183

parser.add_argument(

'-V', '--version',

action='version',

version='%(prog)s {}'.format(C.VERSION),

)

subparsers = parser.add_subparsers(dest='command')

autoupdate_parser = subparsers.add_parser(

'autoupdate',

help="Auto-update pre-commit config to the latest repos' versions.",

)

_add_color_option(autoupdate_parser)

_add_config_option(autoupdate_parser)

autoupdate_parser.add_argument(

'--tags-only', action='store_true', help='LEGACY: for compatibility',

)

autoupdate_parser.add_argument(

'--bleeding-edge', action='store_true',

help=(

'Update to the bleeding edge of `master` instead of the latest '

'tagged version (the default behavior).'

)

autoupdate_parser.add_argument(

'--repo', dest='repos', action='append', metavar='REPO',

help='Only update this repository -- may be specified multiple times.',

)

clean_parser = subparsers.add_parser(

'clean', help='Clean out pre-commit files.',

)

_add_color_option(clean_parser)

_add_config_option(clean_parser)

gc_parser = subparsers.add_parser('gc', help='Clean unused cached repos.')

_add_color_option(gc_parser)

_add_config_option(gc_parser)

init_templatedir_parser = subparsers.add_parser(

'init-templatedir',

help=(

'Install hook script in a directory intended for use with '

'`git config init.templateDir`.'

)

_add_color_option(init_templatedir_parser)

_add_config_option(init_templatedir_parser)

init_templatedir_parser.add_argument(

'directory', help='The directory in which to write the hook script.',

)

_add_hook_type_option(init_templatedir_parser)

install_parser = subparsers.add_parser(

'install', help='Install the pre-commit script.',

)

_add_color_option(install_parser)

_add_config_option(install_parser)

install_parser.add_argument(

'-f', '--overwrite', action='store_true',

help='Overwrite existing hooks / remove migration mode.',

)

install_parser.add_argument(

'--install-hooks', action='store_true',

help=(

'Whether to install hook environments for all environments '

'in the config file.'

)

_add_hook_type_option(install_parser)

install_parser.add_argument(

'--allow-missing-config', action='store_true', default=False,

help=(

'Whether to allow a missing `pre-commit` configuration file '

'or exit with a failure code.'

)

install_hooks_parser = subparsers.add_parser(

'install-hooks',

help=(

'Install hook environments for all environments in the config '

'file. You may find `pre-commit install --install-hooks` more '

'useful.'

)

_add_color_option(install_hooks_parser)

_add_config_option(install_hooks_parser)

migrate_config_parser = subparsers.add_parser(

'migrate-config',

help='Migrate list configuration to new map configuration.',

)

_add_color_option(migrate_config_parser)

_add_config_option(migrate_config_parser)

run_parser = subparsers.add_parser('run', help='Run hooks.')

_add_color_option(run_parser)

_add_config_option(run_parser)

_add_run_options(run_parser)

sample_config_parser = subparsers.add_parser(

'sample-config', help='Produce a sample {} file'.format(C.CONFIG_FILE),

)

_add_color_option(sample_config_parser)

_add_config_option(sample_config_parser)

try_repo_parser = subparsers.add_parser(

'try-repo',

help='Try the hooks in a repository, useful for developing new hooks.',

)

_add_color_option(try_repo_parser)

_add_config_option(try_repo_parser)

try_repo_parser.add_argument(

'repo', help='Repository to source hooks from.',

)

try_repo_parser.add_argument(

'--ref', '--rev',

help=(

'Manually select a rev to run against, otherwise the `HEAD` '

'revision will be used.'

)

_add_run_options(try_repo_parser)

uninstall_parser = subparsers.add_parser(

'uninstall', help='Uninstall the pre-commit script.',

)

_add_color_option(uninstall_parser)

_add_config_option(uninstall_parser)

_add_hook_type_option(uninstall_parser)

help = subparsers.add_parser(

'help', help='Show help for a specific command.',

)

help.add_argument('help_cmd', nargs='?', help='Command to show help for.')

# argparse doesn't really provide a way to use a `default` subparser

if len(argv) == 0:

argv = ['run']

args = parser.parse_args(argv)

if args.command == 'help' and args.help_cmd:

parser.parse_args([args.help_cmd, '--help'])

elif args.command == 'help':

parser.parse_args(['--help'])

with error_handler(), logging_handler(args.color):

if args.command not in COMMANDS_NO_GIT:

_adjust_args_and_chdir(args)

git.check_for_cygwin_mismatch()

store = Store()

store.mark_config_used(args.config)

if args.command == 'autoupdate':

if args.tags_only:

logger.warning('--tags-only is the default')

return autoupdate(

args.config, store,

tags_only=not args.bleeding_edge,

repos=args.repos,

)

elif args.command == 'clean':

return clean(store)

elif args.command == 'gc':

return gc(store)

elif args.command == 'install':

return install(

args.config, store,

hook_types=args.hook_types,

overwrite=args.overwrite, hooks=args.install_hooks,

skip_on_missing_config=args.allow_missing_config,

)

elif args.command == 'init-templatedir':

return init_templatedir(

args.config, store, args.directory,

hook_types=args.hook_types,

)

elif args.command == 'install-hooks':

return install_hooks(args.config, store)

elif args.command == 'migrate-config':

return migrate_config(args.config)

elif args.command == 'run':

return run(args.config, store, args)

elif args.command == 'sample-config':

return sample_config()

elif args.command == 'try-repo':

return try_repo(args)

elif args.command == 'uninstall':

return uninstall(hook_types=args.hook_types)

else:

raise NotImplementedError(

'Command {} not implemented.'.format(args.command),

)

raise AssertionError(

'Command {} failed to exit with a returncode'.format(args.command),

)

if __name__ == '__main__':

exit(main())

pre-commit/pre_commit/main.py at wip_progress_dots · pre-commit/pre-commit · GitHub

Repository navigation

Expand file tree

Breadcrumbs

main.py

Latest commit

History

Breadcrumbs

main.py

File metadata and controls