您好，针对您提出的“工伤后离职，学校非要写离职原因是因工伤离职才能开离职证明，这样是否合法”的问题，我可以为您详细解答。

首先，我们需要明确离职证明的法律性质和功能。根据《中华人民共和国劳动合同法》第50条的规定，用人单位在解除或终止劳动合同时，应当出具解除或终止劳动合同的证明，即我们通常所说的离职证明。离职证明的主要功能有两个：一是证明劳动者已经与原用人单位终结劳动关系以及原劳动合同的基本信息，为其再就业提供帮助；二是劳动者在申请失业保险待遇时，用来证明自己处于失业状态。

接下来，我们分析离职证明的内容要求。根据《中华人民共和国劳动合同法实施条例》第二十四条的规定，用人单位出具的解除、终止劳动合同的证明，应当写明劳动合同期限、解除或者终止劳动合同的日期、工作岗位、在本单位的工作年限。这意味着，离职证明应当仅包含与劳动合同履行、终结相关的基本信息，而不应包含其他主观评价或额外条件。

现在，我们回到您的问题上。学校要求离职证明中必须写明离职原因是因工伤离职，这一要求并不符合离职证明的法定内容要求。离职证明应当客观、中立地反映劳动合同的履行和终结情况，而不应涉及离职的具体原因，尤其是当这种原因可能涉及劳动者隐私或存在争议时。

此外，学校以必须写明离职原因为条件才出具离职证明的做法，也可能构成对劳动者权益的侵害。因为出具离职证明是用人单位的法定义务，不得附加任何条件。如果学校因此拒绝出具离职证明，给劳动者造成损失的，还应当承担相应的赔偿责任。

综上所述，学校要求离职证明中必须写明离职原因是因工伤离职的做法是不合法的。您可以与学校进行沟通协商，要求其依法出具离职证明。如果学校拒绝改正，您可以向劳动监察部门投诉或寻求法律援助来维护自己的合法权益。希望这个解答能够对您有所帮助。