1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
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
|
# -*- coding: utf-8 -*-
#
#!/usr/bin/env python
#
# Copyright (C) 2016 Dr. Robert C. Green II.
#
# This file is part of Combox.
#
# Combox is free software: you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Combox is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Combox (see COPYING). If not, see
# <http://www.gnu.org/licenses/>.
import os
import time
import yaml
from argparse import ArgumentParser
from os import path
from os.path import expanduser
from sys import exit
from threading import Lock
from watchdog.observers import Observer
from combox.config import config_cb, get_nodedirs
from combox.events import ComboxDirMonitor, NodeDirMonitor
from combox.gui import ComboxConfigDialog
from combox.log import log_i, log_e
## Function adapted from Watchdog's docs:
## http://pythonhosted.org/watchdog/quickstart.html#quickstart
def run_cb(config):
"""Runs combox.
- Creates an instance of :class:`.ComboxDirMonitor` to monitor the
combox directory.
- Creates an instance of :class:`.NodeDirMonitor` for each node
directory.
Exits on Ctrt-C.
:param dict config:
A dictionary that contains configuration information about
combox.
"""
db_lock = Lock()
monitor_lock = Lock()
# start combox directory (cd) monitor (cdm)
combox_dir = path.abspath(config['combox_dir'])
cd_monitor = ComboxDirMonitor(config, db_lock, monitor_lock)
cd_observer = Observer()
cd_observer.schedule(cd_monitor, combox_dir, recursive=True)
cd_observer.start()
# start a node directory monitor for each of the node directories.
node_dirs = get_nodedirs(config)
num_nodes = len(get_nodedirs(config))
nd_monitors = []
nd_observers = []
for node in node_dirs:
nd_monitor = NodeDirMonitor(config, db_lock,
monitor_lock)
nd_observer = Observer()
nd_observer.schedule(nd_monitor, node, recursive=True)
nd_observer.start()
nd_monitors.append(nd_monitor)
nd_observers.append(nd_observer)
# Make the first node monitor do the housekeeping.
nd_monitors[0].housekeep()
log_i("Hit Ctrl-C to quit")
try:
while True:
time.sleep(1)
except KeyboardInterrupt:
cd_observer.stop()
for i in range(num_nodes):
nd_observers[i].stop()
nd_observers[i].join()
cd_observer.join()
log_i("combox exiting. Bye!")
def main():
"""Parses args; starts combox configuration if necessary; starts combox.
"""
parser = ArgumentParser()
parser.add_argument("-t", "--test",
help="Use the combox config file in testing area.",
action="store_true")
parser.add_argument("-nw", "--cli",
help="Use CLI interface only",
action="store_true")
args = parser.parse_args()
if args.test:
CONFIG_DIR = path.join('tests', 'test-config')
else:
CONFIG_DIR = path.join(expanduser("~"),'.combox')
config_file = path.join(CONFIG_DIR, 'config.yaml')
log_i(CONFIG_DIR)
if (not path.exists(CONFIG_DIR) or
not path.exists(config_file)):
# combox not configured.
if not args.cli:
ComboxConfigDialog("combox configuration", CONFIG_DIR)
else:
config_cb(CONFIG_DIR)
try:
config = yaml.load(file(config_file, 'r'))
except (IOError, yaml.YAMLError) as exc:
log_e("Unable to open configuration file.")
log_e("Looks like combox is not configured yet. Exiting.")
exit(1)
# run combox.
run_cb(config)
if __name__ == "__main__":
main()
|
