Working directory sorununu asmak icin bir wrapper fonksiyon ekleyelim..
Simdi testleri gozden gecirecegiz, hala calismiyor lakin korkmaya gerek yok.
This commit is contained in:
+10
-1
@@ -77,11 +77,20 @@ class ArchiveZip(ArchiveBase):
|
|||||||
self.zip.close()
|
self.zip.close()
|
||||||
|
|
||||||
def add_to_archive(self, fileName):
|
def add_to_archive(self, fileName):
|
||||||
|
"""A wrapper function to handle working directory sh*t"""
|
||||||
|
cwd = os.getcwd()
|
||||||
|
pathName = os.path.dirname(fileName)
|
||||||
|
fileName = os.path.basename(fileName)
|
||||||
|
os.chdir(pathName)
|
||||||
|
self.add_file(fileName)
|
||||||
|
os.chdir(cwd)
|
||||||
|
|
||||||
|
def add_file(self, fileName):
|
||||||
"""Add file or directory to a zip file"""
|
"""Add file or directory to a zip file"""
|
||||||
if os.path.isdir(fileName) and not os.path.islink(fileName):
|
if os.path.isdir(fileName) and not os.path.islink(fileName):
|
||||||
self.zip.writestr(fileName + '/', '')
|
self.zip.writestr(fileName + '/', '')
|
||||||
for f in os.listdir(fileName):
|
for f in os.listdir(fileName):
|
||||||
self.add_to_archive(fileName + '/' + f)
|
self.add_file(fileName + '/' + f)
|
||||||
else:
|
else:
|
||||||
if os.path.islink(fileName):
|
if os.path.islink(fileName):
|
||||||
dest = os.readlink(fileName)
|
dest = os.readlink(fileName)
|
||||||
|
|||||||
Reference in New Issue
Block a user