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
|
#
# -------------------------------------------------------------------------
# Copyright (c) 2019 AT&T Intellectual Property
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
# -------------------------------------------------------------------------
#
#!/usr/bin/env python3
import argparse
import json
import os
import sys
import lib.tables as tables
from lib.song import Song
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from valet.engine.db_connect.db_handler import DBHandler
from valet.engine.db_connect.locks import Locks, later
from valet.utils.logger import Logger
def options():
parser = argparse.ArgumentParser(description='\033[34mManual Locking And Unlocking Of Valet Regions.\033[0m', add_help=False)
Song.add_args(parser)
g1 = parser.add_argument_group('Control Music Locks')
ex = g1.add_mutually_exclusive_group()
ex.add_argument('-unlock', metavar="region", help='unlock this region')
ex.add_argument('-lock', metavar="region", help='lock this region')
g2 = parser.add_argument_group('Update Locks In Region Table')
ex = g2.add_mutually_exclusive_group()
ex.add_argument('-delete', metavar="region", help='delete region from table')
ex.add_argument('-add', metavar="region", help='update/add region to table')
g2.add_argument('-timeout', metavar="seconds", help='seconds till region expires (for -add)')
group = parser.add_argument_group("Change The Output")
group.add_argument("-?", "--help", action="help", help="show this help message and exit")
group.add_argument('-show', action='store_true', help='print out regions (locking) table')
return parser.parse_args()
# MAIN
if __name__ == "__main__":
opts = options()
logger = Logger().get_logger('console')
config = json.loads(open(opts.config).read())
music_config = config.get("music")
music_config["keyspace"] = config.get("db")["keyspace"]
music = Song(opts, config, logger)
dbh = DBHandler(music, config.get("db"), logger)
if opts.add:
timeout = opts.timeout if opts.timeout else config["engine"]["timeout"]
dbh.add_region(opts.add, later(seconds=int(timeout)))
logger.debug("added region to table")
if opts.lock:
if Locks(dbh, 0).got_lock(opts.lock) == "ok":
logger.debug("added region lock")
else:
logger.debug("failed to add region lock")
if opts.unlock:
Locks.unlock(dbh, opts.unlock)
logger.debug("deleted region lock '%s'" % opts.unlock)
if opts.delete:
dbh.delete_region(opts.delete)
logger.debug("deleted region from table")
if opts.show:
music.keyspace = dbh.keyspace
tables.Regions(music, logger).read(names=True)
sys.exit(0)
|
