import sys
import olrcdb
import os

# Globals
COUNT = 0


class FileParser(object):
    '''Object used to parse through a directory for all it's files. Collects
    the paths of all the files and stores a record of these in a new table in
    the database.

    The Schema of the database is:

    NewTable(path, uploaded=false)
    '''

    def __init__(self, directory, table_name):
        self.directory = directory
        self.table_name = table_name


def prepare_upload(connect, directory, table_name):
    '''Given a database connection, directory and table_name,
    -Create the table in the database
    -populate the table with (path, uploaded=false)
    where each path is a file in the given directory.'''

    global COUNT

    for filename in os.listdir(directory):

        file_path = os.path.join(directory, filename)

        # Add file name to the list.
        if os.path.isfile(file_path):
            connect.insert_path(file_path, table_name)
            COUNT += 1
        else:
            prepare_upload(connect, file_path, table_name)


if __name__ == "__main__":

    # Check for proper parameters
    if len(sys.argv) != 3:
        sys.stderr.write(
            'Usage: python prepareupload.py path-to-drive table-name\n'
        )
        sys.exit(1)

    connect = olrcdb.DatabaseConnection()
    connect.create_table(sys.argv[2])
    prepare_upload(connect, sys.argv[1], sys.argv[2])

import sys
import olrcdb
import os

# Globals
COUNT = 0


class FileParser(object):
    '''Object used to parse through a directory for all it's files. Collects
    the paths of all the files and stores a record of these in a new table in
    the database.

    The Schema of the database is:

    NewTable(path, uploaded=false)
    '''

    def __init__(self, directory, table_name):
        self.directory = directory
        self.table_name = table_name


def prepare_upload(connect, directory, table_name):
    '''Given a database connection, directory and table_name,
    -Create the table in the database
    -populate the table with (path, uploaded=false)
    where each path is a file in the given directory.'''

    global COUNT

    for filename in os.listdir(directory):

        file_path = os.path.join(directory, filename)

        # Add file name to the list.
        if os.path.isfile(file_path):
            connect.insert_path(file_path, table_name)
            COUNT += 1

            sys.stdout.flush()
            sys.stdout.write("\r{0} parsed. \n".format(COUNT))
        else:
            prepare_upload(connect, file_path, table_name)


if __name__ == "__main__":

    # Check for proper parameters
    if len(sys.argv) != 3:
        sys.stderr.write(
            'Usage: python prepareupload.py path-to-drive table-name\n'
        )
        sys.exit(1)

    connect = olrcdb.DatabaseConnection()
    connect.create_table(sys.argv[2])
    prepare_upload(connect, sys.argv[1], sys.argv[2])

Print messages for prepare upload.
Print messages for prepare upload.

  import sys
  import olrcdb
  import os
  
  # Globals
  COUNT = 0
  
  
  class FileParser(object):
      '''Object used to parse through a directory for all it's files. Collects
      the paths of all the files and stores a record of these in a new table in
      the database.
  
      The Schema of the database is:
  
      NewTable(path, uploaded=false)
      '''
  
      def __init__(self, directory, table_name):
          self.directory = directory
          self.table_name = table_name
  
  
  def prepare_upload(connect, directory, table_name):
      '''Given a database connection, directory and table_name,
      -Create the table in the database
      -populate the table with (path, uploaded=false)
      where each path is a file in the given directory.'''
  
      global COUNT
  
      for filename in os.listdir(directory):
  
          file_path = os.path.join(directory, filename)
  
          # Add file name to the list.
          if os.path.isfile(file_path):
              connect.insert_path(file_path, table_name)
              COUNT += 1
+ 
+             sys.stdout.flush()
+             sys.stdout.write("\r{0} parsed. \n".format(COUNT))
          else:
              prepare_upload(connect, file_path, table_name)
  
  
  if __name__ == "__main__":
  
      # Check for proper parameters
      if len(sys.argv) != 3:
          sys.stderr.write(
              'Usage: python prepareupload.py path-to-drive table-name\n'
          )
          sys.exit(1)
  
      connect = olrcdb.DatabaseConnection()
      connect.create_table(sys.argv[2])
      prepare_upload(connect, sys.argv[1], sys.argv[2])
  
Print messages for prepare upload.
## Code Before:
import sys
import olrcdb
import os

# Globals
COUNT = 0


class FileParser(object):
    '''Object used to parse through a directory for all it's files. Collects
    the paths of all the files and stores a record of these in a new table in
    the database.

    The Schema of the database is:

    NewTable(path, uploaded=false)
    '''

    def __init__(self, directory, table_name):
        self.directory = directory
        self.table_name = table_name


def prepare_upload(connect, directory, table_name):
    '''Given a database connection, directory and table_name,
    -Create the table in the database
    -populate the table with (path, uploaded=false)
    where each path is a file in the given directory.'''

    global COUNT

    for filename in os.listdir(directory):

        file_path = os.path.join(directory, filename)

        # Add file name to the list.
        if os.path.isfile(file_path):
            connect.insert_path(file_path, table_name)
            COUNT += 1
        else:
            prepare_upload(connect, file_path, table_name)


if __name__ == "__main__":

    # Check for proper parameters
    if len(sys.argv) != 3:
        sys.stderr.write(
            'Usage: python prepareupload.py path-to-drive table-name\n'
        )
        sys.exit(1)

    connect = olrcdb.DatabaseConnection()
    connect.create_table(sys.argv[2])
    prepare_upload(connect, sys.argv[1], sys.argv[2])

## Instruction:
Print messages for prepare upload.
## Code After:
import sys
import olrcdb
import os

# Globals
COUNT = 0


class FileParser(object):
    '''Object used to parse through a directory for all it's files. Collects
    the paths of all the files and stores a record of these in a new table in
    the database.

    The Schema of the database is:

    NewTable(path, uploaded=false)
    '''

    def __init__(self, directory, table_name):
        self.directory = directory
        self.table_name = table_name


def prepare_upload(connect, directory, table_name):
    '''Given a database connection, directory and table_name,
    -Create the table in the database
    -populate the table with (path, uploaded=false)
    where each path is a file in the given directory.'''

    global COUNT

    for filename in os.listdir(directory):

        file_path = os.path.join(directory, filename)

        # Add file name to the list.
        if os.path.isfile(file_path):
            connect.insert_path(file_path, table_name)
            COUNT += 1

            sys.stdout.flush()
            sys.stdout.write("\r{0} parsed. \n".format(COUNT))
        else:
            prepare_upload(connect, file_path, table_name)


if __name__ == "__main__":

    # Check for proper parameters
    if len(sys.argv) != 3:
        sys.stderr.write(
            'Usage: python prepareupload.py path-to-drive table-name\n'
        )
        sys.exit(1)

    connect = olrcdb.DatabaseConnection()
    connect.create_table(sys.argv[2])
    prepare_upload(connect, sys.argv[1], sys.argv[2])
