PY-16577 Project Structure settings don't work for IPython Notebook

python/ipnb/src/org/jetbrains/plugins/ipnb/editor/panels/IpnbFilePanel.java

@@ -604,4 +604,8 @@ public class IpnbFilePanel extends JPanel implements Scrollable, DataProvider, D
   public VirtualFile getVirtualFile() {
     return myVirtualFile;
   }
+
+  public Document getDocument() {
+    return myDocument;
+  }
 }

python/ipnb/src/org/jetbrains/plugins/ipnb/psi/IpnbPyFragment.java

@@ -72,6 +72,13 @@ public class IpnbPyFragment extends PyFileImpl {
     return getManager().findDirectory(parentFile);
   }
 
+  @NotNull
+  @Override
+  public PsiFile getOriginalFile() {
+    final PsiFile psiFile = PsiDocumentManager.getInstance(getProject()).getPsiFile(myFilePanel.getDocument());
+    return psiFile != null ? psiFile : super.getOriginalFile();
+  }
+
   @Override
   public PsiElement getNextSibling() {
     return null;