25 lines
705 B
Python
25 lines
705 B
Python
|
import distutils, os
|
||
|
|
from setuptools import Command
|
||
|
|
from setuptools.command.setopt import edit_config, option_base
|
||
|
|
|
||
|
|
class saveopts(option_base):
|
||
|
|
"""Save command-line options to a file"""
|
||
|
|
|
||
|
|
description = "save supplied options to setup.cfg or other config file"
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
dist = self.distribution
|
||
|
|
settings = {}
|
||
|
|
|
||
|
|
for cmd in dist.command_options:
|
||
|
|
|
||
|
|
if cmd=='saveopts':
|
||
|
|
continue # don't save our own options!
|
||
|
|
|
||
|
|
for opt,(src,val) in dist.get_option_dict(cmd).items():
|
||
|
|
if src=="command line":
|
||
|
|
settings.setdefault(cmd,{})[opt] = val
|
||
|
|
|
||
|
|
edit_config(self.filename, settings, self.dry_run)
|
||
|
|
|