From 570239cd48b32bb6e0042e6f3471200ae05df6d2 Mon Sep 17 00:00:00 2001
From: Suraj Shetty <surajshetty3416@gmail.com>
Date: Fri, 26 Jul 2019 10:51:06 +0530
Subject: [PATCH] fix: Patch to create file in right location

---
 frappe/core/doctype/file/file.py              | 15 +++++----
 .../patches/v12_0/fix_public_private_files.py | 33 +++++++++++++++++++
 2 files changed, 42 insertions(+), 6 deletions(-)
 create mode 100644 frappe/patches/v12_0/fix_public_private_files.py

diff --git a/frappe/core/doctype/file/file.py b/frappe/core/doctype/file/file.py
index 6e6cce52bf..8fac317100 100755
--- a/frappe/core/doctype/file/file.py
+++ b/frappe/core/doctype/file/file.py
@@ -473,7 +473,7 @@ class File(NestedSet):
 		return None
 
 
-	def save_file(self, content=None, decode=False):
+	def save_file(self, content=None, decode=False, ignore_existing_file_check=False):
 		file_exists = False
 		self.content = content
 		if decode:
@@ -490,12 +490,15 @@ class File(NestedSet):
 		self.content_hash = get_content_hash(self.content)
 		self.content_type = mimetypes.guess_type(self.file_name)[0]
 
+		_file = False
+
 		# check if a file exists with the same content hash and is also in the same folder (public or private)
-		_file = frappe.get_value("File", {
-				"content_hash": self.content_hash,
-				"is_private": self.is_private
-			},
-			["file_url"])
+		if not ignore_existing_file_check:
+			_file = frappe.get_value("File", {
+					"content_hash": self.content_hash,
+					"is_private": self.is_private
+				},
+				["file_url"])
 
 		if _file:
 			self.file_url  = _file
diff --git a/frappe/patches/v12_0/fix_public_private_files.py b/frappe/patches/v12_0/fix_public_private_files.py
new file mode 100644
index 0000000000..001f9a2d2e
--- /dev/null
+++ b/frappe/patches/v12_0/fix_public_private_files.py
@@ -0,0 +1,33 @@
+import frappe
+import os
+
+def execute():
+	files = frappe.get_all('File',
+		fields=['is_private', 'file_url', 'name'],
+		filters={'is_folder': 0})
+
+	for file in files:
+		file_url = file.file_url
+		if file.is_private:
+			if not file_url.startswith('/private/files/'):
+				generate_file(file.name)
+		else:
+			if file_url.startswith('/private/files/'):
+				generate_file(file.name)
+
+def generate_file(file_name):
+	try:
+		file_doc = frappe.get_doc('File', file_name)
+		# private
+		new_doc = frappe.new_doc('File')
+		new_doc.is_private = file_doc.is_private
+		new_doc.file_name = file_doc.file_name
+		# to create copy of file in right location
+		# if the file doc is private file will be created in /private folder
+		# if the file doc is public file will be created in /files folder
+		new_doc.save_file(content=file_doc.get_content(), ignore_existing_file_check=True)
+
+		file_doc.file_url = new_doc.file_url
+		file_doc.save()
+	except IOError:
+		pass
\ No newline at end of file