def do_upload():
upload = request.files.get('upload')
url = request.forms.get('url')
content_type = upload.content_type
name, ext = os.path.splitext(upload.filename)
save_path = '/tmp'
upload.filename = str(abs(hash(name)))+ ext
upload.save(save_path,overwrite=True) # appends upload.filename automatically
fileHandler = FileHandler(5)
fileId,physicalName = fileHandler.save('/tmp/'+upload.filename ,url)
return physicalName
class TestHander(unittest.TestCase):
dbHandler = None;
fileHandler = None;
def setUp(self):
self.dbHandler = DBHandler()
self.fileHandler = FileHandler(5)
def testFileHanderSave(self,url = 'aa'):
path = '/tmp/license.txt'
import os
if not os.path.isfile(path):
os.mknod(path)
fileId,physicalName = self.fileHandler.save(path,url)
return fileId,physicalName
def testGetFilePathByPhysicalName(self):
fileId,physicalName = self.testFileHanderSave()
path = self.fileHandler.getFilePathByDiskName(physicalName)
print 'path :'+ path
def testGetFiletypeByURL(self):
url = 'http://www.abc.com'
self.testFileHanderSave(url)
print 'file type is :' + self.fileHandler.getTypeByUrl(url)
def testGetFileByID(self):
fileId,physicalName = self.testFileHanderSave()
opendFile = self.fileHandler.getFileByID(fileId)
def testGetFiletypeByID(self):
fileId,physicalName = self.testFileHanderSave()
print self.fileHandler.getTypeByID(fileId)
def testGetTotalFileCount(self):
num = self.fileHandler.getTotalFileCount()
print "file count : " + str(num)
def testExecute(self):
sql = 'select * from files'
rows = self.dbHandler.excute(sql)