anoopmb
/
frappe
1
0
Fork 0
Koodi Ongelmat 0 Pull-pyynnöt 0 Julkaisut 0 Wiki Toiminta
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
252 Commitit
1 Haara
314 MiB
 
 
 
 
 
 
Puu: 98acabbb96
Branchit Tagit
${ item.name }
Create branch ${ searchTerm }
from '98acabbb96'
${ noResults }
frappe/docs/_build/html/_modules/webnotes/handler.html

533 rivejä
65 KiB
Raaka Blame Historia 

  1. 

  2. 

  3. <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN"

  4.   "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">

  5. 

  6. <html xmlns="http://www.w3.org/1999/xhtml">

  7.   <head>

  8.     <meta http-equiv="Content-Type" content="text/html; charset=utf-8" />

  9.     

  10.     <title>webnotes.handler &mdash; WNFramework v1.8 documentation</title>

  11.     <link rel="stylesheet" href="../../_static/default.css" type="text/css" />

  12.     <link rel="stylesheet" href="../../_static/pygments.css" type="text/css" />

  13.     <script type="text/javascript">

  14.       var DOCUMENTATION_OPTIONS = {

  15.         URL_ROOT:    '../../',

  16.         VERSION:     '1.8',

  17.         COLLAPSE_INDEX: false,

  18.         FILE_SUFFIX: '.html',

  19.         HAS_SOURCE:  true

  20.       };

  21.     </script>

  22.     <script type="text/javascript" src="../../_static/jquery.js"></script>

  23.     <script type="text/javascript" src="../../_static/underscore.js"></script>

  24.     <script type="text/javascript" src="../../_static/doctools.js"></script>

  25.     <link rel="top" title="WNFramework v1.8 documentation" href="../../index.html" />

  26.     <link rel="up" title="webnotes" href="../webnotes.html" /> 

  27.   </head>

  28.   <body>

  29.     <div class="related">

  30.       <h3>Navigation</h3>

  31.       <ul>

  32.         <li class="right" style="margin-right: 10px">

  33.           <a href="../../genindex.html" title="General Index"

  34.              accesskey="I">index</a></li>

  35.         <li class="right" >

  36.           <a href="../../py-modindex.html" title="Python Module Index"

  37.              >modules</a> |</li>

  38.         <li><a href="../../index.html">WNFramework v1.8 documentation</a> &raquo;</li>

  39.           <li><a href="../index.html" >Module code</a> &raquo;</li>

  40.           <li><a href="../webnotes.html" accesskey="U">webnotes</a> &raquo;</li> 

  41.       </ul>

  42.     </div>  

  43. 

  44.     <div class="document">

  45.       <div class="documentwrapper">

  46.         <div class="bodywrapper">

  47.           <div class="body">

  48.             

  49.   <h1>Source code for webnotes.handler</h1><div class="highlight"><pre>

  50. <span class="kn">import</span> <span class="nn">sys</span><span class="o">,</span> <span class="nn">os</span>

  51. <span class="kn">import</span> <span class="nn">webnotes</span>

  52. <span class="kn">import</span> <span class="nn">webnotes.defs</span>

  53. <span class="kn">import</span> <span class="nn">webnotes.utils</span>

  54. 

  55. <span class="n">form</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">form</span>

  56. <span class="n">form_dict</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">form_dict</span>

  57. 

  58. <span class="n">sql</span> <span class="o">=</span> <span class="bp">None</span>

  59. <span class="n">session</span> <span class="o">=</span> <span class="bp">None</span>

  60. <span class="n">errdoc</span> <span class="o">=</span> <span class="s">&#39;&#39;</span>

  61. <span class="n">errdoctype</span> <span class="o">=</span> <span class="s">&#39;&#39;</span>

  62. <span class="n">errmethod</span> <span class="o">=</span> <span class="s">&#39;&#39;</span>

  63. <span class="n">fw_folder</span> <span class="o">=</span> <span class="s">&#39;/Users/rushabh/workbench/www/&#39;</span>

  64. 

  65. 

  66. <span class="c"># Logs</span>

  67. 

  68. <span class="c"># refresh / start page</span>

  69. <span class="c"># ------------------------------------------------------------------------------------</span>

  70. 

  71. <div class="viewcode-block" id="startup"><a class="viewcode-back" href="../../webnotes.html#webnotes.handler.startup">[docs]</a><span class="k">def</span> <span class="nf">startup</span><span class="p">():</span>

  72. 	<span class="kn">import</span> <span class="nn">webnotes</span>

  73. 	<span class="kn">import</span> <span class="nn">webnotes.session_cache</span>

  74. 

  75. 	<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="o">.</span><span class="n">update</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">session_cache</span><span class="o">.</span><span class="n">get</span><span class="p">())</span>

  76. </div>

  77. <div class="viewcode-block" id="cleanup_docs"><a class="viewcode-back" href="../../webnotes.html#webnotes.handler.cleanup_docs">[docs]</a><span class="k">def</span> <span class="nf">cleanup_docs</span><span class="p">():</span>

  78. 	<span class="kn">import</span> <span class="nn">webnotes.model.doclist</span>

  79. 	<span class="k">if</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;docs&#39;</span><span class="p">)</span> <span class="ow">and</span> <span class="nb">type</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;docs&#39;</span><span class="p">])</span><span class="o">!=</span><span class="nb">dict</span><span class="p">:</span>

  80. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;docs&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">model</span><span class="o">.</span><span class="n">doclist</span><span class="o">.</span><span class="n">compress</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;docs&#39;</span><span class="p">])</span>

  81. 

  82. <span class="c"># server calls</span>

  83. <span class="c"># ------------------------------------------------------------------------------------</span>

  84. </div>

  85. <div class="viewcode-block" id="runserverobj"><a class="viewcode-back" href="../../webnotes.html#webnotes.handler.runserverobj">[docs]</a><span class="k">def</span> <span class="nf">runserverobj</span><span class="p">(</span><span class="n">arg</span><span class="o">=</span><span class="bp">None</span><span class="p">):</span>

  86. 	<span class="kn">import</span> <span class="nn">webnotes.widgets.form</span>

  87. 	<span class="n">webnotes</span><span class="o">.</span><span class="n">widgets</span><span class="o">.</span><span class="n">form</span><span class="o">.</span><span class="n">runserverobj</span><span class="p">()</span>

  88. </div>

  89. <div class="viewcode-block" id="logout"><a class="viewcode-back" href="../../webnotes.html#webnotes.handler.logout">[docs]</a><span class="k">def</span> <span class="nf">logout</span><span class="p">():</span>

  90. 	<span class="n">webnotes</span><span class="o">.</span><span class="n">login_manager</span><span class="o">.</span><span class="n">logout</span><span class="p">()</span>

  91. 

  92. <span class="c"># DocType Mapper</span>

  93. <span class="c"># ------------------------------------------------------------------------------------</span>

  94. </div>

  95. <div class="viewcode-block" id="dt_map"><a class="viewcode-back" href="../../webnotes.html#webnotes.handler.dt_map">[docs]</a><span class="k">def</span> <span class="nf">dt_map</span><span class="p">():</span>

  96. 	<span class="kn">import</span> <span class="nn">webnotes</span>

  97. 	<span class="kn">import</span> <span class="nn">webnotes.model.doclist</span>

  98. 	<span class="kn">from</span> <span class="nn">webnotes.model.code</span> <span class="kn">import</span> <span class="n">get_obj</span>

  99. 	<span class="kn">from</span> <span class="nn">webnotes.model.doc</span> <span class="kn">import</span> <span class="n">Document</span>

  100. 	

  101. 	<span class="n">form_dict</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">form_dict</span>

  102. 	

  103. 	<span class="n">dt_list</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">model</span><span class="o">.</span><span class="n">doclist</span><span class="o">.</span><span class="n">expand</span><span class="p">(</span><span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;docs&#39;</span><span class="p">))</span>

  104. 	<span class="n">from_doctype</span> <span class="o">=</span> <span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;from_doctype&#39;</span><span class="p">)</span>

  105. 	<span class="n">to_doctype</span> <span class="o">=</span> <span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;to_doctype&#39;</span><span class="p">)</span>

  106. 	<span class="n">from_docname</span> <span class="o">=</span> <span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;from_docname&#39;</span><span class="p">)</span>

  107. 	<span class="n">from_to_list</span> <span class="o">=</span> <span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;from_to_list&#39;</span><span class="p">)</span>

  108. 	

  109. 	<span class="n">dm</span> <span class="o">=</span> <span class="n">get_obj</span><span class="p">(</span><span class="s">&#39;DocType Mapper&#39;</span><span class="p">,</span> <span class="n">from_doctype</span> <span class="o">+</span><span class="s">&#39;-&#39;</span> <span class="o">+</span> <span class="n">to_doctype</span><span class="p">)</span>

  110. 	<span class="n">doclist</span> <span class="o">=</span> <span class="n">dm</span><span class="o">.</span><span class="n">dt_map</span><span class="p">(</span><span class="n">from_doctype</span><span class="p">,</span> <span class="n">to_doctype</span><span class="p">,</span> <span class="n">from_docname</span><span class="p">,</span> <span class="n">Document</span><span class="p">(</span><span class="n">fielddata</span> <span class="o">=</span> <span class="n">dt_list</span><span class="p">[</span><span class="mi">0</span><span class="p">]),</span> <span class="p">[],</span> <span class="n">from_to_list</span><span class="p">)</span>

  111. 	

  112. 	<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;docs&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">doclist</span>

  113. 

  114. <span class="c"># Load Month Events</span>

  115. <span class="c"># ------------------------------------------------------------------------------------</span>

  116. </div>

  117. <div class="viewcode-block" id="load_month_events"><a class="viewcode-back" href="../../webnotes.html#webnotes.handler.load_month_events">[docs]</a><span class="k">def</span> <span class="nf">load_month_events</span><span class="p">():</span>

  118. 	<span class="kn">import</span> <span class="nn">webnotes</span>

  119. 	<span class="n">form</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">form</span>

  120. 

  121. 	<span class="n">mm</span> <span class="o">=</span> <span class="n">form</span><span class="o">.</span><span class="n">getvalue</span><span class="p">(</span><span class="s">&#39;month&#39;</span><span class="p">)</span>

  122. 	<span class="n">yy</span> <span class="o">=</span> <span class="n">form</span><span class="o">.</span><span class="n">getvalue</span><span class="p">(</span><span class="s">&#39;year&#39;</span><span class="p">)</span>

  123. 	<span class="n">m_st</span> <span class="o">=</span> <span class="nb">str</span><span class="p">(</span><span class="n">yy</span><span class="p">)</span> <span class="o">+</span> <span class="s">&#39;-&#39;</span> <span class="o">+</span> <span class="nb">str</span><span class="p">(</span><span class="n">mm</span><span class="p">)</span> <span class="o">+</span> <span class="s">&#39;-01&#39;</span>

  124. 	<span class="n">m_end</span> <span class="o">=</span> <span class="nb">str</span><span class="p">(</span><span class="n">yy</span><span class="p">)</span> <span class="o">+</span> <span class="s">&#39;-&#39;</span> <span class="o">+</span> <span class="nb">str</span><span class="p">(</span><span class="n">mm</span><span class="p">)</span> <span class="o">+</span> <span class="s">&#39;-31&#39;</span>

  125. 

  126. 	<span class="kn">import</span> <span class="nn">webnotes.widgets.event</span>

  127. 	<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;docs&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">widgets</span><span class="o">.</span><span class="n">event</span><span class="o">.</span><span class="n">get_cal_events</span><span class="p">(</span><span class="n">m_st</span><span class="p">,</span> <span class="n">m_end</span><span class="p">)</span>

  128. 

  129. <span class="c"># Data import</span>

  130. <span class="c"># ------------------------------------------------------------------------------------</span>

  131. </div>

  132. <div class="viewcode-block" id="import_csv"><a class="viewcode-back" href="../../webnotes.html#webnotes.handler.import_csv">[docs]</a><span class="k">def</span> <span class="nf">import_csv</span><span class="p">():</span>

  133. 	<span class="kn">import</span> <span class="nn">webnotes.model.import_docs</span>

  134. 	<span class="n">form</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">form</span>

  135. 	<span class="kn">from</span> <span class="nn">webnotes.utils</span> <span class="kn">import</span> <span class="n">cint</span>

  136. 	

  137. 	<span class="n">i</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">model</span><span class="o">.</span><span class="n">import_docs</span><span class="o">.</span><span class="n">CSVImport</span><span class="p">()</span>

  138. 	<span class="n">r</span> <span class="o">=</span> <span class="n">i</span><span class="o">.</span><span class="n">import_csv</span><span class="p">(</span><span class="n">form</span><span class="o">.</span><span class="n">getvalue</span><span class="p">(</span><span class="s">&#39;csv_file&#39;</span><span class="p">),</span> <span class="n">form</span><span class="o">.</span><span class="n">getvalue</span><span class="p">(</span><span class="s">&#39;dateformat&#39;</span><span class="p">),</span> <span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;overwrite&#39;</span><span class="p">,</span> <span class="mi">0</span><span class="p">)</span> <span class="ow">and</span> <span class="mi">1</span><span class="p">)</span>

  139. 	

  140. 	<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;type&#39;</span><span class="p">]</span><span class="o">=</span><span class="s">&#39;iframe&#39;</span>

  141. 	<span class="n">rhead</span> <span class="o">=</span> <span class="s">&#39;&#39;&#39;&lt;style&gt;body, html {font-family: Arial; font-size: 12px;}&lt;/style&gt;&#39;&#39;&#39;</span>

  142. 	<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;result&#39;</span><span class="p">]</span><span class="o">=</span> <span class="n">rhead</span> <span class="o">+</span> <span class="n">r</span>

  143. </div>

  144. <div class="viewcode-block" id="get_template"><a class="viewcode-back" href="../../webnotes.html#webnotes.handler.get_template">[docs]</a><span class="k">def</span> <span class="nf">get_template</span><span class="p">():</span>

  145. 	<span class="kn">import</span> <span class="nn">webnotes.model.import_docs</span>

  146. 	<span class="n">webnotes</span><span class="o">.</span><span class="n">model</span><span class="o">.</span><span class="n">import_docs</span><span class="o">.</span><span class="n">get_template</span><span class="p">()</span>

  147. 	

  148. 

  149. <span class="c"># File Upload</span>

  150. <span class="c"># ------------------------------------------------------------------------------------</span>

  151. </div>

  152. <div class="viewcode-block" id="uploadfile"><a class="viewcode-back" href="../../webnotes.html#webnotes.handler.uploadfile">[docs]</a><span class="k">def</span> <span class="nf">uploadfile</span><span class="p">():</span>

  153. 	<span class="kn">import</span> <span class="nn">webnotes.utils.file_manager</span>

  154. 	<span class="k">if</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;from_form&#39;</span><span class="p">):</span>

  155. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">utils</span><span class="o">.</span><span class="n">file_manager</span><span class="o">.</span><span class="n">upload</span><span class="p">()</span>

  156. 	<span class="k">else</span><span class="p">:</span>

  157. 		<span class="c"># save the file</span>

  158. 		<span class="n">fid</span><span class="p">,</span> <span class="n">fname</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">utils</span><span class="o">.</span><span class="n">file_manager</span><span class="o">.</span><span class="n">save_uploaded</span><span class="p">()</span>

  159. 		

  160. 		<span class="c"># do something with the uploaded file</span>

  161. 		<span class="k">if</span> <span class="n">fid</span> <span class="ow">and</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;server_obj&#39;</span><span class="p">):</span>

  162. 			<span class="kn">from</span> <span class="nn">webnotes.model.code</span> <span class="kn">import</span> <span class="n">get_obj</span>

  163. 			<span class="nb">getattr</span><span class="p">(</span><span class="n">get_obj</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;server_obj&#39;</span><span class="p">)),</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;method&#39;</span><span class="p">))(</span><span class="n">fid</span><span class="p">,</span> <span class="n">fname</span><span class="p">)</span>

  164. 			

  165. 		<span class="c"># return the upload</span>

  166. 		<span class="k">if</span> <span class="n">fid</span><span class="p">:</span>

  167. 			<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;result&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="s">&#39;&lt;script&gt;window.parent.upload_callback(&quot;&#39;</span><span class="o">+</span><span class="n">webnotes</span><span class="o">.</span><span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;uploader_id&#39;</span><span class="p">)</span><span class="o">+</span><span class="s">&#39;&quot;, &quot;&#39;</span><span class="o">+</span><span class="n">fid</span><span class="o">+</span><span class="s">&#39;&quot;)&lt;/script&gt;&#39;</span>

  168. 	

  169. <span class="c"># File upload (from scripts)</span>

  170. <span class="c"># ------------------------------------------------------------------------------------</span>

  171. </div>

  172. <div class="viewcode-block" id="upload_many"><a class="viewcode-back" href="../../webnotes.html#webnotes.handler.upload_many">[docs]</a><span class="k">def</span> <span class="nf">upload_many</span><span class="p">():</span>

  173. 	<span class="kn">from</span> <span class="nn">webnotes.model.code</span> <span class="kn">import</span> <span class="n">get_obj</span>

  174. 

  175. 	<span class="c"># pass it on to upload_many method in Control Panel</span>

  176. 	<span class="n">cp</span> <span class="o">=</span> <span class="n">get_obj</span><span class="p">(</span><span class="s">&#39;Control Panel&#39;</span><span class="p">)</span>

  177. 	<span class="n">cp</span><span class="o">.</span><span class="n">upload_many</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">form</span><span class="p">)</span>

  178. 	

  179. 	<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;result&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="s">&quot;&quot;&quot;</span>

  180. <span class="s">&lt;script type=&#39;text/javascript&#39;&gt;</span>

  181. <span class="si">%s</span><span class="s"></span>

  182. <span class="s">&lt;/script&gt;</span>

  183. <span class="si">%s</span><span class="s"></span>

  184. <span class="si">%s</span><span class="s">&quot;&quot;&quot;</span> <span class="o">%</span> <span class="p">(</span><span class="n">cp</span><span class="o">.</span><span class="n">upload_callback</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">form</span><span class="p">),</span> <span class="s">&#39;</span><span class="se">\n</span><span class="s">----</span><span class="se">\n</span><span class="s">&#39;</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">message_log</span><span class="p">)</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span><span class="s">&quot;&#39;&quot;</span><span class="p">,</span> <span class="s">&quot;</span><span class="se">\&#39;</span><span class="s">&quot;</span><span class="p">),</span> <span class="s">&#39;</span><span class="se">\n</span><span class="s">----</span><span class="se">\n</span><span class="s">&#39;</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">debug_log</span><span class="p">)</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span><span class="s">&quot;&#39;&quot;</span><span class="p">,</span> <span class="s">&quot;</span><span class="se">\&#39;</span><span class="s">&quot;</span><span class="p">)</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span><span class="s">&quot;</span><span class="se">\n</span><span class="s">&quot;</span><span class="p">,</span><span class="s">&quot;&lt;br&gt;&quot;</span><span class="p">))</span>

  185. 	<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;type&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="s">&#39;iframe&#39;</span>

  186. 

  187. 

  188. <span class="c"># File download</span>

  189. <span class="c"># ------------------------------------------------------------------------------------</span></div>

  190. <div class="viewcode-block" id="get_file"><a class="viewcode-back" href="../../webnotes.html#webnotes.handler.get_file">[docs]</a><span class="k">def</span> <span class="nf">get_file</span><span class="p">():</span>

  191. 	<span class="kn">import</span> <span class="nn">webnotes.utils.file_manager</span>

  192. 

  193. 	<span class="n">res</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">utils</span><span class="o">.</span><span class="n">file_manager</span><span class="o">.</span><span class="n">get_file</span><span class="p">(</span><span class="n">form</span><span class="o">.</span><span class="n">getvalue</span><span class="p">(</span><span class="s">&#39;fname&#39;</span><span class="p">))</span>

  194. 	<span class="k">if</span> <span class="n">res</span><span class="p">:</span>

  195. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;type&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="s">&#39;download&#39;</span>

  196. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;filename&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">res</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span>

  197. 		

  198. 		<span class="k">if</span> <span class="nb">hasattr</span><span class="p">(</span><span class="n">res</span><span class="p">[</span><span class="mi">1</span><span class="p">],</span> <span class="s">&#39;tostring&#39;</span><span class="p">):</span>

  199. 			<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;filecontent&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">res</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span><span class="o">.</span><span class="n">tostring</span><span class="p">()</span>

  200. 		<span class="k">else</span><span class="p">:</span> 

  201. 			<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;filecontent&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">res</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span>

  202. 	<span class="k">else</span><span class="p">:</span>

  203. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">msgprint</span><span class="p">(</span><span class="s">&#39;[get_file] Unknown file name&#39;</span><span class="p">)</span>

  204. 		

  205. <span class="c"># Get Graph</span>

  206. <span class="c"># ------------------------------------------------------------------------------------</span></div>

  207. <div class="viewcode-block" id="get_graph"><a class="viewcode-back" href="../../webnotes.html#webnotes.handler.get_graph">[docs]</a><span class="k">def</span> <span class="nf">get_graph</span><span class="p">():</span>

  208. 	<span class="n">form</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">form</span>

  209. 

  210. 	<span class="kn">import</span> <span class="nn">StringIO</span>

  211. 	<span class="n">f</span> <span class="o">=</span> <span class="n">StringIO</span><span class="o">.</span><span class="n">StringIO</span><span class="p">()</span>

  212. 

  213. 	<span class="c"># call the object</span>

  214. 	<span class="n">obj</span> <span class="o">=</span> <span class="n">server</span><span class="o">.</span><span class="n">get_obj</span><span class="p">(</span><span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;dt&#39;</span><span class="p">))</span>

  215. 	<span class="n">plt</span> <span class="o">=</span> <span class="n">server</span><span class="o">.</span><span class="n">run_server_obj</span><span class="p">(</span><span class="n">obj</span><span class="p">,</span> <span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;method&#39;</span><span class="p">),</span> <span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;arg&#39;</span><span class="p">))</span>

  216. 	<span class="n">plt</span><span class="o">.</span><span class="n">savefig</span><span class="p">(</span><span class="n">f</span><span class="p">)</span>

  217. 

  218. 	<span class="c"># stream out</span>

  219. 	<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;type&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="s">&#39;download&#39;</span>

  220. 	<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;filename&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">user</span><span class="o">.</span><span class="n">get_random_password</span><span class="p">()</span> <span class="o">+</span> <span class="s">&#39;.png&#39;</span>

  221. 	<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;filecontent&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">f</span><span class="o">.</span><span class="n">getvalue</span><span class="p">()</span>

  222. 

  223. <span class="c"># Reset Password</span>

  224. <span class="c"># ------------------------------------------------------------------------------------</span>

  225. </div>

  226. <div class="viewcode-block" id="reset_password"><a class="viewcode-back" href="../../webnotes.html#webnotes.handler.reset_password">[docs]</a><span class="k">def</span> <span class="nf">reset_password</span><span class="p">():</span>

  227. 	<span class="n">form_dict</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">form_dict</span>

  228. 	

  229. 	<span class="n">act</span> <span class="o">=</span> <span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;account&#39;</span><span class="p">,</span> <span class="s">&#39;&#39;</span><span class="p">)</span>

  230. 	<span class="n">user</span> <span class="o">=</span> <span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;user&#39;</span><span class="p">,</span> <span class="s">&#39;&#39;</span><span class="p">)</span>

  231. 	<span class="k">if</span> <span class="n">act</span><span class="p">:</span>

  232. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="o">.</span><span class="n">set_db</span><span class="p">(</span><span class="n">act</span><span class="p">)</span>

  233. 

  234. 	<span class="k">try</span><span class="p">:</span>

  235. 		<span class="n">p</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">profile</span><span class="o">.</span><span class="n">Profile</span><span class="p">(</span><span class="n">user</span><span class="p">)</span>

  236. 		<span class="n">p</span><span class="o">.</span><span class="n">reset_password</span><span class="p">()</span>

  237. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">msgprint</span><span class="p">(</span><span class="s">&quot;Password has been reset and sent to your email id.&quot;</span><span class="p">)</span>

  238. 	<span class="k">except</span> <span class="ne">Exception</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span>

  239. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">msgprint</span><span class="p">(</span><span class="nb">str</span><span class="p">(</span><span class="n">e</span><span class="p">))</span>

  240. 

  241. <span class="c"># Resume session</span>

  242. <span class="c"># ------------------------------------------------------------------------------------</span>

  243. </div>

  244. <div class="viewcode-block" id="resume_session"><a class="viewcode-back" href="../../webnotes.html#webnotes.handler.resume_session">[docs]</a><span class="k">def</span> <span class="nf">resume_session</span><span class="p">():</span>

  245. 	<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;message&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">session_obj</span><span class="o">.</span><span class="n">resume</span><span class="p">()</span>

  246. 

  247. <span class="c"># -------------</span>

  248. <span class="c"># Create Backup</span>

  249. <span class="c"># -------------</span>

  250. </div>

  251. <div class="viewcode-block" id="backupdb"><a class="viewcode-back" href="../../webnotes.html#webnotes.handler.backupdb">[docs]</a><span class="k">def</span> <span class="nf">backupdb</span><span class="p">(</span><span class="n">form_dict</span><span class="p">,</span> <span class="n">session</span><span class="p">):</span>

  252. 	<span class="n">db_name</span> <span class="o">=</span> <span class="n">server</span><span class="o">.</span><span class="n">decrypt</span><span class="p">(</span><span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;db_name&#39;</span><span class="p">))</span>

  253. 

  254. 	<span class="n">server</span><span class="o">.</span><span class="n">backup_db</span><span class="p">(</span><span class="n">db_name</span><span class="p">)</span>

  255. 

  256. 	<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;type&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="s">&#39;download&#39;</span>

  257. 	<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;filename&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">db_name</span><span class="o">+</span><span class="s">&#39;.tar.gz&#39;</span>

  258. 	<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;filecontent&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="nb">open</span><span class="p">(</span><span class="s">&#39;../backups/&#39;</span> <span class="o">+</span> <span class="n">db_name</span><span class="o">+</span><span class="s">&#39;.tar.gz&#39;</span><span class="p">,</span><span class="s">&#39;rb&#39;</span><span class="p">)</span><span class="o">.</span><span class="n">read</span><span class="p">()</span>

  259. 

  260. <span class="c"># ---------------------------------------------------------------------</span>

  261. </div>

  262. <div class="viewcode-block" id="validate_cmd"><a class="viewcode-back" href="../../webnotes.html#webnotes.handler.validate_cmd">[docs]</a><span class="k">def</span> <span class="nf">validate_cmd</span><span class="p">(</span><span class="n">cmd</span><span class="p">):</span>

  263. 	<span class="c"># check if there is no direct possibility of malicious script injection</span>

  264. 	<span class="k">if</span> <span class="n">cmd</span><span class="o">.</span><span class="n">startswith</span><span class="p">(</span><span class="s">&#39;webnotes.model.code&#39;</span><span class="p">):</span>

  265. 		<span class="k">raise</span> <span class="ne">Exception</span><span class="p">,</span> <span class="s">&#39;Cannot call any methods from webnotes.model.code directly from the handler&#39;</span>

  266. 

  267. 	<span class="k">if</span> <span class="n">cmd</span><span class="o">.</span><span class="n">startswith</span><span class="p">(</span><span class="s">&#39;webnotes.model.db_schema&#39;</span><span class="p">):</span>

  268. 		<span class="k">raise</span> <span class="ne">Exception</span><span class="p">,</span> <span class="s">&#39;Cannot call any methods from webnotes.model.db_schema directly from the handler&#39;</span>

  269. 

  270. 	<span class="k">if</span> <span class="n">cmd</span><span class="o">.</span><span class="n">startswith</span><span class="p">(</span><span class="s">&#39;webnotes.conn&#39;</span><span class="p">):</span>

  271. 		<span class="k">raise</span> <span class="ne">Exception</span><span class="p">,</span> <span class="s">&#39;Cannot call database connection method directly from the handler&#39;</span>

  272. 	

  273. <span class="c"># Execution Starts Here</span>

  274. <span class="c"># ---------------------------------------------------------------------</span>

  275. </div>

  276. <span class="kn">import</span> <span class="nn">webnotes.auth</span>

  277. <span class="kn">import</span> <span class="nn">webnotes.db</span>

  278. 

  279. <span class="c"># reset password</span>

  280. <span class="c"># ---------------------------------------------------------------------</span>

  281. 

  282. <span class="k">if</span> <span class="n">form_dict</span><span class="o">.</span><span class="n">has_key</span><span class="p">(</span><span class="s">&#39;cmd&#39;</span><span class="p">)</span> <span class="ow">and</span> <span class="p">(</span><span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;cmd&#39;</span><span class="p">)</span><span class="o">==</span><span class="s">&#39;reset_password&#39;</span><span class="p">):</span>

  283. 	<span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">db</span><span class="o">.</span><span class="n">Database</span><span class="p">(</span><span class="n">use_default</span> <span class="o">=</span> <span class="mi">1</span><span class="p">)</span>

  284. 	<span class="n">sql</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="o">.</span><span class="n">sql</span>

  285. 	<span class="n">sql</span><span class="p">(</span><span class="s">&quot;START TRANSACTION&quot;</span><span class="p">)</span>

  286. 	<span class="k">try</span><span class="p">:</span>

  287. 		<span class="n">reset_password</span><span class="p">()</span>

  288. 		<span class="n">sql</span><span class="p">(</span><span class="s">&quot;COMMIT&quot;</span><span class="p">)</span>

  289. 	<span class="k">except</span> <span class="ne">Exception</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span>

  290. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">errprint</span><span class="p">(</span><span class="nb">str</span><span class="p">(</span><span class="n">e</span><span class="p">))</span>

  291. 		<span class="n">sql</span><span class="p">(</span><span class="s">&quot;ROLLBACK&quot;</span><span class="p">)</span>

  292. 	

  293. <span class="c"># pre-login access - for registration etc.</span>

  294. <span class="c"># ---------------------------------------------------------------------</span>

  295. 

  296. <span class="k">elif</span> <span class="n">form_dict</span><span class="o">.</span><span class="n">has_key</span><span class="p">(</span><span class="s">&#39;cmd&#39;</span><span class="p">)</span> <span class="ow">and</span> <span class="p">(</span><span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;cmd&#39;</span><span class="p">)</span><span class="o">==</span><span class="s">&#39;prelogin&#39;</span><span class="p">):</span>

  297. 	<span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">db</span><span class="o">.</span><span class="n">Database</span><span class="p">(</span><span class="n">use_default</span> <span class="o">=</span> <span class="mi">1</span><span class="p">)</span>

  298. 	<span class="n">sql</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="o">.</span><span class="n">sql</span>

  299. 	<span class="n">webnotes</span><span class="o">.</span><span class="n">session</span> <span class="o">=</span> <span class="p">{</span><span class="s">&#39;user&#39;</span><span class="p">:</span><span class="s">&#39;Administrator&#39;</span><span class="p">}</span>

  300. 

  301. 	<span class="kn">import</span> <span class="nn">webnotes.model.code</span>

  302. 	

  303. 	<span class="n">sql</span><span class="p">(</span><span class="s">&quot;START TRANSACTION&quot;</span><span class="p">)</span>

  304. 	<span class="k">try</span><span class="p">:</span>

  305. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;message&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">model</span><span class="o">.</span><span class="n">code</span><span class="o">.</span><span class="n">get_obj</span><span class="p">(</span><span class="s">&#39;Profile Control&#39;</span><span class="p">)</span><span class="o">.</span><span class="n">prelogin</span><span class="p">(</span><span class="n">form_dict</span><span class="p">)</span> <span class="ow">or</span> <span class="s">&#39;&#39;</span>

  306. 		<span class="n">sql</span><span class="p">(</span><span class="s">&quot;COMMIT&quot;</span><span class="p">)</span>

  307. 	<span class="k">except</span><span class="p">:</span>

  308. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">errprint</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">utils</span><span class="o">.</span><span class="n">getTraceback</span><span class="p">())</span>

  309. 		<span class="n">sql</span><span class="p">(</span><span class="s">&quot;ROLLBACK&quot;</span><span class="p">)</span>

  310. 

  311. <span class="c"># main stuff</span>

  312. <span class="c"># ---------------------------------------------------------------------</span>

  313. 

  314. <span class="k">else</span><span class="p">:</span>

  315. 

  316. 	<span class="k">try</span><span class="p">:</span>

  317. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">request</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">auth</span><span class="o">.</span><span class="n">HTTPRequest</span><span class="p">()</span>

  318. 	

  319. 		<span class="k">if</span> <span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;cmd&#39;</span><span class="p">)</span> <span class="o">!=</span> <span class="s">&#39;login&#39;</span> <span class="ow">and</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="p">:</span>

  320. 			<span class="n">sql</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="o">.</span><span class="n">sql</span>

  321. 		

  322. 			<span class="c"># NOTE:</span>

  323. 			<span class="c"># guest should only be allowed: </span>

  324. 			<span class="c"># getdoc (if Guest access)</span>

  325. 			<span class="c"># runserverobj (if Guest access)</span>

  326. 		

  327. 			<span class="c"># get command cmd</span>

  328. 			<span class="n">cmd</span> <span class="o">=</span> <span class="n">form_dict</span><span class="o">.</span><span class="n">has_key</span><span class="p">(</span><span class="s">&#39;cmd&#39;</span><span class="p">)</span> <span class="ow">and</span> <span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;cmd&#39;</span><span class="p">)</span> <span class="ow">or</span> <span class="s">&#39;&#39;</span>

  329. 			<span class="n">read_only</span> <span class="o">=</span> <span class="n">form_dict</span><span class="o">.</span><span class="n">has_key</span><span class="p">(</span><span class="s">&#39;_read_only&#39;</span><span class="p">)</span> <span class="ow">and</span> <span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;_read_only&#39;</span><span class="p">)</span> <span class="ow">or</span> <span class="bp">None</span>

  330. 

  331. 			<span class="n">validate_cmd</span><span class="p">(</span><span class="n">cmd</span><span class="p">)</span>

  332. 

  333. 			<span class="n">module</span> <span class="o">=</span> <span class="s">&#39;&#39;</span>

  334. 			<span class="k">if</span> <span class="s">&#39;.&#39;</span> <span class="ow">in</span> <span class="n">cmd</span><span class="p">:</span>

  335. 				<span class="n">module</span> <span class="o">=</span> <span class="s">&#39;.&#39;</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">cmd</span><span class="o">.</span><span class="n">split</span><span class="p">(</span><span class="s">&#39;.&#39;</span><span class="p">)[:</span><span class="o">-</span><span class="mi">1</span><span class="p">])</span>

  336. 				<span class="n">cmd</span> <span class="o">=</span> <span class="n">cmd</span><span class="o">.</span><span class="n">split</span><span class="p">(</span><span class="s">&#39;.&#39;</span><span class="p">)[</span><span class="o">-</span><span class="mi">1</span><span class="p">]</span>

  337. 

  338. 				<span class="k">exec</span> <span class="s">&#39;from </span><span class="si">%s</span><span class="s"> import </span><span class="si">%s</span><span class="s">&#39;</span> <span class="o">%</span> <span class="p">(</span><span class="n">module</span><span class="p">,</span> <span class="n">cmd</span><span class="p">)</span> <span class="ow">in</span> <span class="nb">locals</span><span class="p">()</span>			

  339. 	

  340. 	

  341. 			<span class="c"># execute</span>

  342. 			<span class="k">if</span> <span class="nb">locals</span><span class="p">()</span><span class="o">.</span><span class="n">has_key</span><span class="p">(</span><span class="n">cmd</span><span class="p">):</span>

  343. 				<span class="k">if</span> <span class="p">(</span><span class="ow">not</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="o">.</span><span class="n">in_transaction</span><span class="p">)</span> <span class="ow">and</span> <span class="p">(</span><span class="ow">not</span> <span class="n">read_only</span><span class="p">):</span>

  344. 					<span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="o">.</span><span class="n">begin</span><span class="p">()</span>

  345. 				

  346. 				<span class="k">if</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;arg&#39;</span><span class="p">):</span>

  347. 					<span class="c"># direct method call</span>

  348. 					<span class="n">ret</span> <span class="o">=</span> <span class="nb">locals</span><span class="p">()[</span><span class="n">cmd</span><span class="p">](</span><span class="n">webnotes</span><span class="o">.</span><span class="n">form_dict</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;arg&#39;</span><span class="p">))</span>

  349. 				<span class="k">else</span><span class="p">:</span>

  350. 					<span class="n">ret</span> <span class="o">=</span> <span class="nb">locals</span><span class="p">()[</span><span class="n">cmd</span><span class="p">]()</span>

  351. 				

  352. 				<span class="c"># returns with a message</span>

  353. 				<span class="k">if</span> <span class="n">ret</span><span class="p">:</span>

  354. 					<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;message&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">ret</span>

  355. 						

  356. 				<span class="c"># update session</span>

  357. 				<span class="n">webnotes</span><span class="o">.</span><span class="n">session_obj</span><span class="o">.</span><span class="n">update</span><span class="p">()</span>

  358. 				

  359. 				<span class="k">if</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="o">.</span><span class="n">in_transaction</span><span class="p">:</span>

  360. 					<span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="o">.</span><span class="n">commit</span><span class="p">()</span>

  361. 			<span class="k">else</span><span class="p">:</span>

  362. 				<span class="k">if</span> <span class="n">cmd</span><span class="o">!=</span><span class="s">&#39;login&#39;</span><span class="p">:</span>

  363. 					<span class="n">webnotes</span><span class="o">.</span><span class="n">msgprint</span><span class="p">(</span><span class="s">&#39;No Method: </span><span class="si">%s</span><span class="s">&#39;</span> <span class="o">%</span> <span class="n">cmd</span><span class="p">)</span>

  364. 

  365. 	<span class="k">except</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">ValidationError</span><span class="p">:</span>

  366. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="o">.</span><span class="n">rollback</span><span class="p">()</span>

  367. 	<span class="k">except</span><span class="p">:</span>

  368. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">errprint</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">utils</span><span class="o">.</span><span class="n">getTraceback</span><span class="p">())</span>

  369. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span> <span class="ow">and</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="o">.</span><span class="n">rollback</span><span class="p">()</span>

  370. 			

  371. 

  372. <span class="c">#### cleanup</span>

  373. <span class="c">#-----------</span>

  374. 

  375. <span class="k">if</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="p">:</span>

  376. 	<span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="o">.</span><span class="n">close</span><span class="p">()</span>

  377. 

  378. <span class="c">#### go</span>

  379. 

  380. <span class="kn">import</span> <span class="nn">string</span>

  381. <span class="kn">import</span> <span class="nn">os</span>

  382. 

  383. <span class="n">acceptsGzip</span><span class="p">,</span> <span class="n">out_buf</span><span class="p">,</span> <span class="n">str_out</span> <span class="o">=</span> <span class="mi">0</span><span class="p">,</span> <span class="bp">None</span><span class="p">,</span> <span class="bp">None</span>

  384. <span class="k">try</span><span class="p">:</span>

  385. 	<span class="k">if</span> <span class="n">string</span><span class="o">.</span><span class="n">find</span><span class="p">(</span><span class="n">os</span><span class="o">.</span><span class="n">environ</span><span class="p">[</span><span class="s">&quot;HTTP_ACCEPT_ENCODING&quot;</span><span class="p">],</span> <span class="s">&quot;gzip&quot;</span><span class="p">)</span> <span class="o">!=</span> <span class="o">-</span><span class="mi">1</span><span class="p">:</span>

  386. 		<span class="n">acceptsGzip</span> <span class="o">=</span> <span class="mi">1</span> <span class="c"># problem in win ?</span>

  387. <span class="k">except</span><span class="p">:</span>

  388. 	<span class="k">pass</span>

  389. 

  390. <div class="viewcode-block" id="compressBuf"><a class="viewcode-back" href="../../webnotes.html#webnotes.handler.compressBuf">[docs]</a><span class="k">def</span> <span class="nf">compressBuf</span><span class="p">(</span><span class="n">buf</span><span class="p">):</span>

  391. 	<span class="kn">import</span> <span class="nn">gzip</span><span class="o">,</span> <span class="nn">cStringIO</span>

  392. 	<span class="n">zbuf</span> <span class="o">=</span> <span class="n">cStringIO</span><span class="o">.</span><span class="n">StringIO</span><span class="p">()</span>

  393. 	<span class="n">zfile</span> <span class="o">=</span> <span class="n">gzip</span><span class="o">.</span><span class="n">GzipFile</span><span class="p">(</span><span class="n">mode</span> <span class="o">=</span> <span class="s">&#39;wb&#39;</span><span class="p">,</span>  <span class="n">fileobj</span> <span class="o">=</span> <span class="n">zbuf</span><span class="p">,</span> <span class="n">compresslevel</span> <span class="o">=</span> <span class="mi">5</span><span class="p">)</span>

  394. 	<span class="n">zfile</span><span class="o">.</span><span class="n">write</span><span class="p">(</span><span class="n">buf</span><span class="p">)</span>

  395. 	<span class="n">zfile</span><span class="o">.</span><span class="n">close</span><span class="p">()</span>

  396. 	<span class="k">return</span> <span class="n">zbuf</span><span class="o">.</span><span class="n">getvalue</span><span class="p">()</span>

  397. 

  398. <span class="c"># CSV</span>

  399. <span class="c"># -------------------------------------------------------------------</span>

  400. </div>

  401. <span class="k">if</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;type&#39;</span><span class="p">)</span><span class="o">==</span><span class="s">&#39;csv&#39;</span><span class="p">:</span>

  402. 	<span class="k">print</span> <span class="s">&quot;Content-Type: text/csv&quot;</span>

  403. 	<span class="k">print</span> <span class="s">&quot;Content-Disposition: attachment; filename=&quot;</span><span class="o">+</span><span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;doctype&#39;</span><span class="p">]</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span><span class="s">&#39; &#39;</span><span class="p">,</span> <span class="s">&#39;_&#39;</span><span class="p">)</span><span class="o">+</span><span class="s">&quot;.csv&quot;</span>

  404. 	<span class="k">print</span>

  405. 	<span class="k">print</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;result&#39;</span><span class="p">]</span>

  406. 

  407. <span class="c"># IFRAME</span>

  408. <span class="c"># -------------------------------------------------------------------</span>

  409. 

  410. <span class="k">elif</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;type&#39;</span><span class="p">)</span><span class="o">==</span><span class="s">&#39;iframe&#39;</span><span class="p">:</span>

  411. 	<span class="k">print</span> <span class="s">&quot;Content-Type: text/html&quot;</span>

  412. 	<span class="k">print</span>

  413. 	<span class="k">if</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;result&#39;</span><span class="p">):</span>

  414. 		<span class="k">print</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;result&#39;</span><span class="p">]</span>

  415. 	<span class="k">if</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">debug_log</span><span class="p">:</span>

  416. 		<span class="k">print</span> <span class="s">&#39;&#39;&#39;&lt;script type=&#39;text/javascript&#39;&gt;alert(&quot;</span><span class="si">%s</span><span class="s">&quot;);&lt;/script&gt;&#39;&#39;&#39;</span> <span class="o">%</span> <span class="p">(</span><span class="s">&#39;-------&#39;</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">debug_log</span><span class="p">)</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span><span class="s">&#39;&quot;&#39;</span><span class="p">,</span> <span class="s">&#39;&#39;</span><span class="p">)</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span><span class="s">&#39;</span><span class="se">\n</span><span class="s">&#39;</span><span class="p">,</span><span class="s">&#39;&#39;</span><span class="p">))</span>

  417. 

  418. <span class="c"># file</span>

  419. <span class="c"># -------------------------------------------------------------------</span>

  420. 

  421. <span class="k">elif</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;type&#39;</span><span class="p">)</span><span class="o">==</span><span class="s">&#39;download&#39;</span><span class="p">:</span>

  422. 	<span class="kn">import</span> <span class="nn">mimetypes</span>

  423. 	<span class="k">print</span> <span class="s">&quot;Content-Type: </span><span class="si">%s</span><span class="s">&quot;</span> <span class="o">%</span> <span class="p">(</span><span class="n">mimetypes</span><span class="o">.</span><span class="n">guess_type</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;filename&#39;</span><span class="p">])[</span><span class="mi">0</span><span class="p">]</span> <span class="ow">or</span> <span class="s">&#39;application/unknown&#39;</span><span class="p">)</span>

  424. 	<span class="k">print</span> <span class="s">&quot;Content-Disposition: filename=&quot;</span><span class="o">+</span><span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;filename&#39;</span><span class="p">]</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span><span class="s">&#39; &#39;</span><span class="p">,</span> <span class="s">&#39;_&#39;</span><span class="p">)</span>

  425. 	<span class="k">print</span>

  426. 	<span class="k">print</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;filecontent&#39;</span><span class="p">]</span>

  427. 

  428. <span class="c"># JSON</span>

  429. <span class="c"># -------------------------------------------------------------------</span>

  430. 

  431. <span class="k">else</span><span class="p">:</span>

  432. 	<span class="k">if</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">debug_log</span><span class="p">:</span>

  433. 		<span class="n">save_log</span> <span class="o">=</span> <span class="mi">1</span>

  434. 		<span class="k">if</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">debug_log</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">startswith</span><span class="p">(</span><span class="s">&#39;[Validation Error]&#39;</span><span class="p">):</span>

  435. 			<span class="n">save_log</span> <span class="o">=</span> <span class="mi">0</span>

  436. 

  437. 		<span class="n">t</span> <span class="o">=</span> <span class="s">&#39;</span><span class="se">\n</span><span class="s">----------------</span><span class="se">\n</span><span class="s">&#39;</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">debug_log</span><span class="p">)</span>

  438. 		<span class="k">if</span> <span class="n">errdoctype</span><span class="p">:</span> 

  439. 			<span class="n">t</span> <span class="o">=</span> <span class="n">t</span> <span class="o">+</span> <span class="s">&#39;</span><span class="se">\n</span><span class="s">DocType: &#39;</span> <span class="o">+</span> <span class="n">errdoctype</span>

  440. 		<span class="k">if</span> <span class="n">errdoc</span><span class="p">:</span> 

  441. 			<span class="n">t</span> <span class="o">=</span> <span class="n">t</span> <span class="o">+</span> <span class="s">&#39;</span><span class="se">\n</span><span class="s">Name: &#39;</span> <span class="o">+</span> <span class="n">errdoc</span>

  442. 		<span class="k">if</span> <span class="n">errmethod</span><span class="p">:</span> 

  443. 			<span class="n">t</span> <span class="o">=</span> <span class="n">t</span> <span class="o">+</span> <span class="s">&#39;</span><span class="se">\n</span><span class="s">Method: &#39;</span> <span class="o">+</span> <span class="n">errmethod</span>

  444. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;exc&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="s">&#39;&lt;pre&gt;&#39;</span><span class="o">+</span><span class="n">t</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span><span class="s">&#39;</span><span class="se">\n</span><span class="s">&#39;</span><span class="p">,</span><span class="s">&#39;&lt;br&gt;&#39;</span><span class="p">)</span><span class="o">+</span><span class="s">&#39;&lt;/pre&gt;&#39;</span>

  445. 

  446. 		<span class="k">if</span> <span class="n">save_log</span><span class="p">:</span> <span class="c"># don&#39;t save validation errors</span>

  447. 			<span class="k">try</span><span class="p">:</span>  <span class="n">save_log</span><span class="p">(</span><span class="n">t</span><span class="p">,</span> <span class="s">&#39;Server&#39;</span><span class="p">)</span>

  448. 			<span class="k">except</span><span class="p">:</span> <span class="k">pass</span>

  449. 

  450. 	<span class="k">if</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">message_log</span><span class="p">:</span>

  451. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;server_messages&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="s">&#39;</span><span class="se">\n</span><span class="s">----------------</span><span class="se">\n</span><span class="s">&#39;</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">message_log</span><span class="p">)</span>

  452. 

  453. 	<span class="n">cleanup_docs</span><span class="p">()</span>

  454. 	

  455. 	<span class="c"># Convert to JSON</span>

  456. 	<span class="c"># ---------------</span>

  457. 	<span class="k">try</span><span class="p">:</span>

  458. 		<span class="kn">import</span> <span class="nn">json</span>

  459. 	<span class="k">except</span><span class="p">:</span> <span class="c"># python 2.4</span>

  460. 		<span class="kn">import</span> <span class="nn">simplejson</span> <span class="kn">as</span> <span class="nn">json</span>

  461. 	

  462. 	<span class="k">try</span><span class="p">:</span>

  463. 		<span class="n">str_out</span> <span class="o">=</span> <span class="n">json</span><span class="o">.</span><span class="n">dumps</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">)</span>

  464. 	<span class="k">except</span><span class="p">:</span>

  465. 		<span class="n">str_out</span> <span class="o">=</span> <span class="nb">str</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">)</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span><span class="s">&#39;, None&#39;</span><span class="p">,</span> <span class="s">&#39;, &quot;&quot;&#39;</span><span class="p">)</span>

  466. 	

  467. 	<span class="k">if</span> <span class="n">acceptsGzip</span> <span class="ow">and</span> <span class="nb">len</span><span class="p">(</span><span class="n">str_out</span><span class="p">)</span><span class="o">&gt;</span><span class="mi">512</span><span class="p">:</span>

  468. 		<span class="n">out_buf</span> <span class="o">=</span> <span class="n">compressBuf</span><span class="p">(</span><span class="n">str_out</span><span class="p">)</span>

  469. 		<span class="k">print</span> <span class="s">&quot;Content-Encoding: gzip&quot;</span>

  470. 		<span class="k">print</span> <span class="s">&quot;Content-Length: </span><span class="si">%d</span><span class="s">&quot;</span> <span class="o">%</span> <span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">out_buf</span><span class="p">))</span>

  471. 		

  472. 	<span class="k">print</span> <span class="s">&quot;Content-Type: text/html; Charset: ISO-8859-1&quot;</span>

  473. 	

  474. 	<span class="c"># if there ar additional cookies defined during the request, add them here</span>

  475. 	<span class="k">if</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">cookies</span> <span class="ow">or</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">add_cookies</span><span class="p">:</span> 

  476. 		<span class="k">for</span> <span class="n">c</span> <span class="ow">in</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">add_cookies</span><span class="o">.</span><span class="n">keys</span><span class="p">():</span>

  477. 			<span class="n">webnotes</span><span class="o">.</span><span class="n">cookies</span><span class="p">[</span><span class="n">c</span><span class="p">]</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">add_cookies</span><span class="p">[</span><span class="n">c</span><span class="p">]</span>

  478. 			

  479. 		<span class="k">print</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">cookies</span>

  480. 		

  481. 	<span class="k">print</span> <span class="c"># Headers end</span>

  482. 	

  483. <span class="k">if</span> <span class="n">out_buf</span><span class="p">:</span>

  484. 	<span class="n">sys</span><span class="o">.</span><span class="n">stdout</span><span class="o">.</span><span class="n">write</span><span class="p">(</span><span class="n">out_buf</span><span class="p">)</span>

  485. <span class="k">elif</span> <span class="n">str_out</span><span class="p">:</span>

  486. 	<span class="k">print</span> <span class="n">str_out</span>

  487. </pre></div>

  488. 

  489.           </div>

  490.         </div>

  491.       </div>

  492.       <div class="sphinxsidebar">

  493.         <div class="sphinxsidebarwrapper">

  494.             <p class="logo"><a href="../../index.html">

  495.               <img class="logo" src="../../_static/wnflogo200.gif" alt="Logo"/>

  496.             </a></p>

  497. <div id="searchbox" style="display: none">

  498.   <h3>Quick search</h3>

  499.     <form class="search" action="../../search.html" method="get">

  500.       <input type="text" name="q" size="18" />

  501.       <input type="submit" value="Go" />

  502.       <input type="hidden" name="check_keywords" value="yes" />

  503.       <input type="hidden" name="area" value="default" />

  504.     </form>

  505.     <p class="searchtip" style="font-size: 90%">

  506.     Enter search terms or a module, class or function name.

  507.     </p>

  508. </div>

  509. <script type="text/javascript">$('#searchbox').show(0);</script>

  510.         </div>

  511.       </div>

  512.       <div class="clearer"></div>

  513.     </div>

  514.     <div class="related">

  515.       <h3>Navigation</h3>

  516.       <ul>

  517.         <li class="right" style="margin-right: 10px">

  518.           <a href="../../genindex.html" title="General Index"

  519.              >index</a></li>

  520.         <li class="right" >

  521.           <a href="../../py-modindex.html" title="Python Module Index"

  522.              >modules</a> |</li>

  523.         <li><a href="../../index.html">WNFramework v1.8 documentation</a> &raquo;</li>

  524.           <li><a href="../index.html" >Module code</a> &raquo;</li>

  525.           <li><a href="../webnotes.html" >webnotes</a> &raquo;</li> 

  526.       </ul>

  527.     </div>

  528.     <div class="footer">

  529.         &copy; Copyright 2011, Rushabh Mehta, Web Notes Technologies.

  530.       Created using <a href="http://sphinx.pocoo.org/">Sphinx</a> 1.0.7.

  531.     </div>

  532.   </body>

  533. </html>