archive/dotfile-automator
Archived
1
Fork 0
Code Releases Packages Activity

PEP8

Browse source
This commit is contained in:
Jake Howard 2015-12-02 22:39:40 +00:00
parent ba732934d0
commit a9ec77cce0
2 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
project/config/cli.py
View file

@ -1,8 +1,8 @@
import click import click
from project.utils import get_config from project.utils import get_config
@click.command('config') @click.command('config')
@click.argument('key') @click.argument('key')
def cli(key): def cli(key):
print(get_config(key)) print(get_config(key))
pass

5
project/utils.py
View file

@ -6,9 +6,11 @@ try:
except ImportError: except ImportError:
from yaml import Loader from yaml import Loader
BASE_DIR = os.path.dirname(os.path.abspath(__file__)) BASE_DIR = os.path.dirname(os.path.abspath(__file__))
USER_CONFIG_DIR = "~/.dfa.yml" USER_CONFIG_DIR = "~/.dfa.yml"
def get_config_data(filename): def get_config_data(filename):
try: try:
config_file = open(filename) config_file = open(filename)
@ -17,10 +19,11 @@ def get_config_data(filename):
return [] return []
DEFAULT_CONFIG = get_config_data(os.path.join(BASE_DIR, 'config' ,'defaults.yml')) DEFAULT_CONFIG = get_config_data(os.path.join(BASE_DIR, 'config', 'defaults.yml'))
USER_CONFIG = get_config_data(os.path.expanduser(USER_CONFIG_DIR)) USER_CONFIG = get_config_data(os.path.expanduser(USER_CONFIG_DIR))
def get_config(key): def get_config(key):
if USER_CONFIG and key in USER_CONFIG: if USER_CONFIG and key in USER_CONFIG:
return USER_CONFIG[key] return USER_CONFIG[key]