สร้างไฟล์อย่างปลอดภัยถ้าไม่มีอยู่ใน python

ฉันต้องการเขียนลงในไฟล์โดยขึ้นอยู่กับว่าไฟล์นั้นมีอยู่แล้วหรือไม่เพียงเขียนเมื่อไม่มีอยู่แล้วเท่านั้น (ในทางปฏิบัติฉันต้องการลองไฟล์ต่อไปจนกว่าฉันจะพบไฟล์ที่ไม่มีอยู่)

โค้ดต่อไปนี้แสดงวิธีที่ผู้โจมตีอาจแทรก symlink ตามที่แนะนำในโพสต์นี้ระหว่างการทดสอบไฟล์และไฟล์ที่กำลังเขียน หากรหัสถูกเรียกใช้โดยมีสิทธิ์สูงเพียงพอสิ่งนี้อาจเขียนทับไฟล์โดยพลการ

มีวิธีใดในการแก้ปัญหานี้หรือไม่?

import os
import errno

file_to_be_attacked = 'important_file'

with open(file_to_be_attacked, 'w') as f:
    f.write('Some important content!\n')

test_file = 'testfile'

try:
    with open(test_file) as f: pass
except IOError, e:

    # symlink created here
    os.symlink(file_to_be_attacked, test_file)

    if e.errno != errno.ENOENT:
        raise
    else:
        with open(test_file, 'w') as f:
            f.write('Hello, kthxbye!\n')

แก้ไข : ดูคำตอบของ Dave Jones ด้วย : จาก Python 3.3 คุณสามารถใช้xแฟล็กopen()เพื่อจัดเตรียมฟังก์ชันนี้

คำตอบเดิมด้านล่าง

ใช่ แต่ไม่ได้ใช้การopen()เรียกมาตรฐานของ Python คุณจะต้องใช้os.open()แทนซึ่งช่วยให้คุณระบุแฟล็กให้กับรหัส C ที่อยู่ข้างใต้

O_CREAT | O_EXCLโดยเฉพาะอย่างยิ่งที่คุณต้องการใช้ จากหน้าคนสำหรับopen(2)ภายใต้O_EXCLระบบ Unix ของฉัน:

ตรวจสอบให้แน่ใจว่าการเรียกนี้สร้างไฟล์: หากระบุแฟล็กนี้ร่วมกับO_CREATและมีชื่อพา ธ อยู่แล้วopen()จะล้มเหลว ลักษณะการทำงานของO_EXCLis undefined ถ้าO_CREATไม่ได้ระบุ

เมื่อระบุแฟล็กทั้งสองนี้จะไม่มีการติดตามลิงก์สัญลักษณ์: หากชื่อพา ธ เป็นลิงก์สัญลักษณ์จะopen()ล้มเหลวไม่ว่าลิงก์สัญลักษณ์จะชี้ไปที่ใด

O_EXCL ได้รับการสนับสนุนบน NFS เท่านั้นเมื่อใช้ NFSv3 หรือใหม่กว่าบนเคอร์เนล 2.6 หรือใหม่กว่า ในสภาพแวดล้อมที่ไม่มีO_EXCLการสนับสนุนNFS โปรแกรมที่ใช้ในการดำเนินการล็อกจะมีเงื่อนไขการแข่งขัน

ดังนั้นมันจึงไม่สมบูรณ์แบบ แต่ AFAIK เป็นสิ่งที่ใกล้เคียงที่สุดที่คุณจะหลีกเลี่ยงสภาพการแข่งขันนี้ได้

แก้ไข: กฎอื่น ๆ ในการใช้os.open()แทนที่จะopen()ยังคงมีผลบังคับใช้ โดยเฉพาะอย่างยิ่งถ้าคุณต้องการใช้อธิบายไฟล์กลับมาอ่านหรือเขียนคุณจะต้องเป็นหนึ่งในO_RDONLY, O_WRONLYหรือO_RDWRธงเช่นกัน

O_*แฟล็กทั้งหมดอยู่ในosโมดูลของ Python ดังนั้นคุณจะต้องimport osใช้และos.O_CREATอื่น ๆ

ตัวอย่าง:

import os
import errno

flags = os.O_CREAT | os.O_EXCL | os.O_WRONLY

try:
    file_handle = os.open('filename', flags)
except OSError as e:
    if e.errno == errno.EEXIST:  # Failed as the file already exists.
        pass
    else:  # Something unexpected went wrong so reraise the exception.
        raise
else:  # No exception, so the file must have been created successfully.
    with os.fdopen(file_handle, 'w') as file_obj:
        # Using `os.fdopen` converts the handle to an object that acts like a
        # regular Python file object, and the `with` context manager means the
        # file will be automatically closed when we're done with it.
        file_obj.write("Look, ma, I'm writing to a new file!")

สำหรับการอ้างอิง Python 3.3 จะใช้'x'โหมดใหม่ในopen()ฟังก์ชันเพื่อให้ครอบคลุมกรณีการใช้งานนี้ (สร้างเท่านั้นล้มเหลวหากมีไฟล์อยู่) โปรดทราบว่า'x'โหมดถูกระบุด้วยตัวมันเอง การใช้'wx'ผลลัพธ์ในValueErroras 'w'ซ้ำซ้อน (สิ่งเดียวที่คุณสามารถทำได้หากการโทรประสบความสำเร็จคือการเขียนลงไฟล์อย่างไรก็ตามจะไม่สามารถมีอยู่ได้หากการโทรสำเร็จ):

>>> f1 = open('new_binary_file', 'xb')
>>> f2 = open('new_text_file', 'x')

สำหรับงูหลาม 3.2 และด้านล่าง (รวมทั้งงูหลาม 2.x) โปรดดูคำตอบที่ได้รับการยอมรับ

รหัสนี้จะสร้าง FILE ได้อย่างง่ายดายหากไม่มีอยู่

import os
if not os.path.exists('file'):
    open('file', 'w').close()