投稿者「管理人」のアーカイブ

TCPポートスキャナ

コメントを残す

TCP 1~65535までのポートスキャンを行う。
高速化のためスレッドを30000作成した。
PCが重たかったらif i%30000のところで調整する。

使用例)
python tcp_scan.py 192.168.x.y

import socket
import sys
import threading

opend_port = []

def port_scan(ip,port):
    global opend_port
    sd = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    sd.settimeout(1)
    res = sd.connect_ex((ip,port))
    sd.close()
    if res == 0:
        opend_port.append(port)

def main():
    ip = sys.argv[1]
    
    print(f"target is {ip}")

    threads = []
    max_port = 65535
    for i in range(1,max_port+1):
        print(f"{i} of {max_port}\r",end="")
        port = i
        thread = threading.Thread(target=port_scan,args=(ip,port,))
        threads.append(thread)
        thread.start()
        if i%30000 == 0:
            for thread in threads:
                thread.join()
            threads = []
        
    if len(threads):
        for thread in threads:
            thread.join()

    print("opened tcp ports >")
    for i in opend_port:
        print(i)
    return

if __name__ == "__main__":
    main()

Pythonで画像のExif情報を削除する

コメントを残す

指定したファイルまたはディレクトリ以下のファイルのExif情報を削除する。

ソースコード1
指定したファイルのExifを削除する

from PIL import Image
from PIL.ExifTags import TAGS
import os

def delete_exif(filename):
    image = Image.open(filename)
    image_data = image.getdata()
    image_mode = image.mode
    image_size = image.size

    new_image = Image.new(image_mode,image_size)
    new_image.putdata(image_data)
    old_filename = os.path.splitext(filename)[0]
    old_ext = os.path.splitext(filename)[1][1:]
    new_file = os.path.join(filename,f"{old_filename}_noexif.{old_ext}")
    new_image.save(new_file)
    print(f"        New file > {new_file}")

def main():
    filename = input("path > ")
    delete_exif(filename)

    return

if __name__ == "__main__":
    main()

ソースコード2
指定したディレクトリ以下の「jpg/jpeg」、「png」画像ファイルのExif情報を削除する

from PIL import Image
import os

def main():
    print("Exif delete tool")
    root_path = input("Directory path > ")

    print("Searching...")
    for root,dirs,files in os.walk(root_path):
        for file in files:
            path = os.path.join(root,file)
            ext = os.path.splitext(file)[1][1:]
            target_ext = ["jpg","jpeg","png"]
            
            if ext in target_ext:
                print(f"    file > {file}")
                image = Image.open(path)
                image_data = image.getdata()
                image_mode = image.mode
                image_size = image.size

                new_image = Image.new(image_mode,image_size)
                new_image.putdata(image_data)
                old_filename = os.path.splitext(file)[0]
                old_ext = os.path.splitext(file)[1][1:]
                new_file = os.path.join(root,f"{old_filename}_noexif.{old_ext}")
                new_image.save(new_file)
                print(f"        New file > {new_file}")
            
    input("END")
    return

if __name__ == "__main__":
    main()

Pythonで再帰的にファイル名を取得する

コメントを残す

再帰的にファイル名を取得する方法のテストプログラム。
カレントディレクトリに下記の構造でテストディレクトリを生成する。
生成後、os.walk()関数で生成したディレクトリ、ファイルと同様のパスが取得できているか確認を行う。

example
    ├─dir_0
    │      file_0
    │      file_1
    │      file_2
    │      file_3
    │      file_4
    │      file_5
    │      file_6
    │      file_7
    │      file_8
    │      file_9
    │
    ├─dir_1
    │      file_0
    │      file_1
    │      file_2
    │      file_3
    │      file_4
    │      file_5
    │      file_6
    │      file_7
    │      file_8
    │      file_9
    │
    ├─dir_2
    │      file_0
    │      file_1
    │      file_2
    │      file_3
    │      file_4
    │      file_5
    │      file_6
    │      file_7
    │      file_8
    │      file_9
    │
    ├─dir_3
    │      file_0
    │      file_1
    │      file_2
    │      file_3
    │      file_4
    │      file_5
    │      file_6
    │      file_7
    │      file_8
    │      file_9
    │
    └─dir_4
            file_0
            file_1
            file_2
            file_3
            file_4
            file_5
            file_6
            file_7
            file_8
            file_9

ソースコード

# walk.py

import os
import shutil

def make_test_files(path):
    # ファイルを10個生成
    for i in range(10):
        filename = os.path.join(path,f"file_{i}")
        fd = open(filename,"w")
        fd.close()

    return

def make_test_dirs():
    root_dir = "example"

    # ディレクトリが存在した場合は、削除する(mkdirエラー対策)
    if os.path.exists(root_dir):
        shutil.rmtree(root_dir)

    os.mkdir(root_dir)

    # ディレクトリを5個生成
    for i in range(5):
        sub_dir = os.path.join(root_dir,f"dir_{i}")
        if os.path.exists(sub_dir) == False:
            os.makedirs(sub_dir)
            make_test_files(sub_dir)

    return
    
def main():
    # カレントディレクトリにテスト用ファイルを生成
    make_test_dirs()

    # カレントディレクトリのファイル一覧を表示
    root_path = "."
    for root,dirs,files in os.walk(root_path):
        for file in files:
            path = os.path.join(root,file)
            print(path)

    return

if __name__ == "__main__":
    main()

実行と結果

python --version
Python 3.12.2
python walk.py
.\walk.py
.\example\dir_0\file_0
.\example\dir_0\file_1
.\example\dir_0\file_2
.\example\dir_0\file_3
.\example\dir_0\file_4
.\example\dir_0\file_5
.\example\dir_0\file_6
.\example\dir_0\file_7
.\example\dir_0\file_8
.\example\dir_0\file_9
.\example\dir_1\file_0
.\example\dir_1\file_1
.\example\dir_1\file_2
.\example\dir_1\file_3
.\example\dir_1\file_4
.\example\dir_1\file_5
.\example\dir_1\file_6
.\example\dir_1\file_7
.\example\dir_1\file_8
.\example\dir_1\file_9
.\example\dir_2\file_0
.\example\dir_2\file_1
.\example\dir_2\file_2
.\example\dir_2\file_3
.\example\dir_2\file_4
.\example\dir_2\file_5
.\example\dir_2\file_6
.\example\dir_2\file_7
.\example\dir_2\file_8
.\example\dir_2\file_9
.\example\dir_3\file_0
.\example\dir_3\file_1
.\example\dir_3\file_2
.\example\dir_3\file_3
.\example\dir_3\file_4
.\example\dir_3\file_5
.\example\dir_3\file_6
.\example\dir_3\file_7
.\example\dir_3\file_8
.\example\dir_3\file_9
.\example\dir_4\file_0
.\example\dir_4\file_1
.\example\dir_4\file_2
.\example\dir_4\file_3
.\example\dir_4\file_4
.\example\dir_4\file_5
.\example\dir_4\file_6
.\example\dir_4\file_7
.\example\dir_4\file_8
.\example\dir_4\file_9

Pythonでコマンドライン引数を扱う

コメントを残す

Pythonでコマンドライン引数を取得する。

例) python argv.py kinako mochi

sys.argv[0] → argv.py
sys.argv[1] → kinako
sys.argv[2] → mochi
len(sys.argv) → 3

ソースコード

# argv.py

import sys

def main():
    print(f"sys.argvに格納されている要素 → {sys.argv}")
    print(f"sys.argvの要素数 → {len(sys.argv)}")

    print("sys.argvのデータへのアクセス①")
    for data in sys.argv:
        print(data)
    
    print("sys.argvのデータへのアクセス②")
    for i in range(len(sys.argv)):
        print(sys.argv[i])

    return

if __name__ == "__main__":
    main()

実行と結果

python --version
Python 3.12.2
python argv.py kinako mochi

sys.argvに格納されている要素 → ['argv.py', 'kinako', 'mochi']
sys.argvの要素数 → 3
sys.argvのデータへのアクセス①
argv.py
kinako
mochi
sys.argvのデータへのアクセス②
argv.py
kinako
mochi