Merge pull request #13 from jarun/argparse

Switch to argparse

buku

@@ -20,7 +20,7 @@
 import sys
 import os
 import sqlite3
-from getopt import getopt, GetoptError
+import argparse
 import readline
 import webbrowser
 import html.parser as HTMLParser
@@ -31,6 +31,7 @@ import io
 import signal
 import json
 import shutil
+import textwrap
 
 # Import libraries needed for encryption
 try:
@@ -50,28 +51,15 @@ except ImportError:
 
 
 # Globals
-addurl = False          # Add a URL
-addindex = None         # DB index to insert URL into
-delete = False          # Delete bookmark(s)
-empty = False           # List all bookmarks with no title or tag
-openurl = None          # Open URL in browser
-showindex = None        # Index of bookmark to show
-showOpt = 0             # Modify show. 1: show only URL, 2: show URL and tag
-showTags = False        # List all unique tags
-search = False          # Search for keywords
-searchAll = False       # Match all keywords in search
-entry = None            # DB index to update or delete
 update = False          # Update a bookmark in DB
-debug = False           # Enable debug logs
 titleData = None        # Title fetched from a page
 titleManual = None      # Manually add a title offline
-replace = False         # Replace a tag
-encrypt = False         # Lock database file
-decrypt = False         # Unlock database file
-iterations = 8          # Number of hash iterations to generate key
 jsonOutput = False      # Output json formatted result
+showOpt = 0             # Modify show. 1: show only URL, 2: show URL and tag
+debug = False           # Enable debug logs
 pipeargs = []           # Holds arguments piped to the program
 _VERSION_ = 1.9         # Program version
+BLANK = 'blank'
 
 
 class BMHTMLParser(HTMLParser.HTMLParser):
@@ -358,11 +346,11 @@ def isBookmarkAdded(cur, url):
 
 
 
-def AddUpdateEntry(conn, cur, keywords, index):
+def AddUpdateEntry(conn, cur, keywords, updateindex, insertindex=0):
-    """Add a new bookmark or update an existing record at index
+    """Add a new bookmark or update an existing record at
-    or insert a new record at addindex (if empty)
+    updateindex or insert a new record at insertindex (if empty)
 
-    Params: connection, cursor, index to update
+    Params: connection, cursor, keywords, index to update, index to insert at
     """
 
     global titleManual
@@ -373,7 +361,7 @@ def AddUpdateEntry(conn, cur, keywords, index):
     """In case of an add or insert operation ensure
     that the URL does not exist in DB already
     """
-    if index is None:
+    if updateindex == 0:
         id = isBookmarkAdded(cur, url)
         if id != -1:
             print("URL already exists at index %d" % id)
@@ -398,11 +386,8 @@ def AddUpdateEntry(conn, cur, keywords, index):
     if tags[-1] != ',':
         tags += ','
 
-    if titleManual != None:
+    if titleManual is not None:
-        if titleManual == "none":
+        meta = titleManual
-            meta = ''
-        else:
-            meta = titleManual
     else:
         meta = fetchTitle(url)
         if meta == '':
@@ -410,28 +395,28 @@ def AddUpdateEntry(conn, cur, keywords, index):
         else:
             print("Title: [%s]" % meta)
 
-    if index == None: # Insert a new entry
+    if updateindex == 0: # Add or insert a new entry
         try:
-            if addindex == None: # addindex is index number to insert record at
+            if insertindex == 0: # insertindex is index number to insert record at
                 cur.execute('INSERT INTO bookmarks(URL, metadata, tags) VALUES (?, ?, ?)', (url, meta, tags,))
             else:
-                cur.execute('INSERT INTO bookmarks(id, URL, metadata, tags) VALUES (?, ?, ?, ?)', (int(addindex), url, meta, tags,))
+                cur.execute('INSERT INTO bookmarks(id, URL, metadata, tags) VALUES (?, ?, ?, ?)', (insertindex, url, meta, tags,))
             conn.commit()
             print("Added at index %d\n" % cur.lastrowid)
-            printdb(cur, str(cur.lastrowid))
+            printdb(cur, cur.lastrowid)
         except sqlite3.IntegrityError:
             for row in cur.execute("SELECT id from bookmarks where URL LIKE ?", (url,)):
                 print("URL already exists at index %s" % row[0])
                 return
 
-            print("Index %s exists" % addindex)
+            print("Index %d exists" % insertindex)
     else: # Update an existing entry
         try:
-            cur.execute("UPDATE bookmarks SET URL = ?, metadata = ?, tags = ? WHERE id = ?", (url, meta, tags, int(index),))
+            cur.execute("UPDATE bookmarks SET URL = ?, metadata = ?, tags = ? WHERE id = ?", (url, meta, tags, updateindex,))
             conn.commit()
             if cur.rowcount == 1:
-                print("Updated index %d\n" % int(index))
+                print("Updated index %d\n" % updateindex)
-                printdb(cur, int(index))
+                printdb(cur, updateindex)
             else:
                 print("No matching index")
         except sqlite3.IntegrityError:
@@ -470,10 +455,7 @@ def dbRefresh(conn, cur, index):
             conn.commit()
             print("Updated index %d\n" % row[0])
     else:
-        if titleManual == "none":
+        title = titleManual
-            title = ''
-        else:
-            title = titleManual
 
         for row in resultset:
             cur.execute("UPDATE bookmarks SET metadata = ? WHERE id = ?", (title, row[0],))
@@ -482,11 +464,11 @@ def dbRefresh(conn, cur, index):
 
 
 
-def searchdb(cur, keywords):
+def searchdb(cur, keywords, all_keywords=False):
     """Search the database for an entries with tags or URL
     or title info matching keywords and list those.
 
-    Params: cursor, keywords to search
+    Params: cursor, keywords to search, search any or all keywords
     """
 
     global jsonOutput
@@ -494,7 +476,7 @@ def searchdb(cur, keywords):
     placeholder = "'%' || ? || '%'"
     query = "SELECT id, url, metadata, tags FROM bookmarks WHERE"
 
-    if searchAll == True: # Match all keywords in URL or Title
+    if all_keywords == True: # Match all keywords in URL or Title
         for token in keywords:
             query += " (tags LIKE (%s) OR URL LIKE (%s) OR metadata LIKE (%s)) AND" % (placeholder, placeholder, placeholder)
             arguments.append(token)
@@ -584,7 +566,7 @@ def cleardb(conn, cur, index):
     Params: connection, cursor, index to delete
     """
 
-    if int(index) == 0: # Remove the table
+    if index == 0: # Remove the table
         resp = input("ALL bookmarks will be removed. Enter \x1b[1my\x1b[21m to confirm: ")
         if resp != 'y':
             print("No bookmarks deleted")
@@ -595,11 +577,11 @@ def cleardb(conn, cur, index):
         print("All bookmarks deleted")
     else: # Remove a single entry
         try:
-            cur.execute('DELETE FROM bookmarks WHERE id = ?', (int(index),))
+            cur.execute('DELETE FROM bookmarks WHERE id = ?', (index,))
             conn.commit()
             if cur.rowcount == 1:
-                print("Removed index %d" % int(index))
+                print("Removed index %d" % index)
-                compactDB(conn, cur, int(index))
+                compactDB(conn, cur, index)
             else:
                 print("No matching index")
         except IndexError:
@@ -609,7 +591,8 @@ def cleardb(conn, cur, index):
 
 def printdb(cur, index, empty=False):
     """Print bookmark details at index or all bookmarks if index is None
-    Print only bookmarks with empty title or tags if empty is True
+    Print only bookmarks with blank title or tag if empty is True
+    Note: URL is printed on top because title may be blank
 
     Params: cursor, index to print, flag to show only bookmarks with no title or tags
     """
@@ -618,7 +601,7 @@ def printdb(cur, index, empty=False):
     global jsonOutput
 
     resultset = None
-    if int(index) == 0: # Show all entries
+    if index == 0: # Show all entries
         if empty == False:
             cur.execute('SELECT * FROM bookmarks')
             resultset = cur.fetchall()
@@ -642,7 +625,7 @@ def printdb(cur, index, empty=False):
             print(formatJson(resultset))
     else: # Show record at index
         try:
-            resultset = cur.execute("SELECT * FROM bookmarks WHERE id = ?", (int(index),))
+            resultset = cur.execute("SELECT * FROM bookmarks WHERE id = ?", (index,))
         except IndexError:
             print("Index out of bound")
             return
@@ -713,7 +696,7 @@ def showUniqueTags(cur):
 
 
 
-def replaceTags(conn, cur, orig, new):
+def replaceTags(conn, cur, orig, new=None):
     """Replace orig tags with new tags in DB for all records.
     Remove orig tag is new tag is empty.
 
@@ -724,14 +707,32 @@ def replaceTags(conn, cur, orig, new):
     delete = False
 
     orig = ',' + orig + ','
-    new = new.strip(',')
+    if new is None:
-    if new == '':
+        newtags = ','
-        new = ','
         delete = True
     else:
-        new = ',' + new + ','
+        newtags = ','
+        for tag in new:
+            if tag[-1] == ',':
+                tag = tag.strip(',') + ','      # if delimiter is present, maintain it
+            else:
+                tag = tag.strip(',')            # a token in a multi-word tag
 
-    if orig == new:
+            if tag == ',':
+                continue
+
+            if newtags[-1] == ',':
+                newtags += tag
+            else:
+                newtags += ' ' + tag
+
+        if newtags[-1] != ',':
+            newtags += ','
+
+        if newtags == ',':
+            delete = True
+
+    if orig == newtags:
         print("Tags are same.")
         return
 
@@ -740,10 +741,11 @@ def replaceTags(conn, cur, orig, new):
 
     for row in results:
         if delete == False:
-            if row[1].find(new) >= 0:
+            # Check if tag newtags is already added
-                new = ','
+            if row[1].find(newtags) >= 0:
+                newtags = ','
 
-        newtags = row[1].replace(orig, new)
+        newtags = row[1].replace(orig, newtags)
         cur.execute("UPDATE bookmarks SET tags = ? WHERE id = ?", (newtags, row[0],))
         print("Updated index %d" % row[0])
         update = True
@@ -760,7 +762,7 @@ def fetchopen(index):
     """
 
     try:
-        for row in cur.execute("SELECT URL FROM bookmarks WHERE id = ?", (int(index),)):
+        for row in cur.execute("SELECT URL FROM bookmarks WHERE id = ?", (index,)):
             url = unquote(row[0])
             browser_open(url)
             return
@@ -828,7 +830,7 @@ def get_filehash(filepath):
 
 
 
-def encrypt_file():
+def encrypt_file(iterations):
     """Encrypt the bookmarks database file"""
 
     dbpath = os.path.join(getDataPath(), 'bookmarks.db')
@@ -889,7 +891,7 @@ def encrypt_file():
 
 
 
-def decrypt_file():
+def decrypt_file(iterations):
     """Decrypt the bookmarks database file"""
 
     dbpath = os.path.join(getDataPath(), 'bookmarks.db')
@@ -906,7 +908,7 @@ def decrypt_file():
     password = ''
     password = getpass.getpass()
     if password == '':
-        print("Decryption failed");
+        printmsg("Decryption failed", "ERROR");
         sys.exit(1)
 
     with open(encpath, 'rb') as infile:
@@ -938,13 +940,12 @@ def decrypt_file():
     dbhash = get_filehash(dbpath)
     if dbhash != enchash:
         os.remove(dbpath)
-        print("Decryption failed");
+        printmsg("Decryption failed", "ERROR");
+        sys.exit(1)
     else:
         os.remove(encpath)
         print("File decrypted")
 
-    sys.exit(0)
-
 
 
 def sigint_handler(signum, frame):
@@ -970,53 +971,64 @@ def printmsg(msg, level=None):
 
 
 
-def usage():
+class customUpdateAction(argparse.Action):
-    """Show buku usage, options, general information and exit"""
+    """Class to capture if an optional param
+    is actually used, even if sans arguments
+    """
 
-    print("Usage: OPTIONS [URL] [TAGS] [KEYWORDS ...]\n\n"
+    def __call__(self, parser, args, values, option_string=None):
+        global update
 
-    "A private cmdline bookmark manager. Your mini web!\n\n"
+        update = True
+        # NOTE: the following converts a None argument to an empty array []
+        setattr(args, self.dest, values)
 
-    "General options\n"
-    "  -a URL [tags]        add URL as bookmark with comma separated tags\n"
-    "  -d N                 delete entry at DB index N (from -p 0), N=0 deletes all\n"
-    "  -g                   list all tags alphabetically\n"
-    "  -m title             manually set title, for -a, -i, -u; '-m none' clears title\n"
-    "  -s keyword(s)        search bookmarks for any keyword\n"
-    "  -S keyword(s)        search bookmarks with all keywords\n"
-    "  -u N [URL] [tags]    update fields of the entry at DB index N\n"
-    "                       The first keyword, if available, is treated as the URL.\n"
-    "                       If URL is omitted (and -m is not used) the title of entry at\n"
-    "                       index N is refreshed from the web, N=0 refreshes all titles.\n\n"
 
-    "Power toys\n"
-    "  -e                   show bookmarks with empty titles or no tags\n"
-    "  -i N                 insert new bookmark at free DB index N\n"
-    "  -j                   show results in Json format\n"
-    "  -k                   decrypt (unlock) database file\n"
-    "  -l                   encrypt (lock) database file\n"
-    "  -o N                 open URL at DB index N in browser\n"
-    "  -p N                 show details of bookmark record at DB index N, N=0 shows all\n"
-    "  -r oldtag [newtag]   replace oldtag with newtag, delete oldtag if newtag empty\n"
-    "  -t N                 use N (> 0) hash iterations to generate key, for -k, -l\n"
-    "  -x N                 modify -p behaviour, N=1: show only URL, N=2: show URL & tag\n"
-    "  -z                   show debug information\n\n"
 
-    "Prompt keys\n"
+class customTitleAction(argparse.Action):
-    "  1-N                  open Nth search result in browser\n"
+    """Class to capture if an optional param
-    "  Enter                exit buku\n\n"
+    is actually used, even if sans arguments
+    """
 
-    "Version %.1f\n"
+    def __call__(self, parser, args, values, option_string=None):
-    "Copyright (C) 2015 Arun Prakash Jana <engineerarun@gmail.com>\n"
+        global titleManual
-    "License: GPLv3\n"
+
-    "Webpage: https://github.com/jarun/buku\n" % _VERSION_)
+        titleManual = ''
-    sys.exit(1)
+        if titleManual is not None:
+            print("titleManual is not None")
+        # NOTE: the following converts a None argument to an empty array []
+        setattr(args, self.dest, values)
+
+
+
+class ExtendedArgumentParser(argparse.ArgumentParser):
+    """Extend classic argument parser"""
+
+    # Print additional help and info
+    @staticmethod
+    def print_extended_help(file=None):
+        file.write(textwrap.dedent('''
+        prompt keys:
+          1-N                  open the Nth search result in web browser
+          Enter                exit buku
+
+        Version %.1f
+        Copyright (C) 2015 Arun Prakash Jana <engineerarun@gmail.com>
+        License: GPLv3
+        Webpage: https://github.com/jarun/buku
+        ''' % _VERSION_))
+
+    # Help
+    def print_help(self, file=None):
+        super(ExtendedArgumentParser, self).print_help(file)
+        self.print_extended_help(file)
 
 
 
 """main starts here"""
+
+# Handle piped input
 def main(argv = sys.argv):
-    # detects whether have pipe line parsing in
     if not sys.stdin.isatty():
         pipeargs.extend(sys.argv)
         for s in sys.stdin.readlines():
@@ -1028,129 +1040,108 @@ if __name__ == "__main__":
     except KeyboardInterrupt:
         pass
 
-optlist = None
+# If piped input, set argument vector
-keywords = None
+if len(pipeargs) > 0:
+    sys.argv = pipeargs
 
+# Setup custom argument parser
+argparser = ExtendedArgumentParser(
+    description='A private cmdline bookmark manager. Your mini web!',
+    formatter_class=argparse.RawTextHelpFormatter,
+    usage='''buku [-a URL [tags ...]] [-u [N [URL tags ...]]]
+            [-t [...]] [-d [N]] [-h]
+            [-s keyword [...]] [-S keyword [...]]
+            [-k [N]] [-l [N]] [-p [N]] [-f N]
+            [-r oldtag [newtag ...]] [-j] [-o N] [-z]''',
+    add_help=False
+)
+
+# General options
+general_group = argparser.add_argument_group(title="general options",
+    description='''-a, --add URL [tags ...]
+                     bookmark URL with comma separated tags
+-u, --update [N [URL tags ...]]
+                     update fields of bookmark at DB index N
+                     refresh all titles, if no arguments
+                     if URL omitted and -t is unused, update
+                     title of bookmark at index N from web
+-t, --title [...]    manually set title, works with -a, -u
+                     do not set title, if no arguments
+-d, --delete [N]     delete bookmark at DB index N
+                     delete all bookmarks, if no arguments
+-h, --help           show this information''')
+general_group.add_argument('-a', '--add', nargs='+', dest='addurl', metavar=('URL', 'tags'), help=argparse.SUPPRESS)
+general_group.add_argument('-u', '--update', nargs='*', dest='update', action=customUpdateAction, metavar=('N', 'URL tags'), help=argparse.SUPPRESS)
+general_group.add_argument('-t', '--title', nargs='*', dest='title', action=customTitleAction, metavar='title', help=argparse.SUPPRESS)
+general_group.add_argument('-d', '--delete', nargs='?', dest='delete', type=int, const=0, metavar='N', help=argparse.SUPPRESS)
+general_group.add_argument('-h', '--help', dest='help', action='store_true', help=argparse.SUPPRESS)
+
+# Search options
+search_group=argparser.add_argument_group(title="search options",
+    description='''-s, --sany keyword [...]
+                     search bookmarks for ANY matching keyword
+-S, --sall keyword [...]
+                     search bookmarks with ALL keywords
+                     special keywords -
+                     "tags" : list all tags alphabetically
+                     "blank": list entries with empty title/tag''')
+search_group.add_argument('-s', '--sany', nargs='+', metavar='keyword', help=argparse.SUPPRESS)
+search_group.add_argument('-S', '--sall', nargs='+', metavar='keyword', help=argparse.SUPPRESS)
+
+# Encryption options
+crypto_group=argparser.add_argument_group(title="encryption options",
+    description='''-l, --lock [N]       encrypt DB file with N (> 0, default 8)
+                     hash iterations to generate key
+-k, --unlock [N]     decrypt DB file with N (> 0, default 8)
+                     hash iterations to generate key''')
+crypto_group.add_argument('-k', '--unlock', nargs='?', dest='decrypt', type=int, const=8, metavar='N', help=argparse.SUPPRESS)
+crypto_group.add_argument('-l', '--lock', nargs='?', dest='encrypt', type=int, const=8, metavar='N', help=argparse.SUPPRESS)
+
+# Power toys
+power_group=argparser.add_argument_group(title="power toys",
+    description='''-p, --print [N]      show details of bookmark at DB index N
+                     show all bookmarks, if no arguments
+-f, --format N       modify -p output
+                     N=1: show only URL, N=2: show URL and tag
+-r, --replace oldtag [newtag ...]
+                     replace oldtag with newtag in all bookmarks
+                     delete oldtag, if no newtag
+-j, --jason          Json formatted output, works with -p, -s
+-o, --open           open bookmark at DB index N in web browser
+-z, --debug          show debug information and additional logs''')
+power_group.add_argument('-p', '--print', nargs='?', dest='printindex', type=int, const=0, metavar='N', help=argparse.SUPPRESS)
+power_group.add_argument('-f', '--format', dest='showOpt', type=int, choices=[1, 2], metavar='N', help=argparse.SUPPRESS)
+power_group.add_argument('-r', '--replace', nargs='+', dest='replace', metavar=('oldtag', 'newtag'), help=argparse.SUPPRESS)
+power_group.add_argument('-j', '--json', dest='jsonOutput', action='store_true', help=argparse.SUPPRESS)
+power_group.add_argument('-o', '--open', dest='openurl', type=int, metavar='N', help=argparse.SUPPRESS)
+power_group.add_argument('-z', '--debug', dest='debug', action='store_true', help=argparse.SUPPRESS)
+
+#addarg = argparser.add_argument
+#addarg('-i', '--insert', nargs='+', dest='insert', metavar=('N', 'URL tags'),
+#       help=" insert new bookmark with URL and tags at free DB index N; frees index if URL and tags are omitted")
+
+# Show help if no arguments passed
 if len(sys.argv) < 2:
-    usage()
+    argparser.print_help(sys.stderr)
-
-# Check cmdline options
-try:
-
-    if len(pipeargs) > 0:
-        optlist, keywords = getopt(pipeargs[1:], "d:i:m:o:p:t:u:x:aegjklrsSz")
-    else:
-        optlist, keywords = getopt(sys.argv[1:], "d:i:m:o:p:t:u:x:aegjklrsSz")
-    if len(optlist) < 1:
-        usage()
-
-    for opt in optlist:
-        if opt[0] == "-a":
-            if update == True or delete == True:
-                print("You can either add or update or delete in one instance\n")
-                usage()
-
-            addurl = True
-        elif opt[0] == "-d":
-            if addurl == True or update == True:
-                print("You can either add or update or delete in one instance\n")
-                usage()
-
-            if not opt[1].isdigit():
-                usage()
-
-            entry = opt[1]
-            if int(entry) < 0:
-                usage()
-
-            delete = True
-        elif opt[0] == "-e":
-            empty = True
-        elif opt[0] == "-g":
-            showTags = True
-        elif opt[0] == "-i":
-            if update == True or delete == True:
-                print("You can either add or update or delete in one instance\n")
-                usage()
-
-            if not opt[1].isdigit():
-                usage()
-
-            addindex = opt[1]
-            if int(addindex) <= 0:
-                usage()
-
-            addurl = True
-        elif opt[0] == "-j":
-            jsonOutput = True
-        elif opt[0] == "-k":
-            if no_crypto == True:
-                printmsg("PyCrypto missing", "ERROR")
-                sys.exit(0)
-
-            decrypt = True
-        elif opt[0] == "-l":
-            if no_crypto == True:
-                printmsg("PyCrypto missing", "ERROR")
-                sys.exit(0)
-
-            encrypt = True
-        elif opt[0] == "-m":
-            titleManual = opt[1]
-        elif opt[0] == "-o":
-            if not opt[1].isdigit():
-                usage()
-
-            openurl = opt[1]
-            if int(openurl) <= 0:
-                usage()
-        elif opt[0] == "-p":
-            if not opt[1].isdigit():
-                usage()
-
-            showindex = opt[1]
-            if int(showindex) < 0:
-                usage()
-        elif opt[0] == "-r":
-            replace = True
-        elif opt[0] == "-s":
-            search = True
-        elif opt[0] == "-S":
-            searchAll = True
-            search = True
-        elif opt[0] == "-t":
-            if not opt[1].isdigit():
-                usage()
-
-            iterations = int(opt[1])
-            if iterations <= 0:
-                usage()
-        elif opt[0] == "-u":
-            if addurl == True or delete == True:
-                print("You can either add or update or delete in one instance\n")
-                usage()
-
-            if not opt[1].isdigit():
-                usage()
-
-            entry = opt[1]
-            if int(entry) < 0:
-                usage()
-
-            update = True
-        elif opt[0] == "-x":
-            if not opt[1].isdigit():
-                usage()
-
-            showOpt = int(opt[1])
-            if showOpt < 1 or showOpt > 2:
-                usage()
-        elif opt[0] == "-z":
-            debug = True
-except GetoptError as e:
-    print("buku:", e)
     sys.exit(1)
 
+# Parse the arguments
+args = argparser.parse_args()
+
+# Show help and exit if help requested
+if args.help == True:
+    argparser.print_help(sys.stderr)
+    sys.exit(0)
+
+# Assign the values to globals
+if args.showOpt is not None:
+    showOpt = args.showOpt
+if titleManual is not None and len(args.title) > 0:
+    titleManual = " ".join(args.title)
+jsonOutput = args.jsonOutput
+debug = args.debug
+
+# Show version in debug logs
 if debug:
     print("Version %.1f" % _VERSION_)
 
@@ -1158,73 +1149,111 @@ if debug:
 moveOldDatabase()
 
 # Handle encrypt/decrypt options at top priority
-if encrypt == True:
+if args.encrypt is not None:
-    encrypt_file()
+    if no_crypto:
+        printmsg("PyCrypto missing", "ERROR")
+        sys.exit(1)
+    if args.encrypt < 1:
+        printmsg("Iterations must be >= 1", "ERROR")
+        sys.exit(1)
+    encrypt_file(args.encrypt)
 
-if decrypt == True:
+if args.decrypt is not None:
-    decrypt_file()
+    if no_crypto:
+        printmsg("PyCrypto missing", "ERROR")
+        sys.exit(1)
+    if args.decrypt < 1:
+        printmsg("Decryption failed", "ERROR");
+        sys.exit(1)
+    decrypt_file(args.decrypt)
 
-# Initilize the database and get handles
+# Initialize the database and get handles
 conn, cur = initdb()
 
-# Replace a tag in DB
+# Add a record
-if replace == True:
+if args.addurl is not None:
-    numargs = len(keywords)
+    AddUpdateEntry(conn, cur, args.addurl, 0)
 
-    if addurl == True or update == True or delete == True:
+# Remove a single record or all records
-        print("Tag replace doesn't work with add or update or delete.\n")
+if args.delete is not None:
+    if args.delete < 0:
+        printmsg("Index must be >= 0", "ERROR")
         conn.close()
-        usage()
+        sys.exit(1)
-    elif numargs < 1 or numargs > 2:
+    cleardb(conn, cur, args.delete)
-        print("Tag replace accepts 1 or 2 arguments\n")
+
-        conn.close()
+# Search URLs, titles, tags for any keyword
-        usage()
+if args.sany is not None:
-    elif numargs == 1:
+    searchdb(cur, args.sany)
-        replaceTags(conn, cur, keywords[0], "")
+
+# Search URLs, titles, tags with all keywords
+if args.sall is not None:
+    if args.sall[0] == 'tags' and len(args.sall) == 1:
+        showUniqueTags(cur)
+    elif args.sall[0] == BLANK and len(args.sall) == 1:
+        printdb(cur, 0, True)
     else:
-        replaceTags(conn, cur, keywords[0], keywords[1])
+        searchdb(cur, args.sall, True)
-
-# Add record
-if addurl == True:
-    if len(keywords) < 1:
-        conn.close()
-        usage()
-
-    AddUpdateEntry(conn, cur, keywords, entry)
 
 # Update record
 if update == True:
-    if len(keywords) < 1:
+    if len(args.update) == 0:
-        dbRefresh(conn, cur, int(entry))
+        dbRefresh(conn, cur, 0)
-    else:
+    elif not args.update[0].isdigit():
-        AddUpdateEntry(conn, cur, keywords, entry)
+        printmsg("Index must be a number >= 0", "ERROR")
-
-# Search tags, URLs, Title info
-if search == True:
-    if len(keywords) < 1:
         conn.close()
-        usage()
+        sys.exit(1)
-
+    elif int(args.update[0]) == 0:
-    searchdb(cur, keywords)
+        dbRefresh(conn, cur, 0)
+    elif len(args.update) == 1:
+        printmsg("At least URL should be provided for non-zero index", "ERROR")
+        conn.close()
+        sys.exit(1)
+    else:
+        AddUpdateEntry(conn, cur, args.update[1:], int(args.update[0]))
 
 # Print all records
-if showindex is not None:
+if args.printindex is not None:
-    printdb(cur, showindex)
+    if args.printindex < 0:
+        printmsg("Index must be >= 0", "ERROR")
+        conn.close()
+        sys.exit(1)
+    printdb(cur, args.printindex)
 
-# Show all unique tags
+# Replace a tag in DB
-if showTags == True:
+if args.replace is not None:
-    showUniqueTags(cur)
+    if len(args.replace) == 1:
-
+        replaceTags(conn, cur, args.replace[0])
-if empty == True:
+    else:
-    printdb(cur, 0, empty)
+        replaceTags(conn, cur, args.replace[0], args.replace[1:])
 
 # Open URL in browser
-if openurl != None:
+if args.openurl is not None:
-    fetchopen(openurl)
+    if args.openurl < 1:
+        printmsg("Index must be >= 1", "ERROR")
+        conn.close()
+        sys.exit(1)
+    fetchopen(args.openurl)
 
-# Remove a single record of all records
+"""NOTE: Insert is functional but commented
-if delete == True:
+because DB compaction serves the purpose.
-    cleardb(conn, cur, entry)
+
+# Insert a record at an index
+if args.insert is not None:
+    if not args.insert[0].isdigit():
+        printmsg("Index must be a number >= 1", "ERROR")
+        conn.close()
+        sys.exit(1)
+    insertindex = int(args.insert[0])
+    if insertindex < 1:
+        printmsg("Index must be a number >= 1", "ERROR")
+        conn.close()
+        sys.exit(1)
+    if len(args.insert) == 1:
+        pass # No operation
+    else:
+        AddUpdateEntry(conn, cur, args.insert[1:], 0, insertindex)
+"""
 
 # Close the connection before exiting
 conn.close()