Merge pull request #41 from bmrussell/file_operations

File operations

examples/python_client_library/file_operations.py

@@ -0,0 +1,123 @@
+"""Demonstrates creating writing, and deleting files/ folders on a NetApp volume.
+"""
+import logging
+from netapp_ontap import NetAppRestError
+from netapp_ontap.resources import Volume
+from netapp_ontap.resources import FileInfo
+from utils import Argument, parse_args, setup_connection
+
+# REFERENCES:
+# https://devnet.netapp.com/restapi.php
+# https://pypi.org/project/netapp-ontap/
+# https://library.netapp.com/ecmdocs/ECMLP2858435/html/resources/volume.html
+# https://library.netapp.com/ecmdocs/ECMLP2885777/html/resources/file_info.html
+# https://community.netapp.com/t5/ONTAP-Rest-API-Discussions/FileInfo-Received-list-directory-more-than-one-record/m-p/440962
+
+
+def list_files(volume, path):
+    """Recursively list files on a volume"""
+    files = FileInfo.get_collection(volume.uuid, path)
+    for file_info in files:
+        if file_info.name not in (".", ".."):
+            if file_info.type == "file":
+                print(f"{path}{file_info.name}")
+            elif file_info.type == "directory" and file_info.name != ".snapshot":
+                print(f"{path}{file_info.name}/")
+                list_files(volume, f"{path}{file_info.name}/")
+
+
+def delete(volume, pathname, recursive=False):
+    """Delete a file or directory on a volume"""
+    try:
+        resource = FileInfo(volume.uuid, path=pathname)
+        resource.delete(recurse=recursive)
+    except NetAppRestError as error:
+        if recursive:
+            extra = "(recursively) "
+        else:
+            extra = ""
+        message = f"delete: File or dir {pathname} wasn't deleted {extra}on {volume.name} ({error})"
+        logging.critical(message)
+
+
+def create_directory(volume, pathname):
+    """Create a directory on a volume"""
+    resource = FileInfo(volume.uuid, pathname)
+    resource.type = "directory"
+    resource.unix_permissions = "644"
+    try:
+        resource.post()
+    except NetAppRestError as error:
+        message = f"create_directory: dir {pathname} wasn't created on {volume.name} ({error})"
+        logging.critical(message)
+
+
+def create_file(volume, pathname, contents=""):
+    """Create a file on a volume
+
+    Args:
+        volume (string): volume object to create file on
+        pathname (string): path to file that is to be created
+        contents (string): contents of file to be written
+    """
+    try:
+        resource = FileInfo(volume.uuid, pathname)
+        resource.post(
+            hydrate=True, data=contents)
+        resource.patch()
+    except NetAppRestError as error:
+        message = f"create_file: File {pathname} was not created on {volume.name} ({error})"
+        logging.critical(message)
+
+
+def file_handling(volume_name):
+    """Demonstrate file handling on NetApp
+    Writes some files and directories to the volume then cleans up what it has written
+    Note: Does not check for existing content.
+    Args:
+        volume_name (string): Volume name to us 
+    """
+    try:
+        all_volumes = list(Volume.get_collection(name=volume_name))
+        for vol in all_volumes:
+            print(f"Volume: {vol.name} ({vol.uuid})")
+            create_file(vol, "alice", "lorem ipsum")
+
+            create_directory(vol, "bobsfiles")
+            create_file(vol, "bobsfiles/bob", "lorem ipsum")
+
+            create_directory(vol, "bobsfiles/charliesfiles")
+            create_file(
+                vol, "bobsfiles/charliesfiles/charlie1", "lorem ipsum")
+            create_file(
+                vol, "bobsfiles/charliesfiles/charlie2", "lorem ipsum")
+            create_file(
+                vol, "bobsfiles/charliesfiles/charlie3", "lorem ipsum")
+
+            list_files(vol, "/")
+
+            print("Cleaning up...")
+            delete(vol, "/alice", False)
+            delete(vol, "/bobsfiles", True)
+            print("Done.")
+    except NetAppRestError as error:
+        print("Exception :" + str(error))
+
+
+def main() -> None:
+    """Main function"""
+
+    arguments = [
+        Argument("-c", "--cluster", "API server IP:port details"),
+        Argument("-v", "--volume_name", "Volume Name")]
+    args = parse_args(
+        "Demonstrates file and directory creation, file and directory creation deletion",
+        arguments,
+    )
+
+    setup_connection(args.cluster, args.api_user, args.api_pass)
+    file_handling(args.volume_name)
+
+
+if __name__ == "__main__":
+    main()