Assignment3_PTable.ipynb

{"cells":[{"metadata":{},"cell_type":"markdown","source":"# Functions"},{"metadata":{"trusted":true,"collapsed":true},"cell_type":"code","source":"def prompt_mode():\n    print(\"1. Get info on element\")\n    print(\"2. Get property info for all elements\")\n    print(\"3. Insert new element\")\n    print(\"4. Edit element\")\n    print()\n    print(\"-9. Exit program\")\n    print()\n\n    mode = int(input(\"Enter mode number: \"))\n    print()\n    \n    return mode","execution_count":2,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"def mode_1(ptable):\n    print(\"1. Get info on element\")\n    print()\n\n    print(\"Elements:\")\n    for symbol in ptable:\n        print(symbol)\n    print()\n\n    key = input(\"Enter element symbol: \").capitalize()\n    print()\n\n    info = ptable.get(key)\n\n    if (info != None):\n        print(\"Symbol       : {}\".format(key))\n        print(\"Name         : {}\".format(info.get(\"name\")))\n        print(\"Atomic Number: {}\".format(info.get(\"atomic number\")))\n        print(\"Row          : {}\".format(info.get(\"row\")))\n        print(\"Column       : {}\".format(info.get(\"col\")))\n    else:\n        print(\"Element not found!\\nConsider adding it yourself!\")","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"def mode_2(ptable):\n    print(\"2. Get property info for all elements\")\n    print()\n\n    print(\"Properties:\")\n    print(\"1. Symbol\")\n    print(\"2. Name\")\n    print(\"3. Atomic number\")\n    print(\"4. Row\")\n    print(\"5. Column\")\n    print()\n\n    property = int(input(\"Enter property number: \"))\n    print()\n    \n    if (property == 1):\n        print(\"Symbols:\")\n        for symbol in ptable:\n            print(symbol)\n    elif (property == 2):\n        print(\"Names:\")\n        for key in ptable:\n            print(ptable.get(key).get(\"name\"))\n    elif (property == 3):\n        print(\"Atomic Numbers:\")\n        for key in ptable:\n            print(ptable.get(key).get(\"atomic number\"))\n    elif (property == 4):\n        print(\"Rows:\")\n        for key in ptable:\n            print(ptable.get(key).get(\"row\"))\n    elif (property == 5):\n        print(\"Columns:\")\n        for key in ptable:\n            print(ptable.get(key).get(\"col\"))\n    else:\n        print(\"Invalid entry\")","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"def mode_3(ptable):\n    print(\"3. Insert new element\")\n    print()\n\n    key = input(\"Enter symbol: \").capitalize()\n    name = input(\"Enter name: \").capitalize()\n    atomic_num = input(\"Enter atomic number: \")\n    row = input(\"Enter row: \")\n    col = input(\"Enter column: \")\n    print()\n\n    ptable[key] = {\"name\": name,\n                   \"atomic number\": atomic_num,\n                   \"row\": row,\n                   \"col\": col}","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"def mode_4(ptable):\n    print(\"4. Edit element\")\n    print()\n\n    print(\"Elements:\")\n    for symbol in ptable:\n        print(symbol)\n    print()\n    \n    key = input(\"Enter element symbol: \").capitalize()\n    print()\n\n    if (key in ptable):\n        name = input(\"Enter new name: \").capitalize()\n        atomic_num = input(\"Enter new atomic number: \")\n        row = input(\"Enter new row: \")\n        col = input(\"Enter new column: \")\n        print()\n\n        ptable[key] = {\"name\": name,\n                       \"atomic number\": atomic_num,\n                       \"row\": row,\n                       \"col\": col}\n    else:\n        print(\"Invalid key\")","execution_count":null,"outputs":[]},{"metadata":{},"cell_type":"markdown","source":"# Main"},{"metadata":{"trusted":true},"cell_type":"code","source":"ptable = {\"H\":\n          {\"name\": \"Hydrogen\",\n           \"atomic number\": \"1\",\n           \"row\": \"1\",\n           \"col\": \"1\"},\n\n          \"Cs\":\n          {\"name\": \"Caesium\",\n           \"atomic number\": \"55\",\n           \"row\": \"6\",\n           \"col\": \"1\"},\n\n          \"Ru\":\n          {\"name\": \"Ruthenium\",\n           \"atomic number\": \"44\",\n           \"row\": \"5\",\n           \"col\": \"8\"}}\n\nmode = 0\n\nwhile (mode != -9):\n    mode = prompt_mode()\n\n    if (mode == 1):\n        mode_1(ptable)\n\n    if (mode == 2):\n        mode_2(ptable)\n\n    if (mode == 3):\n        mode_3(ptable)\n\n    if (mode == 4):\n        mode_4(ptable)\n\n    if (mode == 5):\n        mode_5(ptable)\n    \n    print(\"----\")\n    print()    ","execution_count":null,"outputs":[]}],"metadata":{"kernelspec":{"name":"python3","display_name":"Python 3","language":"python"},"language_info":{"name":"python","version":"3.6.11","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"}},"nbformat":4,"nbformat_minor":4}