fetch-gdrive-object.py

@@ -13,11 +13,16 @@
 # 	Google Sheets: .xslx, .ods, .csv, .tsv, .pdf, and .html
 # 	Google Slides: .pptx, .pdf, .svg, .jpg, .png, and .txt
 # 	Google Drawings: pdf, .svg, .jpg, .png
+# 	Files: 
 
 # todo 
 #	test on all machines (py2 or py3 code?)
 #	look for -i error in log to determine success? (in fetchObject())
-# 	support presentation drawings text # slide is the language used elsewhere 
+# 	support for other object types: 
+#		file - done
+#		presentation 
+#		drawings # slide is the language used elsewhere 
+#		text 
 #	pylint 
 #	test under osx so that people can use it on their local machines to fetch workflows 
 
@@ -33,6 +38,9 @@ def fetchObject(documentId, objectType, outputFormat, outputName, logFile):
 	elif objectType == 'doc':
 		location = 'document/d/'
 
+	elif objectType == 'file':
+		location = 'file/d/'
+
 	# DOC_ID=1olFTlH4ssJIv01n40-44BIDda0OkXcFhXj-SLVtZwks
 
 	command = 'wget --output-file=' + logFile + ' \"' + base + location + documentId + '/export?format=' + outputFormat 
@@ -43,6 +51,7 @@ def fetchObject(documentId, objectType, outputFormat, outputName, logFile):
 		command = command + '\" '
 
 	command = command + '-O ' + outputName
+
 	_status = os.system(command) 
 
 	return _status
@@ -52,7 +61,7 @@ if __name__ == "__main__":
 
 	parser = argparse.ArgumentParser(description='fetch-gdrive-object')
 	parser.add_argument('-d','--documentId', type=str, help='GDrive Document ID', required=True)
-	parser.add_argument('-t','--objectType', type=str, choices=['doc', 'sheet'], help='Object type {doc, sheet}', required=True)
+	parser.add_argument('-t','--objectType', type=str, choices=['doc', 'sheet', 'file'], help='Object type {doc, sheet, file}', required=True)
 	parser.add_argument('-f','--outputFormat', type=str, help='Output format (must fit type)', required=True)
 	parser.add_argument('-o','--outputName', type=str, help='Output file name', required=True)
 	args = parser.parse_args()
@@ -68,6 +77,9 @@ if __name__ == "__main__":
 	elif args.objectType == 'sheet' and args.outputFormat in ['xslx', 'ods', 'csv', 'tsv', 'pdf', 'html']:
 		status = fetchObject(args.documentId, args.objectType, args.outputFormat, args.outputName, logFile)
 
+	elif args.objectType == 'file' and args.outputFormat in ['xslx', 'ods', 'csv', 'tsv', 'pdf', 'html', 'jpg', 'png']:
+		status = fetchObject(args.documentId, args.objectType, args.outputFormat, args.outputName, logFile)
+
 	else:
 		print('fetch-gdrive-object: error, unsupported objectType or objectType outputFormat combination')
 		exit(-1)