やること

よく使いたくなるが、細かい事は忘れがちなので、
osモジュール系、ファイル読み書き系を備忘録として。

環境

virtualbox 5.2.26
vagrant 2.1.2
python 3.5.2
mac 10.14.2

ファイル読み込み

※読み込みはopenのモード省略可

ファイル全体を文字列として読み込み `read()`

input.py


with open('test.txt', encoding='utf-8') as f:
  contents = f.read()
  print(contents.rstrip())

output.py

huhu
haha
yeah

ファイル全体をリストとして読み込み `readlines()`

input.py

with open('test.txt', encoding='utf-8') as f:
  contents = f.readlines()
  print(contents)

output.py

['huhu\n','haha\n','yeah\n']

ファイルを一行ずつ読み込み `readline()`

input.py

with open('test.txt', encoding='utf-8') as f:
  contents = f.readline()
  print(contents.rstrip())

  contents = f.readline()
  print(contents.rstrip())

  contents = f.readline()
  print(contents.rstrip())

# 読み込む度一行ずつ進んでいく。

output.py

huhu
haha
yeah

forで読み込む

input.py

with open('test.txt', encoding='utf-8') as f:
  for row in f:
    print(row.rstrip())

output.py

huhu
haha
yeah

ファイル書き込み

上書き `mode='w'`

input.py

with open('test.txt', 'w', encoding='utf-8') as f:
  f.write('''yeah
haha
huhu
''')

test.txt

yeah
haha
huhu

末尾に追加書き込み `mode='a'`

input.py

with open('test.py', 'a', encoding='utf-8') as f:
  f.write('hello')

test.txt

yeah
haha
huhu
hello

先頭、途中に書き込みする `mode='r+'`

input.py

with open('text.txt', encoding='utf-8') as f:
  l = f.readlines()
  l.insert(0, 'huhahahaha\n')

# insert(PLACE, VALUE)

with open('text.txt', 'r+', encoding='utf-8') as f:
  f.writelines(l)

# writelines = シーケンス型を書き込む

test.py

huhahahaha
yeah
haha
huhu
hello

OSモジュール

os.path()

`os.path.exists()`

指定フォルダにファイルが存在するか確認する

input.py

import os


def search(target, file_name):
  path = os.path.join(target, file_name)
  # os.path.joinはos関係なくパスを連結してくれる。
  
  if os.path.exists(path):
    print(path + ' は存在するらしい')
  else:
    print(path + ' は存在しない？')

if __name__ == '__main__':
  search('宇宙', '生命体')
  search('test', 'test.py')

output

宇宙/生命体 は存在しない？
test/test.py は存在するらしい

`os.path.isfile()` , `os.path.isdir()`

指定ファイル、指定ディレクトリがファイルかディレクトリか判別

input.py

import os 


def whatIsThis(target, file_name):
  path = os.path.join(target, file_name)

  if os.path.isfile(path):
    print(path + ' はファイルです。')

  if os.path.isdir(path):
    print(path + ' はディレクトリです。')

if __name__ == '__main__':
  whatIsThis(test, t.txt)
  whatIsThis(test, t)

output

test/t.txt はファイルです。
test/t はディレクトリです。

`os.listdir()`

指定ディレクトリ内のファイルやディレクトリのリストを返す

input.py

import os


def dir_list(target):
  for name in os.listdir(target):
    print(name)

if __name__ == '__main__':
  dir_list('.')

output

input.py
test

`os.walk()`

指定ディレクトリ内の子ディレクトリまで全てリストで返す

input.py

import os


# 子ディレクトリ内のファイルも全て連結して表示
def file_all(target):
  for root, dirs, files in os.walk(target):
    for file in files:
      path = os.path.join(root, file)
      print(path)

if __name__ == '__main__':
  dir_all('.')

output

./input.py
./test/test.txt