From 66cabf1af244e18fe488525389ee6bd9e7962af8 Mon Sep 17 00:00:00 2001
From: Rhet Turnbull <rturnbull+git@gmail.com>
Date: Sun, 12 Jan 2020 07:42:30 -0800
Subject: [PATCH] Added cloudasset/incloud options to CLI query

---
 osxphotos/__main__.py  | 42 ++++++++++++++++++++++++++++++++++++++++++
 osxphotos/photoinfo.py |  4 ++++
 2 files changed, 46 insertions(+)

diff --git a/osxphotos/__main__.py b/osxphotos/__main__.py
index 4e132e77..7b92444a 100644
--- a/osxphotos/__main__.py
+++ b/osxphotos/__main__.py
@@ -266,6 +266,10 @@ def list_libraries(cli_obj):
 @click.option(
     "--not-live", is_flag=True, help="Search for photos that are not Apple live photos"
 )
+@click.option("--cloudasset", is_flag=True, help="Search for photos that are part of an iCloud library")
+@click.option("--not-cloudasset", is_flag=True, help="Search for photos that are not part of an iCloud library")
+@click.option("--incloud", is_flag=True, help="Search for photos that are in iCloud (have been synched)")
+@click.option("--not-incloud", is_flag=True, help="Search for photos that are not in iCloud (have not been synched)")
 @click.option(
     "--only-movies",
     is_flag=True,
@@ -315,6 +319,10 @@ def query(
     not_burst,
     live,
     not_live,
+    cloudasset,
+    not_cloudasset,
+    incloud,
+    not_incloud,
 ):
     """ Query the Photos database using 1 or more search options; 
         if more than one option is provided, they are treated as "AND" 
@@ -349,6 +357,10 @@ def query(
             not_burst,
             live,
             not_live,
+            cloudasset,
+            not_cloudasset,
+            incloud,
+            not_incloud,
         ]
     ):
         click.echo(cli.commands["query"].get_help(ctx))
@@ -385,6 +397,14 @@ def query(
         # can't search for both live and not_live
         click.echo(cli.commands["query"].get_help(ctx))
         return
+    elif cloudasset and not_cloudasset:
+        # can't search for both live and not_live
+        click.echo(cli.commands["query"].get_help(ctx))
+        return
+    elif incloud and not_incloud:
+        # can't search for both live and not_live
+        click.echo(cli.commands["query"].get_help(ctx))
+        return
 
     # actually have something to query
     isphoto = ismovie = True  # default searches for everything
@@ -422,6 +442,10 @@ def query(
         not_burst,
         live,
         not_live,
+        cloudasset,
+        not_cloudasset,
+        incloud,
+        not_incloud
     )
     print_photo_info(photos, cli_obj.json or json)
 
@@ -758,6 +782,8 @@ def print_photo_info(photos, json=False):
                 "burst",
                 "live_photo",
                 "path_live_photo",
+                "iscloudasset",
+                "incloud",
             ]
         )
         for p in photos:
@@ -788,6 +814,8 @@ def print_photo_info(photos, json=False):
                     p.burst,
                     p.live_photo,
                     p.path_live_photo,
+                    p.iscloudasset,
+                    p.incloud,
                 ]
             )
         for row in dump:
@@ -823,6 +851,10 @@ def _query(
     not_burst,
     live,
     not_live,
+    cloudasset,
+    not_cloudasset,
+    incloud,
+    not_incloud,
 ):
     """ run a query against PhotosDB to extract the photos based on user supply criteria """
     """ used by query and export commands """
@@ -915,6 +947,16 @@ def _query(
     elif not_live:
         photos = [p for p in photos if not p.live_photo]
 
+    if cloudasset:
+        photos = [p for p in photos if p.iscloudasset]
+    elif not_cloudasset:
+        photos = [p for p in photos if not p.iscloudasset]
+    
+    if incloud:
+        photos = [p for p in photos if p.incloud]
+    elif not_incloud:
+        photos = [p for p in photos if not p.incloud]
+        
     return photos
 
 
diff --git a/osxphotos/photoinfo.py b/osxphotos/photoinfo.py
index 63a405f5..63c2b4f4 100644
--- a/osxphotos/photoinfo.py
+++ b/osxphotos/photoinfo.py
@@ -661,6 +661,8 @@ class PhotoInfo:
             "burst": self.burst,
             "live_photo": self.live_photo,
             "path_live_photo": self.path_live_photo,
+            "iscloudasset": self.iscloudasset,
+            "incloud": self.incloud,
         }
         return yaml.dump(info, sort_keys=False)
 
@@ -692,6 +694,8 @@ class PhotoInfo:
             "burst": self.burst,
             "live_photo": self.live_photo,
             "path_live_photo": self.path_live_photo,
+            "iscloudasset": self.iscloudasset,
+            "incloud": self.incloud,
         }
         return json.dumps(pic)