Commit initial code.

drivepy/config.py

@@ -0,0 +1,114 @@
+import mimetypes
+import os
+
+GOOGLE_DRIVE_SCOPE = 'https://www.googleapis.com/auth/drive'
+
+ALL_FILE_FIELDS = 'files,kind,nextPageToken'
+
+ALL_SINGLE_FILE_FIELDS = 'appProperties,capabilities,contentHints,createdTime,description,explicitlyTrashed,fileExtension,folderColorRgb,fullFileExtension,headRevisionId,iconLink,id,imageMediaMetadata,isAppAuthorized,kind,lastModifyingUser,md5Checksum,mimeType,modifiedByMeTime,modifiedTime,name,originalFilename,ownedByMe,owners,parents,permissions,properties,quotaBytesUsed,shared,sharedWithMeTime,sharingUser,size,spaces,starred,thumbnailLink,trashed,version,videoMediaMetadata,viewedByMe,viewedByMeTime,viewersCanCopyContent,webContentLink,webViewLink,writersCanShare'
+
+ALL_PERMISSION_FIELDS = 'allowFileDiscovery,displayName,domain,emailAddress,id,kind,photoLink,role,type'
+
+class PermissionRole(object):
+    Reader = 'reader'
+    Writer = 'writer'
+    Owner = 'owner'
+    Commenter = 'commenter'
+
+class PermissionType(object):
+    User = 'user'
+    Group = 'group'
+    Domain = 'domain'
+    Anyone = 'anyone'
+
+class DownloadType(object):
+    # Documents
+    HTML = 'text/html'
+    RichText = 'text/rtf'
+    OpenOfficeDoc = 'application/vnd.oasis.opendocument.text'
+    WordDoc = 'application/vnd.openxmlformats-officedocument.wordprocessingml.document'
+
+    # Spreadsheets
+    CSV = 'text/csv'
+    Excel = 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'
+    OpenOfficeSheet = 'application/x-vnd.oasis.opendocument.spreadsheet'
+
+    # Drawings
+    JPEG = 'image/jpeg'
+    PNG = 'image/png'
+    SVG = 'image/svg+xml'
+
+    # Presentations
+    PowerPoint = 'application/vnd.openxmlformats-officedocument.presentationml.presentation'
+
+    # Scripts
+    JSON = 'application/vnd.google-apps.script+json'
+
+    # All except Scripts
+    PDF = 'application/pdf'
+
+    # All except scripts and sheets
+    PlainText = 'text/plain'
+
+class MimeTypes(object):
+    Audio = 'application/vnd.google-apps.audio'
+    Document = 'application/vnd.google-apps.document'
+    Drawing = 'application/vnd.google-apps.drawing'
+    File = 'application/vnd.google-apps.file'
+    Folder = 'application/vnd.google-apps.folder'
+    Form = 'application/vnd.google-apps.form'
+    FusionTable = 'application/vnd.google-apps.fusiontable'
+    Map = 'application/vnd.google-apps.map'
+    Photo = 'application/vnd.google-apps.photo'
+    Presentation = 'application/vnd.google-apps.presentation'
+    Script = 'application/vnd.google-apps.script'
+    Sites = 'application/vnd.google-apps.sites'
+    Spreadsheet = 'application/vnd.google-apps.spreadsheet'
+    Unknown = 'application/vnd.google-apps.unknown'
+    Video = 'application/vnd.google-apps.video'
+
+    TypeDict = {}
+
+    types = {
+        Audio: ['mp3', 'wav', 'aac', 'ogg', 'wma'],
+        Document: ['txt', 'doc', 'docx', 'log', 'odt', 'rtf'],
+        Folder: [''],
+        Presentation: ['ppt', 'pptx'],
+        Spreadsheet: ['csv', 'xls', 'xlsx', 'xlt', 'xml', 'ods'],
+        Video: ['flv', 'mp4', 'mpg', 'mov', 'mkv', 'avi']
+    }
+
+    for mime, exts in types.items():
+        for ext in exts:
+            TypeDict[ext] = mime
+
+    DownloadTypeDict = {
+        Document: DownloadType.WordDoc,
+        Spreadsheet: DownloadType.Excel,
+        Drawing: DownloadType.PDF,
+        Presentation: DownloadType.PowerPoint,
+        Script: DownloadType.JSON
+    }
+
+    @staticmethod
+    def get_download_type(mime_type):
+        """Tries to guess the download type based on the mime_type."""
+        return MimeTypes.DownloadTypeDict.get(mime_type, 'text/plain')
+
+    @staticmethod
+    def guess_type(path):
+        """Guesses the mimetype of a file extension
+
+        Args:
+            path: the file name or path
+
+        Returns:
+            A mimetype string
+        """
+        ext = os.path.split(path)[-1].replace('.', '')
+        guess = MimeTypes.TypeDict.get(ext)
+
+        if guess is None:
+            guess = mimetypes.guess_type(path)[0] or MimeTypes.File
+
+        return guess