python常用文件操作

python常用文件操作

ps…记录一些常用的文件操作

import os


#获取系统模块参数
# print(os.environ.keys())

#系统盘符
# print(os.environ['HOME'])

#用户盘符
# print(os.environ['PWD'])

#window根目录
# print(os.environ['SYSTEMROOT'])

#路径拼接
print(os.path.join('d://','music'))

#当前文件读取指针
open('111.txt').tell()

#移动指针
open('111.txt').seek(0,1)

#当前文件目录
a = os.getcwd()
c = os.environ['PWD']
# print(a,b,c)

#当前文件路径
os.path.realpath(__file__)

#当前文件名
__file__

#创建文件夹
os.mkdir()

#重命名文件
os.rename()


w     以写方式打开，
a     以追加模式打开
r+     以读写模式打开
w+     以读写模式打开
a+     以读写模式打开
rb     以二进制读模式打开
wb     以二进制写模式打开
ab     以二进制追加模式打开
rb+    以二进制读写模式打开
wb+    以二进制读写模式打开
ab+    以二进制读写模式打开


def fillfile():
  global new_folders,loopText,loop,os
  os.mkdir(new_folders)
  root = os.getcwd()
  directory = root + os.sep + new_folders + os.sep
  for i in range(0,100):
    open(directory + repr(i) + '.txt','w')

  return directory

#输出指定目录下的文件
def outputDirs():
  dir = input("请输入需要输入的目录:\n")
  if(os.path.isdir(dir)):
    dirs = os.listdir(dir)
    getDirs(dirs)
  else:
    print("未找到此目录")


def getDirs(dirs):
  for i,e in enumerate(dirs):
    print("No.%d\tname:%s"%(i,e))

#搜索文件
def searchFile():
  file_name = input("请输入要搜索的文件名:\n")
  if os.path.isfile(file_name):
    print("您搜索的文件路径位于:%s"%(file_name))

#搜索文件是否存在
def searchFileStatus(file_name):
  if os.path.isfile(file_name):
    return True
  else:
    return False

#读取文件
def readFile():
      file_name = input("请输入要读取的文件名:")
      if searchFileStatus(file_name):
            while True:
              open_file = open(file_name,'r')
              content = open_file.readlines()
              if len(content) == 0:
                    break
              print(content)
            

def searchFileForRename():
  file_name = input("请输入要查找的文件名:\t(dirname+basebame)\n")
  if os.path.isfile(file_name):
    new_file_name = input("要重命名的文件名:\t(basebame)\n")
    return (file_name,new_file_name)
  else:
    print("未找到此文件")

#重命名文件
def rename():
  (a,b) = searchFileForRename()
  realDirPath = os.path.dirname(a)
  os.rename(a,realDirPath + os.sep + b)
  print("旧文件名:%s\t新文件名:%s"%(a,b))

#写入文件
def writeFile():
  file_name = input("请输入要写入的文件名:\n")
  if searchFileStatus(file_name):
    file = open(file_name,'w')
    content = input("请输入要写入的内容:\n")
    file.write(content)
    print("写入完成")




#GUI
def dashboard():
  mode = input(
    "="*20
    +"\n文件操作小工具"
    +"\n1.搜索文件\n2.重命名文件\n3.列出指定目录下的文件\nt4.读取文件\n5.写入文件\n"+
    "="*20 + "\n"
    )

  if int(mode) == 1:
    searchFile()
  elif int(mode) == 2:
    rename()
  elif int(mode) == 3:
    outputDirs()
  elif int(mode) == 4:
    readFile()
  elif int(mode) == 5:
    writeFile()


# dashboard()

=============面向对象版本================
import os

class FileSystem:
    def __init__(self):
        self.mode = 0
        self.modeDes = ['读取文件','创建文件','列出指定目录下的文件','删除文件','重命名文件']
        self.selectMode()
        self.callMethod()
        
    def selectMode(self):
        print("请输入要操作的方法:\n")
        print("="*20)
        for i,e in enumerate(self.modeDes):
            print("%d.%s"%(i+1,e))
        print("="*20)
        mode = int(input())
        self.mode = mode

    def callMethod(self):
        if self.mode == 1:
            self.readFile()
        elif self.mode == 2:
            self.mkFile()
        elif self.mode == 3:
            self.outputDirs()
        elif self.mode == 4:
            self.deleteFile()
        elif self.mode == 5:
            self.renameFile()

    def readFile(self):
        file_name = self.getPrint("请输入要读取的文件:\n")
        if self.fileExist(file_name):
            open_file = open(file_name,'r')
            while True:
                content = open_file.readline()
                if len(content) == 0:
                    break
                print(content)

    def mkFile(self):
        print("请输入要创建的文件:")
        file_name = input()
        if ~self.fileExist(file_name):
            open(file_name,'w')
            print("创建成功")
        else:
            print("文件已存在,请重命名文件")

    def outputDirs(self):
        dir = self.getPrint("请输入指定目录下的文件")
        if self.dirExist(dir):
            files = os.listdir(dir)
            for file in files:
                print(file)
    
    def deleteFile(self):
        self.getPrint("请输入要删除的文件:\n")
        if self.fileExist(file_name):
            os.remove(file_name)
            print("删除成功")
        else:
            print("文件不存在")

    def renameFile(self):
        file_name = self.getPrint("请输入要重命名的文件:\n")
        if self.fileExist(file_name):
            new_file_name = input("请输入新的文件名:")
            if self.fileExist(new_file_name):
                print("文件名已经存在,请重新输入")
            else:
                os.rename(file_name,new_file_name)
                print("重命名文件名成功")
            
    def fileExist(self,file):
        if os.path.isfile(file):
            return True
        else:
            return False

    def dirExist(self,dir):
        if os.path.isdir(dir):
            return True
        else:
            return False
    
    def getPrint(self,str):
        print(str)
        file_name = input()
        return file_name
        
    
FileSystem()