import os
from zipfile import ZipFile
zip_name = "myfunction.zip"
if os.path.exists(zip_name):
os.remove(zip_name)
for root,dirs,files in os.walk('.'):
with ZipFile(zip_name,'w') as zf:
for fn in files:
zf.write(fn)

import os
from zipfile import ZipFile
zip_name = "myfunction.zip"
if os.path.exists(zip_name):
os.remove(zip_name)
for root,dirs,files in os.walk('.'):
with ZipFile(zip_name,'w') as zf:
for fn in files:
zf.write(fn)