brotherton-erpnext/patches/april_2013/p06_update_file_size.py

# Copyright (c) 2013, Web Notes Technologies Pvt. Ltd.
# License: GNU General Public License v3. See license.txt

import webnotes, os, webnotes.utils

def execute():
	files_path = webnotes.utils.get_path("public", "files")
	webnotes.conn.auto_commit_on_many_writes = 1

	for f in webnotes.conn.sql("""select name, file_name from 
		`tabFile Data`""", as_dict=True):
		if f.file_name:
			filepath = os.path.join(files_path, f.file_name)
			if os.path.exists(filepath):
				webnotes.conn.set_value("File Data", f.name, "file_size", os.stat(filepath).st_size)
				
	webnotes.conn.auto_commit_on_many_writes = 0