Ch refactor

.coveragerc

@@ -0,0 +1,4 @@
+[report]
+omit=
+    *tests/*
+    */python?.?/*

.gitignore

@@ -2,10 +2,11 @@
 __pycache__/
 *.py[cod]
 *$py.class
-
+*.pyc
 # C extensions
 *.so
 
+
 # Distribution / packaging
 .Python
 env/
@@ -44,6 +45,7 @@ nosetests.xml
 coverage.xml
 *,cover
 .hypothesis/
+cover/
 
 # Translations
 *.mo
@@ -87,3 +89,105 @@ ENV/
 
 # Rope project settings
 .ropeproject
+
+
+# Created by https://www.gitignore.io/api/python
+
+### Python ###
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+*.db
+
+# Distribution / packaging
+.Python
+env/
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*,cover
+.hypothesis/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# pyenv
+.python-version
+
+# celery beat schedule file
+celerybeat-schedule
+
+# SageMath parsed files
+*.sage.py
+
+# dotenv
+.env
+
+# virtualenv
+.venv
+venv/
+ENV/
+
+# Spyder project settings
+.spyderproject
+
+# Rope project settings
+.ropeproject
+
+# End of https://www.gitignore.io/api/python

app.py

@@ -0,0 +1,193 @@
+
+"""
+Usage:
+    amity create_room (O|L) <room_name>...
+    amity delete_rooms(args['<room_name>'])
+    amity add_person <first_name> <last_name> (F | S) [--wants_accomodation=value]
+    amity delete_employee(int(args['<person_id>']))
+    amity print_allocations [-output=<filename>]
+    amity reallocate_person <person_id> <new_room>
+    amity print_room <room_name>
+    amity print_unallocated [-output=<filename>]
+    amity load_people <filename>
+    amity delete_rooms(args['<room_name>'])
+    amity save_state (--database=<sqlite_database>)
+    amity load_state
+    amity (-i | --interactive)
+    amity (-h | --help | --version)
+Options:
+    -i, --interactive  Interactive Mode
+    -h, --help  Show this screen and exit.
+    -a, --wants_accomodation=<opt>  Person wants accomodation [default: N]
+    -d, --database=<sqlite_database>  Save state to specified database [default: amity_db]
+"""
+import cmd, os, sys
+from docopt import docopt, DocoptExit
+from pyfiglet import figlet_format
+from termcolor import cprint
+from views import Amity
+
+
+amity = Amity()
+
+def docopt_cmd(func):
+    """
+    This decorator is used to simplify the try/except block and pass the result
+    of the docopt parsing to the called action.
+    """
+    def fn(self, arg):
+        try:
+            opt = docopt(fn.__doc__,arg)
+        except DocoptExit as e:
+            # The DocoptExit is thrown when the args do not match.
+            # We print a message to the user and the usage block.
+
+            print('Invalid command!')
+            print(e)
+            return
+
+        except SystemExit:
+            # The SystemExit exception prints the usage for --help
+            # We do not need to do the print here.
+            return
+        return func(self, opt)
+
+    fn.__name__ = func.__name__
+    fn.__doc__ = func.__doc__
+    fn.__dict__.update(func.__dict__)
+    return fn
+
+def intro():
+    cprint(figlet_format('Space Allocator', font='slant'),
+           'blue', attrs=['bold'])
+    print("Welcome to Amity! Here is a list of commands to get you started." +
+          " Type 'help' anytime to access documented commands")
+    print("\n\n")
+    cprint(__doc__, 'magenta')
+
+
+class AmitySystem(cmd.Cmd):
+    prompt = '(Amity) '
+    file = None
+
+    @docopt_cmd
+    def do_create_room(self, args):
+        """Usage: create_room (O|L) <room_name>..."""
+        room_type = None
+        if args["O"].upper() == 'O':
+            room_type = "O"
+        else:
+            room_type = "L"
+        amity.create_room(room_type, args["<room_name>"])
+
+
+    @docopt_cmd
+    def do_add_person(self, args):
+        """
+        Usage: add_person <first_name> <last_name> (F | S) [--wants_accomodation=value]
+        """
+        if args["F"].upper() == 'F':
+            category = "F"
+        else:
+            category = "S"
+        first_name = args["<first_name>"]
+        last_name = args["<last_name>"]
+        amity.add_person(first_name, last_name, category, args["--wants_accomodation"])
+
+    @docopt_cmd
+    def do_reallocate_person(self, args):
+        """Usage: reallocate_person <person_id> <new_room>"""
+        if args["<person_id>"].isalpha():
+            print("person id cannot be string")
+            return
+        else:
+            (amity.reallocate_person(int(args['<person_id>']),
+                                     args['<new_room>']))
+
+    @docopt_cmd
+    def do_save_state(self, args):
+        """Persists app data into the given db
+        Usage: save_state (--db_name=sqlite_db)
+        """
+        db = args['--db_name']
+        amity.save_state(db)
+
+    @docopt_cmd
+    def do_print_unallocated(self, args):
+        """Usage: print_unallocated [--file=text_file]"""
+        amity.print_unallocated(args['--file'])
+
+    @docopt_cmd
+    def do_print_allocations(self, args):
+        """
+        Prints all rooms and the people in them.
+        Usage: print_allocations [--o=filename]
+        """
+        filename = args["--o"] or ""
+        amity.print_allocations(filename)
+
+    @docopt_cmd
+    def do_delete_employee(self, args):
+        """Usage: delete_employee <person_id>"""
+        if args["<person_id>"].isalpha():
+            print("person id cannot be string")
+            return
+        else:
+            (amity.delete_employee(int(args['<person_id>'])))
+
+    @docopt_cmd
+    def do_delete_rooms(self, args):
+        """Usage: delete_room <room_name>"""
+        amity.delete_rooms(args['<room_name>'])
+
+
+    @docopt_cmd
+    def do_load_state(self, arg):
+        """
+        Loads data from the specified db into the app.
+        Usage: load_state <filename>
+        """
+        amity.load_state(arg["<filename>"])
+
+    @docopt_cmd
+    def do_load_people(self, args):
+        """Usage: load_state <text_file>"""
+        amity.load_people(args["<text_file>"])
+
+
+    @docopt_cmd
+    def do_print_room(self, args):
+        """Usage: print_room <room_name>"""
+
+        room_name = args["<room_name>"]
+        amity.print_room(room_name)
+    @docopt_cmd
+    def do_clear(self, arg):
+        """Clears screen"""
+
+        os.system('clear')
+
+    @docopt_cmd
+    def do_load_people(self, args):
+        """Usage: load_state <text_file>"""
+        amity.load_people(args["<text_file>"])
+
+
+
+    def do_quit(self, arg):
+        """Quits out of Interactive Mode."""
+
+        print('Thank you for using Amity. Adios!')
+        exit()
+
+
+opt = docopt(__doc__, sys.argv[1:])
+
+if opt['--interactive']:
+    os.system('clear')
+    intro()
+    AmitySystem().cmdloop()
+
+print(opt)
+
+

models/model.py

@@ -0,0 +1,53 @@
+import os
+import sys
+import sqlalchemy
+
+from sqlalchemy import Column, Integer, String, ForeignKey, Text, Boolean
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import relationship, sessionmaker
+
+from sqlalchemy import create_engine
+
+
+
+
+Base = declarative_base()
+
+
+class Person(Base):
+    """creates a people's table"""
+    __tablename__ = "People"
+
+    id = Column(Integer, primary_key=True, autoincrement=True, nullable=False)
+    first_name = Column(String, nullable=False)
+    last_name = Column(String, nullable=False)
+    category = Column(String, nullable=False)
+    wants_accomodation = Column(Boolean)
+
+    def __repr__(self):
+        return 'first name{} last name{}'.format(self.first_name,self.last_name)
+
+
+class Room(Base):
+    """creates rooms tables"""
+    __tablename__ = "Rooms"
+
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    name = Column(String, nullable=False)
+    room_type = Column(String, nullable=False)
+    room_capacity = Column(Integer, nullable=False)
+    occupants = Column(String,nullable=True)
+
+
+class DatabaseCreator(object):
+    """creating database connection to object"""
+    def __init__(self, db_name=None):
+        self.db_name = db_name + '.db'
+        self.engine = create_engine('sqlite:///' + self.db_name)
+        Base.metadata.create_all(self.engine)
+        session_maker = sessionmaker(bind=self.engine)
+        self.session = session_maker()
+
+
+
+