os.walk()でカレントフォルダのファイル名だけ欲しい。サブフォルダ内のファイル名を無視したい。

Question

カレントフォルダ内のファイルをzip圧縮して削除、ただし、サブフォルダは無視するコードを作りたい。
下記コードでカレントフォルダ内のファイル圧縮は出来ているのだけれど、カレントフォルダの中のサブフォルダの中にファイルがあるとファイル名がないとエラーで止まります。
サブフォルダの中にカレントフォルダの中のファイルと同じファイル名があると、カレントフォルダのそのファイルが二回入ります。
サブフォルダとかサブフォルダ内のファイルを無視したいです。

    with zipfile.ZipFile('./archive.zip', 'w') as myZip:
        des_folder = './'
        # osモジュールで街頭パスのファイルを取得する。(os.walkはパスの派生フォルダまで取得する。)
        for folder, subfolders, files in os.walk(des_folder):
            for file in files:
                full_name = os.path.join(des_folder, file)
                # writeは圧縮ファイルでファイルを書き込む。パラメータは対処ファイル、格納するパスで指定する。
                if full_name != './archive.zip':
                    myZip.write(full_name, os.path.relpath(full_name, des_folder))
                    os.remove(full_name)

どうしたら良いでしょうか。

Answer 1

元の形を保ちたいのであれば

import os
with zipfile.ZipFile('./archive.zip', 'w') as myZip:
    des_folder = './'
    for folder, subfolders, files in os.walk(des_folder):
        for file in files:
            full_name = os.path.join(des_folder, file)
            if full_name != './archive.zip':
                myZip.write(full_name)
                os.remove(full_name)
        break

ループを1回目だけ実行して、すぐに抜けるという変則的なコードです。
上記のコードは三番目に考えたコードで、その前のコードは以下です。

import os
with zipfile.ZipFile('archive.zip', 'w') as myZip:
    for file in os.walk('./').__next__()[2]:
        if file != 'archive.zip':
            myZip.write(file)
            os.remove(file)

os.walk('./')はジェネレータですので、それを__next__()で呼び出すと長さが3のリストが帰ってきます。その最後の要素には、現在のディレクトリにあるサブディレクトリではないファイルが格納されています。ただ、ジェネレータの__next__メソッドを使うのは一般的ではないので上記のコードにしました。

一番最初に思いついたコード下のコードです。これははotnさんも書かれていた方法です。

import os
import glob
with zipfile.ZipFile('archive.zip', 'w') as myZip:
    for file in glob.glob("*"):
        if os.path.isfile(file) and file != 'archive.zip':
            myZip.write(file)
            os.remove(file)

しかし、質問のタイトルに、「os.walk()で」とあったので、os.walkを使う答えを考えました。
Aki1000さんはいろいろな方法を知りたいというタイプのようですので、いろいろなやり方を書いておきました。行数が一番短いのは二番目の方法ですが、普通こういうのはやらないでしょう。

Comments

[Aki1000]

おお、出来ますね。

>元の形を保ちたいのであれば
もっと綺麗になるならその方が良いですけど、なんか簡単な方法あるのですか？

[ppaul]

回答を追加したのでご覧ください。

[Aki1000]

ありがとうございます！

Answer 2

サブフォルダとかサブフォルダ内のファイルを無視したいです。

だけみて回答すると、

for file in glob.glob("*"):
    if os.path.isfile(file):
        ～～～