From 917e29d0e8152334499978b23d747d33799492e1 Mon Sep 17 00:00:00 2001
From: stark4n6 <48143894+stark4n6@users.noreply.github.com>
Date: Wed, 8 Mar 2023 13:11:28 -0500
Subject: [PATCH] Update Cello.py
Should fix multiuser issue
---
scripts/artifacts/Cello.py | 208 ++++++++++++++++++-------------------
1 file changed, 101 insertions(+), 107 deletions(-)
diff --git a/scripts/artifacts/Cello.py b/scripts/artifacts/Cello.py
index 9fa41e8e..7bd372c0 100755
--- a/scripts/artifacts/Cello.py
+++ b/scripts/artifacts/Cello.py
@@ -1,4 +1,3 @@
-
import os
import shutil
import sqlite3
@@ -10,18 +9,8 @@
is_windows = is_platform_windows()
slash = '\\' if is_windows else '/'
-def get_cello_db_path(files_found):
- for file_found in files_found:
- file_found = str(file_found)
- if not os.path.basename(file_found) == 'cello.db': # skip -journal and other files
- continue
- elif file_found.find('.magisk') >= 0 and file_found.find('mirror') >= 0:
- continue # Skip sbin/.magisk/mirror/data/.. , it should be duplicate data??
- return file_found
- return ''
-
def get_offline_path(files_found, blob_name):
- #offline_path_relative = "/files/shiny_blobs/blobs"
+
for file_found in files_found:
if file_found.endswith(blob_name):
file_found = str(file_found)
@@ -29,109 +18,114 @@ def get_offline_path(files_found, blob_name):
return ''
def get_Cello(files_found, report_folder, seeker, wrap_text):
- file_found = get_cello_db_path(files_found)
- if not file_found:
- logfunc('Error: Could not get Cello.db path')
- return
+ for file_found in files_found:
+
+ if not os.path.basename(file_found) == 'cello.db': # skip -journal and other files
+ continue
+ elif file_found.find('.magisk') >= 0 and file_found.find('mirror') >= 0:
+ continue # Skip sbin/.magisk/mirror/data/.. , it should be duplicate data??
- if report_folder[-1] == slash:
- folder_name = os.path.basename(report_folder[:-1])
- else:
- folder_name = os.path.basename(report_folder)
+ if report_folder[-1] == slash:
+ folder_name = os.path.basename(report_folder[:-1])
+ else:
+ folder_name = os.path.basename(report_folder)
- db = open_sqlite_db_readonly(file_found)
- cursor = db.cursor()
- cursor.execute('''
- SELECT
- case created_date
- when 0 then ''
- else datetime(created_date/1000, 'unixepoch')
- end as created_date,
- title,
- case modified_date
- when 0 then ''
- else datetime(modified_date/1000, 'unixepoch')
- end as modified_date,
- case shared_with_me_date
- when 0 then ''
- else datetime(shared_with_me_date/1000, 'unixepoch')
- end as shared_with_me_date,
- case modified_by_me_date
- when 0 then ''
- else datetime(modified_by_me_date/1000, 'unixepoch')
- end as modified_by_me_date,
- case viewed_by_me_date
- when 0 then ''
- else datetime(viewed_by_me_date/1000, 'unixepoch')
- end as viewed_by_me_date,
- mime_type,
- Quota_bytes,
- case is_folder
- when 0 then ''
- when 1 then 'Yes'
- end as is_folder,
- case is_owner
- when 0 then ''
- when 1 then 'Yes'
- end as is_owner,
- case trashed
- when 0 then ''
- when 1 then 'Yes'
- end as trashed,
- (SELECT value from item_properties where key='offlineStatus' and item_stable_id=stable_id) as offline_status,
- (SELECT json_extract(value, '$.blobKey') from item_properties where key LIKE 'com.google.android.apps.docs:content_metadata%' and item_stable_id=stable_id) as content_metadata
- FROM items
- ''')
+ db = open_sqlite_db_readonly(file_found)
+ cursor = db.cursor()
+ cursor.execute('''
+ SELECT
+ case created_date
+ when 0 then ''
+ else datetime(created_date/1000, 'unixepoch')
+ end as created_date,
+ title,
+ case modified_date
+ when 0 then ''
+ else datetime(modified_date/1000, 'unixepoch')
+ end as modified_date,
+ case shared_with_me_date
+ when 0 then ''
+ else datetime(shared_with_me_date/1000, 'unixepoch')
+ end as shared_with_me_date,
+ case modified_by_me_date
+ when 0 then ''
+ else datetime(modified_by_me_date/1000, 'unixepoch')
+ end as modified_by_me_date,
+ case viewed_by_me_date
+ when 0 then ''
+ else datetime(viewed_by_me_date/1000, 'unixepoch')
+ end as viewed_by_me_date,
+ mime_type,
+ Quota_bytes,
+ case is_folder
+ when 0 then ''
+ when 1 then 'Yes'
+ end as is_folder,
+ case is_owner
+ when 0 then ''
+ when 1 then 'Yes'
+ end as is_owner,
+ case trashed
+ when 0 then ''
+ when 1 then 'Yes'
+ end as trashed,
+ (SELECT value from item_properties where key='offlineStatus' and item_stable_id=stable_id) as offline_status,
+ (SELECT json_extract(value, '$.blobKey') from item_properties where key LIKE 'com.google.android.apps.docs:content_metadata%' and item_stable_id=stable_id) as content_metadata
+ FROM items
+ ''')
- all_rows = cursor.fetchall()
- usageentries = len(all_rows)
- if usageentries > 0:
- report = ArtifactHtmlReport('Cello')
- report.start_artifact_report(report_folder, 'Cello')
- report.add_script()
- data_headers = ('Created Date','File Name','Modified Date','Shared with User Date','Modified by User Date','Viewed by User Date','Mime Type', \
- 'Offline','Quota Size','Folder','User is Owner','Deleted')
- data_list = []
- tsv_list = []
- for row in all_rows:
- doc_name = row[1]
- offline_status = "No"
- if row[11] == 1: # file is offline
- offline_status = "Yes"
- offline_path_name = row[12]
- if offline_path_name not in (None, ''):
- offline_path = get_offline_path(files_found, offline_path_name)
- if offline_path:
- destination_path = get_next_unused_name(os.path.join(report_folder, doc_name))
- shutil.copy2(offline_path, destination_path)
- dest_name = os.path.basename(destination_path)
- doc_name = f"{doc_name}"
+ all_rows = cursor.fetchall()
+ usageentries = len(all_rows)
+ if usageentries > 0:
+
+ account_name = os.path.basename(os.path.dirname(file_found))
+
+ report = ArtifactHtmlReport(f'Cello - {account_name}')
+ report.start_artifact_report(report_folder, f'Cello - {account_name}')
+ report.add_script()
+ data_headers = ('Created Date','File Name','Modified Date','Shared with User Date','Modified by User Date','Viewed by User Date','Mime Type', \
+ 'Offline','Quota Size','Folder','User is Owner','Deleted')
+ data_list = []
+ tsv_list = []
+ for row in all_rows:
+ doc_name = row[1]
+ offline_status = "No"
+ if row[11] == 1: # file is offline
+ offline_status = "Yes"
+ offline_path_name = row[12]
+ if offline_path_name not in (None, ''):
+ offline_path = get_offline_path(files_found, offline_path_name)
+ if offline_path:
+ destination_path = get_next_unused_name(os.path.join(report_folder, doc_name))
+ shutil.copy2(offline_path, destination_path)
+ dest_name = os.path.basename(destination_path)
+ doc_name = f"{doc_name}"
+ else:
+ logfunc(f'File {doc_name} not present offline!')
else:
logfunc(f'File {doc_name} not present offline!')
+ if row[8] == "Yes":
+ doc_name = ' ' + doc_name
else:
- logfunc(f'File {doc_name} not present offline!')
- if row[8] == "Yes":
- doc_name = ' ' + doc_name
- else:
- if doc_name.startswith('