ตรวจสอบให้แน่ใจว่าไม่ได้ใช้รหัสที่ไม่ปลอดภัย

ฟังก์ชั่นf()ใช้eval()(หรือบางสิ่งที่อันตราย) กับข้อมูลที่ฉันสร้างและเก็บไว้ในlocal_fileเครื่องที่ใช้โปรแกรมของฉัน:

import local_file

def f(str_to_eval):
    # code....
    # .... 
    eval(str_to_eval)    
    # ....
    # ....    
    return None


a = f(local_file.some_str)

f() ปลอดภัยที่จะเรียกใช้เนื่องจากสตริงที่ฉันกำหนดให้เป็นของฉันเอง

แต่ถ้าฉันเคยตัดสินใจที่จะใช้มันสำหรับบางสิ่งบางอย่างที่ไม่ปลอดภัย (เช่นผู้ใช้ป้อนข้อมูล) สิ่งที่จะไปผิดมหันต์ นอกจากนี้หากlocal_fileหยุดเป็นท้องถิ่นแล้วมันจะสร้างช่องโหว่เนื่องจากฉันจะต้องเชื่อถือเครื่องที่ให้ไฟล์นั้นเช่นกัน

ฉันจะแน่ใจได้อย่างไรว่าฉันไม่เคย "ลืม" ว่าฟังก์ชั่นนี้ไม่ปลอดภัยที่จะใช้ (เว้นแต่ว่าตรงตามเกณฑ์ที่กำหนด)

_{หมายเหตุ: eval()มีอันตรายและมักจะถูกแทนที่ด้วยสิ่งที่ปลอดภัย}

กำหนดประเภทเพื่อตกแต่งอินพุตที่ "ปลอดภัย" ในฟังก์ชั่นของคุณf()ยืนยันว่าอาร์กิวเมนต์มีประเภทนี้หรือโยนข้อผิดพลาด def g(x): return f(SafeInput(x))สมาร์ทจะเพียงพอที่จะไม่กำหนดทางลัด

ตัวอย่าง:

class SafeInput(object):
  def __init__(self, value):
    self._value = value

  def value(self):
    return self._value

def f(safe_string_to_eval):
  assert type(safe_string_to_eval) is SafeInput, "safe_string_to_eval must have type SafeInput, but was {}".format(type(safe_string_to_eval))
  ...
  eval(safe_string_to_eval.value())
  ...

a = f(SafeInput(local_file.some_str))

ในขณะที่สิ่งนี้สามารถหลีกเลี่ยงได้ง่ายมันทำให้ยากขึ้นโดยบังเอิญ ผู้คนอาจวิจารณ์การตรวจสอบแบบเข้มงวดเช่นนี้ แต่พวกเขาก็พลาดประเด็นนี้ไป ตั้งแต่SafeInputประเภทเป็นเพียงประเภทข้อมูลและไม่ได้เป็นระดับเชิงวัตถุที่สามารถ subclassed ตรวจสอบตัวตนสำหรับประเภทที่มีความปลอดภัยมากขึ้นกว่าการตรวจสอบสำหรับประเภทเข้ากันได้กับtype(x) is SafeInput isinstance(x, SafeInput)เนื่องจากเราต้องการบังคับใช้เฉพาะบางประเภทการเพิกเฉยต่อประเภทที่ชัดเจนและการพิมพ์เป็ดโดยปริยายก็ไม่เป็นที่น่าพอใจเช่นกัน Python มีระบบการพิมพ์ดังนั้นลองใช้มันเพื่อตรวจจับข้อผิดพลาดที่อาจเกิดขึ้นได้!

เมื่อโค้ดของ Joel ชี้ให้ทำรหัสผิดให้ดูผิด (เลื่อนลงไปที่หัวข้อ "ทางออกที่แท้จริง" หรืออ่านได้ดีขึ้นอย่างสมบูรณ์มันก็คุ้มค่าแม้ว่ามันจะอยู่ที่ตัวแปรแทนที่จะเป็นชื่อฟังก์ชั่น) ดังนั้นฉันแนะนำอย่างนอบน้อมให้เปลี่ยนชื่อฟังก์ชั่นของคุณเป็นอย่างอื่นf_unsafe_for_external_use()- คุณอาจจะคิดแบบย่อ ๆ ด้วยตัวเอง

แก้ไข: ฉันคิดว่าข้อเสนอแนะของ amon เป็นรูปแบบที่ดีมาก OO ในรูปแบบเดียวกัน :-)

การทำตามคำแนะนำโดย @amon คุณสามารถคิดถึงการรวมรูปแบบกลยุทธ์ที่นี่รวมถึงรูปแบบวัตถุวิธีการ

class AbstractFoobarizer(object):
    def handle_cond(self, foo, bar):
        """
        Handle the event or something.
        """
        raise NotImplementedError

    def run(self):
        # do some magic
        pass

และจากนั้นการดำเนินการ 'ปกติ'

class PrintingFoobarizer(AbstractFoobarizer):
    def handle_cond(self, foo, bar):
        print("Something went very well regarding {} and {}".format(foo, bar))

และการใช้งานของผู้ดูแลระบบอินพุต eval

class AdminControllableFoobarizer(AbstractFoobarizer):
    def handle_cond(self, foo, bar):
        # Look up code in database/config file/...
        code = get_code_from_settings()
        eval(code)

(หมายเหตุ: ด้วยตัวอย่างของโลกแห่งความจริงฉันอาจให้ตัวอย่างที่ดีกว่า)