added speedial command

kamcli/commands/cmd_speeddial.py

@@ -0,0 +1,123 @@
+import click
+from sqlalchemy import create_engine
+from kamcli.ioutils import ioutils_dbres_print
+from kamcli.cli import pass_context
+from kamcli.cli import parse_user_spec
+
+
+##
+#
+#
[email protected]('speeddial', help='Manage speed dial records')
+@pass_context
+def cli(ctx):
+    pass
+
+
+##
+#
+#
[email protected]('add', short_help='Add a speed dial record')
[email protected]('table', '--table', default='speed_dial',
+            help='Name of database table (default: speed_dial)')
[email protected]('userid', metavar='<userid>')
[email protected]('shortdial', metavar='<shortdial>')
[email protected]('targeturi', metavar='<targeturi>')
[email protected]('desc', metavar='<desc>', nargs=-1)
+@pass_context
+def speeddial_add(ctx, table, userid, shortdial, targeturi, desc):
+    """Add a speed dial record
+
+    \b
+    Parameters:
+        <userid> - username, AoR or SIP URI for subscriber
+        <shortdial> - username, AoR or SIP URI for short dial
+        <targeturi> - username, AoR or SIP URI for target
+        [<desc>] - description for speed dial record
+    """
+    udata = parse_user_spec(ctx, userid)
+    sdata = parse_user_spec(ctx, shortdial)
+    tdata = parse_user_spec(ctx, targeturi)
+    ctx.vlog('Adding for user [%s@%s] short dial [%s@%s] target [sip:%s@%s]',
+            udata['username'], udata['domain'],
+            sdata['username'], sdata['domain'],
+            tdata['username'], tdata['domain'])
+    e = create_engine(ctx.gconfig.get('db', 'rwurl'))
+    if not desc:
+        e.execute('insert into ' + table + ' (username, domain, sd_username, sd_domain, new_uri) values (%s, %s, %s, %s, %s)',
+                udata['username'], udata['domain'], sdata['username'], sdata['domain'],
+                'sip:' + tdata['username'] + '@' + tdata['domain'])
+    else:
+        e.execute('insert into ' + table + ' (username, domain, sd_username, sd_domain, new_uri, description) values (%s, %s, %s, %s, %s, %s)',
+                udata['username'], udata['domain'], sdata['username'], sdata['domain'],
+                'sip:' + tdata['username'] + '@' + tdata['domain'], desc)
+
+
+##
+#
+#
[email protected]('rm', short_help='Remove speed dial records')
[email protected]('table', '--table', default='speed_dial',
+            help='Name of database table (default: speed_dial)')
[email protected]('userid', metavar='<userid>')
[email protected]('shortdial', metavar='<shortdial>', nargs=-1)
+@pass_context
+def speeddial_rm(ctx, table, userid, shortdial):
+    """Remove a user from groups (revoke privilege)
+
+    \b
+    Parameters:
+        <userid> - username, AoR or SIP URI for subscriber
+        <shortdial> - username, AoR or SIP URI for short dial
+    """
+    udata = parse_user_spec(ctx, userid)
+    ctx.log('Removing speed dial for record [%s@%s]', udata['username'], udata['domain'])
+    e = create_engine(ctx.gconfig.get('db', 'rwurl'))
+    if not shortdial:
+        e.execute('delete from ' + table + ' where username=%s and domain=%s',
+                    udata['username'], udata['domain'])
+    else:
+        for s in shortdial:
+            sdata = parse_user_spec(ctx, s)
+            e.execute('delete from ' + table + ' where username=%s and domain=%s and sd_username=%s and sd_domain=%s',
+                    udata['username'], udata['domain'], sdata['username'], sdata['domain'])
+
+
+##
+#
+#
[email protected]('show', short_help='Show speed dial records')
[email protected]('oformat', '--output-format', '-F',
+                type=click.Choice(['raw', 'json', 'table', 'dict']),
+                default=None, help='Format the output')
[email protected]('ostyle', '--output-style', '-S',
+                default=None, help='Style of the output (tabulate table format)')
[email protected]('table', '--table', default='speed_dial',
+            help='Name of database table (default: speed_dial)')
[email protected]('userid', metavar='[<userid>]')
[email protected]('shortdial', nargs=-1, metavar='[<shortdial>]')
+@pass_context
+def speeddial_show(ctx, oformat, ostyle, table, userid, shortdial):
+    """Show details for speed dial records
+
+    \b
+    Parameters:
+        <userid> - username, AoR or SIP URI for user or alias
+        [<shortdial>] - username, AoR or SIP URI for short dial (optional)
+    """
+    udata = parse_user_spec(ctx, userid)
+    e = create_engine(ctx.gconfig.get('db', 'rwurl'))
+
+    ctx.vlog('Showing speed dial records for user [%s@%s]', udata['username'], udata['domain'])
+    if not shortdial:
+        res = e.execute('select * from ' + table + ' where username=%s and domain=%s',
+                udata['username'], udata['domain'])
+        ioutils_dbres_print(ctx, oformat, ostyle, res)
+    else:
+        for s in shortdial:
+            sdata = parse_user_spec(ctx, s)
+            res = e.execute('select * from ' + table + ' where username=%s and domain=%s and sd_username=%s and sd_domain=%s',
+                    udata['username'], udata['domain'], sdata['username'], sdata['domain'])
+            ioutils_dbres_print(ctx, oformat, ostyle, res)
+
+