Lokasi ngalangkungan proxy:   [ UP ]  
[Ngawartoskeun bug]   [Panyetelan cookie]                
Skip to content

Commit 44389d3

Browse files
committed
add mime detection for file upload
1 parent d2b8626 commit 44389d3

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

github3/models.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,8 @@
66
"""
77

88
from .helpers import to_python, to_api
9+
import os
10+
import magic
911
from .config import settings
1012
import requests
1113

@@ -178,8 +180,10 @@ def downloads(self, **params):
178180
self.name, 'downloads'), Download, **params)
179181

180182
def create_download(self, filepath, **params):
181-
import os
182-
data = {'size': os.path.getsize(filepath), 'name': filepath.split('/')[-1]}
183+
m = magic.open(magic.MAGIC_MIME)
184+
m.load()
185+
mime_type = m.file(filepath)
186+
data = {'size': os.path.getsize(filepath), 'name': filepath.split('/')[-1], 'content_type': mime_type}
183187

184188
dlressource = self._gh._post_resource(('repos', self.owner.login,
185189
self.name, 'downloads'),

0 commit comments

Comments
 (0)