Handle base dir, queue dir in Config class

This commit is contained in:
Marcin Kulik
2012-11-03 21:46:35 +01:00
committed by Marcin Kulik
parent 2c4217fc10
commit 477d36ea57
2 changed files with 15 additions and 8 deletions

View File

@@ -5,24 +5,32 @@ import uuid
class Config: class Config:
def __init__(self): def __init__(self):
self.config_filename = os.path.expanduser('~/.ascii.io/config') self.base_dir_path = os.path.expanduser("~/.ascii.io")
self.config_file_path = '%s/config' % self.base_dir_path
self.queue_dir_path = '%s/queue' % self.base_dir_path
self.create_base_dir()
self.parse_config_file()
def create_base_dir(self):
if not os.path.isdir(self.base_dir_path):
os.mkdir(self.base_dir_path)
def parse_config_file(self):
config = ConfigParser.RawConfigParser() config = ConfigParser.RawConfigParser()
config.add_section('user') config.add_section('user')
config.add_section('api') config.add_section('api')
config.add_section('record') config.add_section('record')
try: try:
config.read(self.config_filename) config.read(self.config_file_path)
except ConfigParser.ParsingError: except ConfigParser.ParsingError:
print('Config file %s contains syntax errors' % self.config_filename) print('Config file %s contains syntax errors' %
self.config_file_path)
sys.exit(2) sys.exit(2)
self.config = config self.config = config
# if not os.path.isdir(BASE_DIR):
# os.mkdir(BASE_DIR)
def api_url(self): def api_url(self):
try: try:
api_url = self.config.get('api', 'url') api_url = self.config.get('api', 'url')
@@ -40,7 +48,7 @@ class Config:
user_token = str(uuid.uuid1()) user_token = str(uuid.uuid1())
self.config.set('user', 'token', user_token) self.config.set('user', 'token', user_token)
with open(self.config_filename, 'wb') as configfile: with open(self.config_file_path, 'wb') as configfile:
self.config.write(configfile) self.config.write(configfile)
return user_token return user_token

View File

@@ -2,4 +2,3 @@ import os
import sys import sys
SCRIPT_NAME = os.path.basename(sys.argv[0]) SCRIPT_NAME = os.path.basename(sys.argv[0])
BASE_DIR = os.path.expanduser("~/.ascii.io")