anoopmb
/
frappe
1
0
Форкнуть 0
Код Задачи 0 Pull Request'ы 0 Релизы 0 Вики Активность
Вы не можете выбрать более 25 тем Темы должны начинаться с буквы или цифры, могут содержать дефисы(-) и должны содержать не более 35 символов.
252 коммитов
1 ветка
314 MiB
 
 
 
 
 
 
Дерево: 98acabbb96
Ветки Теги
${ item.name }
Создать ветку ${ searchTerm }
из '98acabbb96'
${ noResults }
frappe/docs/_build/html/_modules/webnotes/auth.html

520 строки
66 KiB
Исходник Вина История 

  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.auth &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.auth</h1><div class="highlight"><pre>

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

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

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

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

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

  55. 

  56. <span class="c"># =================================================================================</span>

  57. <span class="c"># HTTPRequest</span>

  58. <span class="c"># =================================================================================</span>

  59. 

  60. <div class="viewcode-block" id="HTTPRequest"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.HTTPRequest">[docs]</a><span class="k">class</span> <span class="nc">HTTPRequest</span><span class="p">:</span>

  61. 	<span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>

  62. 

  63. 		<span class="c"># Get Environment variables</span>

  64. 		<span class="bp">self</span><span class="o">.</span><span class="n">domain</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">get_env_vars</span><span class="p">(</span><span class="s">&#39;HTTP_HOST&#39;</span><span class="p">)</span>

  65. 		<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">domain</span> <span class="ow">and</span> <span class="bp">self</span><span class="o">.</span><span class="n">domain</span><span class="o">.</span><span class="n">startswith</span><span class="p">(</span><span class="s">&#39;www.&#39;</span><span class="p">):</span>

  66. 			<span class="bp">self</span><span class="o">.</span><span class="n">domain</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">domain</span><span class="p">[</span><span class="mi">4</span><span class="p">:]</span>

  67. 

  68. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">remote_ip</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">get_env_vars</span><span class="p">(</span><span class="s">&#39;REMOTE_ADDR&#39;</span><span class="p">)</span>					

  69. 

  70. 		<span class="c"># load cookies</span>

  71. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">cookie_manager</span> <span class="o">=</span> <span class="n">CookieManager</span><span class="p">()</span>

  72. 

  73. 		<span class="c"># set db</span>

  74. 		<span class="bp">self</span><span class="o">.</span><span class="n">set_db</span><span class="p">()</span>

  75. 

  76. 		<span class="c"># check status</span>

  77. 		<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">get_global</span><span class="p">(</span><span class="s">&quot;__session_status&quot;</span><span class="p">)</span><span class="o">==</span><span class="s">&#39;stop&#39;</span><span class="p">:</span>

  78. 			<span class="n">webnotes</span><span class="o">.</span><span class="n">msgprint</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="o">.</span><span class="n">get_global</span><span class="p">(</span><span class="s">&quot;__session_status_message&quot;</span><span class="p">))</span>

  79. 			<span class="k">raise</span> <span class="ne">Exception</span>

  80. 

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

  82. 		<span class="c"># start transaction</span>

  83. 		<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>

  84. 

  85. 		<span class="c"># login</span>

  86. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">login_manager</span> <span class="o">=</span> <span class="n">LoginManager</span><span class="p">()</span>

  87. 

  88. 		<span class="c"># start session</span>

  89. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">session_obj</span> <span class="o">=</span> <span class="n">Session</span><span class="p">()</span>

  90. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">session</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">data</span>

  91. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">tenant_id</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">session</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;tenant_id&#39;</span><span class="p">,</span> <span class="mi">0</span><span class="p">)</span>

  92. 

  93. 		<span class="c"># write out cookies if sid is supplied (this is a pre-logged in redirect)</span>

  94. 		<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;sid&#39;</span><span class="p">):</span>

  95. 			<span class="n">webnotes</span><span class="o">.</span><span class="n">cookie_manager</span><span class="o">.</span><span class="n">set_cookies</span><span class="p">()</span>

  96. 

  97. 		<span class="c"># run login triggers</span>

  98. 		<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;cmd&#39;</span><span class="p">)</span><span class="o">==</span><span class="s">&#39;login&#39;</span><span class="p">:</span>

  99. 			<span class="n">webnotes</span><span class="o">.</span><span class="n">login_manager</span><span class="o">.</span><span class="n">run_trigger</span><span class="p">(</span><span class="s">&#39;on_login_post_session&#39;</span><span class="p">)</span>

  100. 			

  101. 		<span class="c"># load profile</span>

  102. 		<span class="bp">self</span><span class="o">.</span><span class="n">setup_profile</span><span class="p">()</span>

  103. 

  104. 		<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>

  105. 		<span class="c"># end transaction</span>

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

  107. 

  108. 	<span class="c"># setup profile</span>

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

  110. 

  111. <div class="viewcode-block" id="HTTPRequest.setup_profile"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.HTTPRequest.setup_profile">[docs]</a>	<span class="k">def</span> <span class="nf">setup_profile</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>

  112. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">user</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>

  113. 		<span class="c"># load the profile data</span>

  114. 		<span class="k">if</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">session</span><span class="p">[</span><span class="s">&#39;data&#39;</span><span class="p">]</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;profile&#39;</span><span class="p">):</span>

  115. 			<span class="n">webnotes</span><span class="o">.</span><span class="n">user</span><span class="o">.</span><span class="n">load_from_session</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">session</span><span class="p">[</span><span class="s">&#39;data&#39;</span><span class="p">][</span><span class="s">&#39;profile&#39;</span><span class="p">])</span>

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

  117. 			<span class="n">webnotes</span><span class="o">.</span><span class="n">user</span><span class="o">.</span><span class="n">load_profile</span><span class="p">()</span>	

  118. 

  119. 

  120. 	<span class="c"># get account name</span>

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

  122. </div>

  123. <div class="viewcode-block" id="HTTPRequest.get_ac_name"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.HTTPRequest.get_ac_name">[docs]</a>	<span class="k">def</span> <span class="nf">get_ac_name</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>

  124. 		<span class="c"># login</span>

  125. 		<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;acx&#39;</span><span class="p">):</span>

  126. 			<span class="k">return</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;acx&#39;</span><span class="p">)</span>

  127. 		

  128. 		<span class="c"># in form</span>

  129. 		<span class="k">elif</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;ac_name&#39;</span><span class="p">):</span>

  130. 			<span class="k">return</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;ac_name&#39;</span><span class="p">)</span>

  131. 			

  132. 		<span class="c"># in cookie</span>

  133. 		<span class="k">elif</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">incoming_cookies</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;ac_name&#39;</span><span class="p">):</span>

  134. 			<span class="k">return</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">incoming_cookies</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;ac_name&#39;</span><span class="p">)</span>

  135. 			

  136. 			

  137. 	<span class="c"># set database login</span>

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

  139. </div>

  140. <div class="viewcode-block" id="HTTPRequest.set_db"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.HTTPRequest.set_db">[docs]</a>	<span class="k">def</span> <span class="nf">set_db</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">ac_name</span> <span class="o">=</span> <span class="bp">None</span><span class="p">):</span>

  141. 		

  142. 

  143. 		<span class="c"># select based on subdomain</span>

  144. 		<span class="k">if</span> <span class="nb">getattr</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">defs</span><span class="p">,</span><span class="s">&#39;domain_name_map&#39;</span><span class="p">,</span> <span class="p">{})</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">domain</span><span class="p">):</span>

  145. 			<span class="n">db_name</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">defs</span><span class="o">.</span><span class="n">domain_name_map</span><span class="p">[</span><span class="bp">self</span><span class="o">.</span><span class="n">domain</span><span class="p">]</span>

  146. 

  147. 		<span class="c"># select based on ac_name</span>

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

  149. 			<span class="n">ac_name</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_ac_name</span><span class="p">()</span>

  150. 			<span class="k">if</span> <span class="n">ac_name</span><span class="p">:</span>

  151. 				<span class="n">db_name</span> <span class="o">=</span> <span class="nb">getattr</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">defs</span><span class="p">,</span><span class="s">&#39;db_name_map&#39;</span><span class="p">,{})</span><span class="o">.</span>\

  152. 					<span class="n">get</span><span class="p">(</span><span class="n">ac_name</span><span class="p">,</span> <span class="n">ac_name</span><span class="p">)</span>

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

  154. 				<span class="n">db_name</span> <span class="o">=</span> <span class="nb">getattr</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">defs</span><span class="p">,</span><span class="s">&#39;default_db_name&#39;</span><span class="p">,</span><span class="s">&#39;&#39;</span><span class="p">)</span>

  155. 	

  156. 		<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">user</span> <span class="o">=</span> <span class="n">db_name</span><span class="p">,</span><span class="n">password</span> <span class="o">=</span> <span class="nb">getattr</span><span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">defs</span><span class="p">,</span><span class="s">&#39;db_password&#39;</span><span class="p">,</span><span class="s">&#39;&#39;</span><span class="p">))</span>

  157. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">ac_name</span> <span class="o">=</span> <span class="n">ac_name</span>

  158. 

  159. <span class="c"># =================================================================================</span>

  160. <span class="c"># Login Manager</span>

  161. <span class="c"># =================================================================================</span>

  162. </div></div>

  163. <div class="viewcode-block" id="LoginManager"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.LoginManager">[docs]</a><span class="k">class</span> <span class="nc">LoginManager</span><span class="p">:</span>

  164. 	<span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>

  165. 		<span class="bp">self</span><span class="o">.</span><span class="n">cp</span> <span class="o">=</span> <span class="bp">None</span>

  166. 		<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;cmd&#39;</span><span class="p">)</span><span class="o">==</span><span class="s">&#39;login&#39;</span><span class="p">:</span>

  167. 			<span class="c"># clear cache</span>

  168. 			<span class="kn">from</span> <span class="nn">webnotes.session_cache</span> <span class="kn">import</span> <span class="n">clear_cache</span>

  169. 			<span class="n">clear_cache</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;usr&#39;</span><span class="p">))</span>				

  170. 

  171. 			<span class="bp">self</span><span class="o">.</span><span class="n">authenticate</span><span class="p">()</span>

  172. 			<span class="bp">self</span><span class="o">.</span><span class="n">post_login</span><span class="p">()</span>

  173. 			<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="s">&#39;Logged In&#39;</span>

  174. 

  175. 	<span class="c"># run triggers, write cookies</span>

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

  177. 	

  178. <div class="viewcode-block" id="LoginManager.post_login"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.LoginManager.post_login">[docs]</a>	<span class="k">def</span> <span class="nf">post_login</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>

  179. 		<span class="bp">self</span><span class="o">.</span><span class="n">validate_ip_address</span><span class="p">()</span>

  180. 		<span class="bp">self</span><span class="o">.</span><span class="n">run_trigger</span><span class="p">()</span>

  181. 	

  182. 	<span class="c"># check password</span>

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

  184. 	</div>

  185. <div class="viewcode-block" id="LoginManager.authenticate"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.LoginManager.authenticate">[docs]</a>	<span class="k">def</span> <span class="nf">authenticate</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">user</span><span class="o">=</span><span class="bp">None</span><span class="p">,</span> <span class="n">pwd</span><span class="o">=</span><span class="bp">None</span><span class="p">):</span>

  186. 		<span class="k">if</span> <span class="ow">not</span> <span class="p">(</span><span class="n">user</span> <span class="ow">and</span> <span class="n">pwd</span><span class="p">):</span>	

  187. 			<span class="n">user</span><span class="p">,</span> <span class="n">pwd</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;usr&#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;pwd&#39;</span><span class="p">)</span>

  188. 

  189. 		<span class="k">if</span> <span class="ow">not</span> <span class="p">(</span><span class="n">user</span> <span class="ow">and</span> <span class="n">pwd</span><span class="p">):</span>

  190. 			<span class="n">webnotes</span><span class="o">.</span><span class="n">msgprint</span><span class="p">(</span><span class="s">&#39;Incomplete Login Details&#39;</span><span class="p">,</span> <span class="n">raise_exception</span><span class="o">=</span><span class="mi">1</span><span class="p">)</span>

  191. 		

  192. 		<span class="c"># custom authentication (for single-sign on)</span>

  193. 		<span class="bp">self</span><span class="o">.</span><span class="n">load_control_panel</span><span class="p">()</span>

  194. 		<span class="k">if</span> <span class="nb">hasattr</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">cp</span><span class="p">,</span> <span class="s">&#39;authenticate&#39;</span><span class="p">):</span>

  195. 			<span class="bp">self</span><span class="o">.</span><span class="n">user</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">cp</span><span class="o">.</span><span class="n">authenticate</span><span class="p">()</span>

  196. 		

  197. 		<span class="c"># check the password</span>

  198. 		<span class="k">if</span> <span class="n">user</span><span class="o">==</span><span class="s">&#39;Administrator&#39;</span><span class="p">:</span>

  199. 			<span class="n">p</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><span class="p">(</span><span class="s">&quot;select name from tabProfile where name=</span><span class="si">%s</span><span class="s"> and (`password`=</span><span class="si">%s</span><span class="s"> OR `password`=PASSWORD(</span><span class="si">%s</span><span class="s">))&quot;</span><span class="p">,</span> <span class="p">(</span><span class="n">user</span><span class="p">,</span> <span class="n">pwd</span><span class="p">,</span> <span class="n">pwd</span><span class="p">))</span>

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

  201. 			<span class="n">p</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><span class="p">(</span><span class="s">&quot;select name from tabProfile where name=</span><span class="si">%s</span><span class="s"> and (`password`=</span><span class="si">%s</span><span class="s">  OR `password`=PASSWORD(</span><span class="si">%s</span><span class="s">)) and IFNULL(enabled,0)=1&quot;</span><span class="p">,</span> <span class="p">(</span><span class="n">user</span><span class="p">,</span> <span class="n">pwd</span><span class="p">,</span> <span class="n">pwd</span><span class="p">))</span>

  202. 			

  203. 		<span class="k">if</span> <span class="ow">not</span> <span class="n">p</span><span class="p">:</span>

  204. 			<span class="n">webnotes</span><span class="o">.</span><span class="n">msgprint</span><span class="p">(</span><span class="s">&#39;Authentication Failed&#39;</span><span class="p">,</span> <span class="n">raise_exception</span><span class="o">=</span><span class="mi">1</span><span class="p">)</span>

  205. 			

  206. 		<span class="bp">self</span><span class="o">.</span><span class="n">user</span> <span class="o">=</span> <span class="n">p</span><span class="p">[</span><span class="mi">0</span><span class="p">][</span><span class="mi">0</span><span class="p">]</span>

  207. 	

  208. 	<span class="c"># triggers</span>

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

  210. 	</div>

  211. <div class="viewcode-block" id="LoginManager.load_control_panel"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.LoginManager.load_control_panel">[docs]</a>	<span class="k">def</span> <span class="nf">load_control_panel</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>

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

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

  214. 			<span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">cp</span><span class="p">:</span>

  215. 				<span class="bp">self</span><span class="o">.</span><span class="n">cp</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;Control Panel&#39;</span><span class="p">)</span>

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

  217. 			<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;Control Panel Exception&#39;</span><span class="p">]</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">getTraceback</span><span class="p">()</span>

  218. 	

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

  220. <div class="viewcode-block" id="LoginManager.run_trigger"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.LoginManager.run_trigger">[docs]</a>	<span class="k">def</span> <span class="nf">run_trigger</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">method</span><span class="o">=</span><span class="s">&#39;on_login&#39;</span><span class="p">):</span>

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

  222. 			<span class="kn">import</span> <span class="nn">event_handlers</span>

  223. 			<span class="k">if</span> <span class="nb">hasattr</span><span class="p">(</span><span class="n">event_handlers</span><span class="p">,</span> <span class="n">method</span><span class="p">):</span>

  224. 				<span class="nb">getattr</span><span class="p">(</span><span class="n">event_handlers</span><span class="p">,</span> <span class="n">method</span><span class="p">)(</span><span class="bp">self</span><span class="p">)</span>

  225. 			<span class="k">return</span>

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

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

  228. 	

  229. 		<span class="c"># deprecated</span>

  230. 		<span class="bp">self</span><span class="o">.</span><span class="n">load_control_panel</span><span class="p">()</span>

  231. 		<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">cp</span> <span class="ow">and</span> <span class="nb">hasattr</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">cp</span><span class="p">,</span> <span class="n">method</span><span class="p">):</span>

  232. 			<span class="nb">getattr</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">cp</span><span class="p">,</span> <span class="n">method</span><span class="p">)(</span><span class="bp">self</span><span class="p">)</span>

  233. 

  234. 	<span class="c"># ip validation</span>

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

  236. 	</div>

  237. <div class="viewcode-block" id="LoginManager.validate_ip_address"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.LoginManager.validate_ip_address">[docs]</a>	<span class="k">def</span> <span class="nf">validate_ip_address</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>

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

  239. 			<span class="n">ip</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><span class="p">(</span><span class="s">&quot;select ip_address from tabProfile where name = &#39;</span><span class="si">%s</span><span class="s">&#39;&quot;</span> <span class="o">%</span> <span class="bp">self</span><span class="o">.</span><span class="n">user</span><span class="p">)[</span><span class="mi">0</span><span class="p">][</span><span class="mi">0</span><span class="p">]</span> <span class="ow">or</span> <span class="s">&#39;&#39;</span>

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

  241. 			

  242. 		<span class="n">ip</span> <span class="o">=</span> <span class="n">ip</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span><span class="s">&quot;,&quot;</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="o">.</span><span class="n">split</span><span class="p">(</span><span class="s">&#39;</span><span class="se">\n</span><span class="s">&#39;</span><span class="p">)</span>

  243. 		<span class="n">ip</span> <span class="o">=</span> <span class="p">[</span><span class="n">i</span><span class="o">.</span><span class="n">strip</span><span class="p">()</span> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="n">ip</span><span class="p">]</span>

  244. 			

  245. 		<span class="k">if</span> <span class="n">ret</span> <span class="ow">and</span> <span class="n">ip</span><span class="p">:</span>

  246. 			<span class="k">if</span> <span class="ow">not</span> <span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">remote_ip</span><span class="o">.</span><span class="n">startswith</span><span class="p">(</span><span class="n">ip</span><span class="p">[</span><span class="mi">0</span><span class="p">])</span> <span class="ow">or</span> <span class="p">(</span><span class="n">webnotes</span><span class="o">.</span><span class="n">remote_ip</span> <span class="ow">in</span> <span class="n">ip</span><span class="p">)):</span>

  247. 				<span class="k">raise</span> <span class="ne">Exception</span><span class="p">,</span> <span class="s">&#39;Not allowed from this IP Address&#39;</span>	

  248. 

  249. 	<span class="c"># login as guest</span>

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

  251. 	</div>

  252. <div class="viewcode-block" id="LoginManager.login_as_guest"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.LoginManager.login_as_guest">[docs]</a>	<span class="k">def</span> <span class="nf">login_as_guest</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>

  253. 		<span class="n">res</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><span class="p">(</span><span class="s">&quot;select name from tabProfile where name=&#39;Guest&#39; and ifnull(enabled,0)=1&quot;</span><span class="p">)</span>

  254. 		<span class="k">if</span> <span class="ow">not</span> <span class="n">res</span><span class="p">:</span>

  255. 			<span class="k">raise</span> <span class="ne">Exception</span><span class="p">,</span> <span class="s">&quot;No Guest Access&quot;</span>

  256. 		<span class="bp">self</span><span class="o">.</span><span class="n">user</span> <span class="o">=</span> <span class="s">&#39;Guest&#39;</span>

  257. 		<span class="bp">self</span><span class="o">.</span><span class="n">post_login</span><span class="p">()</span>

  258. 

  259. 	<span class="c"># Logout</span>

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

  261. 	</div>

  262. <div class="viewcode-block" id="LoginManager.call_on_logout_event"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.LoginManager.call_on_logout_event">[docs]</a>	<span class="k">def</span> <span class="nf">call_on_logout_event</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>

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

  264. 		<span class="n">cp</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;Control Panel&#39;</span><span class="p">,</span> <span class="s">&#39;Control Panel&#39;</span><span class="p">)</span>

  265. 		<span class="k">if</span> <span class="nb">hasattr</span><span class="p">(</span><span class="n">cp</span><span class="p">,</span> <span class="s">&#39;on_logout&#39;</span><span class="p">):</span>

  266. 			<span class="n">cp</span><span class="o">.</span><span class="n">on_logout</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span>

  267. </div>

  268. <div class="viewcode-block" id="LoginManager.logout"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.LoginManager.logout">[docs]</a>	<span class="k">def</span> <span class="nf">logout</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">arg</span><span class="o">=</span><span class="s">&#39;&#39;</span><span class="p">):</span>

  269. 		<span class="bp">self</span><span class="o">.</span><span class="n">run_trigger</span><span class="p">(</span><span class="s">&#39;on_logout&#39;</span><span class="p">)</span>

  270. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="s">&#39;update tabSessions set status=&quot;Logged Out&quot; where sid=&quot;</span><span class="si">%s</span><span class="s">&quot;&#39;</span> <span class="o">%</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">session</span><span class="p">[</span><span class="s">&#39;sid&#39;</span><span class="p">])</span>

  271. 		

  272. <span class="c"># =================================================================================</span>

  273. <span class="c"># Cookie Manager</span>

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

  275. </div></div>

  276. <div class="viewcode-block" id="CookieManager"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.CookieManager">[docs]</a><span class="k">class</span> <span class="nc">CookieManager</span><span class="p">:</span>

  277. 	<span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>

  278. 		<span class="kn">import</span> <span class="nn">Cookie</span>

  279. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">cookies</span> <span class="o">=</span> <span class="n">Cookie</span><span class="o">.</span><span class="n">SimpleCookie</span><span class="p">()</span>

  280. 		<span class="bp">self</span><span class="o">.</span><span class="n">get_incoming_cookies</span><span class="p">()</span>

  281. 

  282. 	<span class="c"># get incoming cookies</span>

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

  284. <div class="viewcode-block" id="CookieManager.get_incoming_cookies"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.CookieManager.get_incoming_cookies">[docs]</a>	<span class="k">def</span> <span class="nf">get_incoming_cookies</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>

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

  286. 		<span class="n">cookies</span> <span class="o">=</span> <span class="p">{}</span>

  287. 		<span class="k">if</span> <span class="s">&#39;HTTP_COOKIE&#39;</span> <span class="ow">in</span> <span class="n">os</span><span class="o">.</span><span class="n">environ</span><span class="p">:</span>

  288. 			<span class="n">c</span> <span class="o">=</span> <span class="n">os</span><span class="o">.</span><span class="n">environ</span><span class="p">[</span><span class="s">&#39;HTTP_COOKIE&#39;</span><span class="p">]</span>

  289. 			<span class="n">webnotes</span><span class="o">.</span><span class="n">cookies</span><span class="o">.</span><span class="n">load</span><span class="p">(</span><span class="n">c</span><span class="p">)</span>

  290. 			<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">cookies</span><span class="o">.</span><span class="n">values</span><span class="p">():</span>

  291. 				<span class="n">cookies</span><span class="p">[</span><span class="n">c</span><span class="o">.</span><span class="n">key</span><span class="p">]</span> <span class="o">=</span> <span class="n">c</span><span class="o">.</span><span class="n">value</span>

  292. 					

  293. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">incoming_cookies</span> <span class="o">=</span> <span class="n">cookies</span>

  294. 		

  295. 	<span class="c"># Set cookies</span>

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

  297. 	</div>

  298. <div class="viewcode-block" id="CookieManager.set_cookies"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.CookieManager.set_cookies">[docs]</a>	<span class="k">def</span> <span class="nf">set_cookies</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>

  299. 		<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">cur_db_name</span><span class="p">:</span>

  300. 			<span class="n">webnotes</span><span class="o">.</span><span class="n">cookies</span><span class="p">[</span><span class="s">&#39;account_id&#39;</span><span class="p">]</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">cur_db_name</span>

  301. 		

  302. 		<span class="c"># ac_name	</span>

  303. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">cookies</span><span class="p">[</span><span class="s">&#39;ac_name&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">ac_name</span> <span class="ow">or</span> <span class="s">&#39;&#39;</span>

  304. 		

  305. 		<span class="k">if</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">session</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;sid&#39;</span><span class="p">):</span>

  306. 			<span class="n">webnotes</span><span class="o">.</span><span class="n">cookies</span><span class="p">[</span><span class="s">&#39;sid&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">session</span><span class="p">[</span><span class="s">&#39;sid&#39;</span><span class="p">]</span>

  307. 

  308. 			<span class="c"># sid expires in 3 days</span>

  309. 			<span class="kn">import</span> <span class="nn">datetime</span>

  310. 			<span class="n">expires</span> <span class="o">=</span> <span class="n">datetime</span><span class="o">.</span><span class="n">datetime</span><span class="o">.</span><span class="n">now</span><span class="p">()</span> <span class="o">+</span> <span class="n">datetime</span><span class="o">.</span><span class="n">timedelta</span><span class="p">(</span><span class="n">days</span><span class="o">=</span><span class="mi">3</span><span class="p">)</span>

  311. 

  312. 			<span class="n">webnotes</span><span class="o">.</span><span class="n">cookies</span><span class="p">[</span><span class="s">&#39;sid&#39;</span><span class="p">][</span><span class="s">&#39;expires&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">expires</span><span class="o">.</span><span class="n">strftime</span><span class="p">(</span><span class="s">&#39;%a, </span><span class="si">%d</span><span class="s"> %b %Y %H:%M:%S&#39;</span><span class="p">)</span>		

  313. 

  314. 	<span class="c"># Set Remember Me</span>

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

  316. </div>

  317. <div class="viewcode-block" id="CookieManager.set_remember_me"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.CookieManager.set_remember_me">[docs]</a>	<span class="k">def</span> <span class="nf">set_remember_me</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>

  318. 		<span class="k">if</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">utils</span><span class="o">.</span><span class="n">cint</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;remember_me&#39;</span><span class="p">)):</span>

  319. 			<span class="n">remember_days</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">get_value</span><span class="p">(</span><span class="s">&#39;Control Panel&#39;</span><span class="p">,</span><span class="bp">None</span><span class="p">,</span><span class="s">&#39;remember_for_days&#39;</span><span class="p">)</span> <span class="ow">or</span> <span class="mi">7</span>

  320. 			<span class="n">webnotes</span><span class="o">.</span><span class="n">cookies</span><span class="p">[</span><span class="s">&#39;remember_me&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="mi">1</span>

  321. 

  322. 			<span class="kn">import</span> <span class="nn">datetime</span>

  323. 			<span class="n">expires</span> <span class="o">=</span> <span class="n">datetime</span><span class="o">.</span><span class="n">datetime</span><span class="o">.</span><span class="n">now</span><span class="p">()</span> <span class="o">+</span> <span class="n">datetime</span><span class="o">.</span><span class="n">timedelta</span><span class="p">(</span><span class="n">days</span><span class="o">=</span><span class="n">remember_days</span><span class="p">)</span>

  324. 

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

  326. 				<span class="n">webnotes</span><span class="o">.</span><span class="n">cookies</span><span class="p">[</span><span class="n">k</span><span class="p">][</span><span class="s">&#39;expires&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">expires</span><span class="o">.</span><span class="n">strftime</span><span class="p">(</span><span class="s">&#39;%a, </span><span class="si">%d</span><span class="s"> %b %Y %H:%M:%S&#39;</span><span class="p">)</span>	

  327. 

  328. <span class="c"># =================================================================================</span>

  329. <span class="c"># Session </span>

  330. <span class="c"># =================================================================================</span>

  331. </div></div>

  332. <div class="viewcode-block" id="Session"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.Session">[docs]</a><span class="k">class</span> <span class="nc">Session</span><span class="p">:</span>

  333. 	<span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">user</span><span class="o">=</span><span class="bp">None</span><span class="p">):</span>

  334. 		<span class="bp">self</span><span class="o">.</span><span class="n">user</span> <span class="o">=</span> <span class="n">user</span>

  335. 		<span class="bp">self</span><span class="o">.</span><span class="n">sid</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;sid&#39;</span><span class="p">)</span> <span class="ow">or</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">incoming_cookies</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;sid&#39;</span><span class="p">)</span>

  336. 		<span class="bp">self</span><span class="o">.</span><span class="n">data</span> <span class="o">=</span> <span class="p">{</span><span class="s">&#39;user&#39;</span><span class="p">:</span><span class="n">user</span><span class="p">,</span><span class="s">&#39;data&#39;</span><span class="p">:{}}</span>

  337. 

  338. 		<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;cmd&#39;</span><span class="p">)</span><span class="o">==</span><span class="s">&#39;login&#39;</span><span class="p">:</span>

  339. 			<span class="bp">self</span><span class="o">.</span><span class="n">start</span><span class="p">()</span>

  340. 			<span class="k">return</span>

  341. 			

  342. 		<span class="bp">self</span><span class="o">.</span><span class="n">load</span><span class="p">()</span>

  343. 	

  344. 	<span class="c"># start a session</span>

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

  346. <div class="viewcode-block" id="Session.load"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.Session.load">[docs]</a>	<span class="k">def</span> <span class="nf">load</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>

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

  348. 		

  349. 		<span class="n">r</span><span class="o">=</span><span class="bp">None</span>

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

  351. 			<span class="n">r</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><span class="p">(</span><span class="s">&quot;select user, sessiondata, status from tabSessions where sid=&#39;</span><span class="si">%s</span><span class="s">&#39;&quot;</span> <span class="o">%</span> <span class="bp">self</span><span class="o">.</span><span class="n">sid</span><span class="p">)</span>

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

  353. 			<span class="k">if</span> <span class="n">e</span><span class="o">.</span><span class="n">args</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">==</span><span class="mi">1054</span><span class="p">:</span>

  354. 				<span class="bp">self</span><span class="o">.</span><span class="n">add_status_column</span><span class="p">()</span>

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

  356. 				<span class="k">raise</span> <span class="n">e</span>

  357. 	

  358. 		<span class="k">if</span> <span class="n">r</span><span class="p">:</span>

  359. 			<span class="n">r</span><span class="o">=</span><span class="n">r</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span>

  360. 			

  361. 			<span class="c"># ExipredSession</span>

  362. 			<span class="k">if</span> <span class="n">r</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span><span class="o">==</span><span class="s">&#39;Expired&#39;</span> <span class="ow">and</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;cmd&#39;</span><span class="p">)</span><span class="o">!=</span><span class="s">&#39;resume_session&#39;</span><span class="p">):</span>

  363. 				<span class="k">if</span> <span class="n">r</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">==</span><span class="s">&#39;Guest&#39;</span> <span class="ow">or</span> <span class="p">(</span><span class="ow">not</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;cmd&#39;</span><span class="p">))</span> <span class="ow">or</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;cmd&#39;</span><span class="p">)</span><span class="o">==</span><span class="s">&#39;logout&#39;</span><span class="p">:</span>

  364. 					<span class="n">webnotes</span><span class="o">.</span><span class="n">login_manager</span><span class="o">.</span><span class="n">login_as_guest</span><span class="p">()</span>

  365. 					<span class="bp">self</span><span class="o">.</span><span class="n">start</span><span class="p">()</span>

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

  367. 					<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;session_status&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="s">&#39;Session Expired&#39;</span>

  368. 					<span class="k">raise</span> <span class="ne">Exception</span><span class="p">,</span> <span class="s">&#39;Session Expired&#39;</span>

  369. 			<span class="k">elif</span> <span class="n">r</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span><span class="o">==</span><span class="s">&#39;Logged Out&#39;</span><span class="p">:</span>

  370. 				<span class="n">webnotes</span><span class="o">.</span><span class="n">login_manager</span><span class="o">.</span><span class="n">login_as_guest</span><span class="p">()</span>

  371. 				<span class="bp">self</span><span class="o">.</span><span class="n">start</span><span class="p">()</span>

  372. 				<span class="c"># allow refresh or logout</span>

  373. 				<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;cmd&#39;</span><span class="p">)</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;cmd&#39;</span><span class="p">)</span><span class="o">!=</span><span class="s">&#39;logout&#39;</span><span class="p">:</span>

  374. 					<span class="n">webnotes</span><span class="o">.</span><span class="n">response</span><span class="p">[</span><span class="s">&#39;session_status&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="s">&#39;Logged Out&#39;</span>

  375. 					<span class="k">raise</span> <span class="ne">Exception</span><span class="p">,</span> <span class="s">&#39;Logged Out&#39;</span>

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

  377. 				<span class="bp">self</span><span class="o">.</span><span class="n">data</span> <span class="o">=</span> <span class="p">{</span><span class="s">&#39;data&#39;</span><span class="p">:</span><span class="nb">eval</span><span class="p">(</span><span class="n">r</span><span class="p">[</span><span class="mi">1</span><span class="p">]),</span> <span class="s">&#39;user&#39;</span><span class="p">:</span><span class="n">r</span><span class="p">[</span><span class="mi">0</span><span class="p">],</span> <span class="s">&#39;sid&#39;</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">sid</span><span class="p">}</span>

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

  379. 			<span class="n">webnotes</span><span class="o">.</span><span class="n">login_manager</span><span class="o">.</span><span class="n">login_as_guest</span><span class="p">()</span>

  380. 			<span class="bp">self</span><span class="o">.</span><span class="n">start</span><span class="p">()</span>

  381. 			

  382. 

  383. 	<span class="c"># start a session</span>

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

  385. <div class="viewcode-block" id="Session.start"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.Session.start">[docs]</a>	<span class="k">def</span> <span class="nf">start</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>

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

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

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

  389. 		

  390. 		<span class="c"># generate sid</span>

  391. 		<span class="bp">self</span><span class="o">.</span><span class="n">data</span><span class="p">[</span><span class="s">&#39;user&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">login_manager</span><span class="o">.</span><span class="n">user</span>

  392. 		<span class="bp">self</span><span class="o">.</span><span class="n">data</span><span class="p">[</span><span class="s">&#39;sid&#39;</span><span class="p">]</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">generate_hash</span><span class="p">()</span>

  393. 		<span class="bp">self</span><span class="o">.</span><span class="n">data</span><span class="p">[</span><span class="s">&#39;data&#39;</span><span class="p">][</span><span class="s">&#39;session_ip&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">os</span><span class="o">.</span><span class="n">environ</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;REMOTE_ADDR&#39;</span><span class="p">);</span>

  394. 		<span class="bp">self</span><span class="o">.</span><span class="n">data</span><span class="p">[</span><span class="s">&#39;data&#39;</span><span class="p">][</span><span class="s">&#39;tenant_id&#39;</span><span class="p">]</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;tenant_id&#39;</span><span class="p">,</span> <span class="mi">0</span><span class="p">)</span>

  395. 

  396. 		<span class="c"># get ipinfo</span>

  397. 		<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">get_global</span><span class="p">(</span><span class="s">&#39;get_ip_info&#39;</span><span class="p">):</span>

  398. 			<span class="bp">self</span><span class="o">.</span><span class="n">get_ipinfo</span><span class="p">()</span>

  399. 		

  400. 		<span class="c"># insert session</span>

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

  402. 			<span class="bp">self</span><span class="o">.</span><span class="n">insert_session_record</span><span class="p">()</span>

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

  404. 			<span class="k">if</span> <span class="n">e</span><span class="o">.</span><span class="n">args</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">==</span><span class="mi">1054</span><span class="p">:</span>

  405. 				<span class="bp">self</span><span class="o">.</span><span class="n">add_status_column</span><span class="p">()</span>

  406. 				<span class="bp">self</span><span class="o">.</span><span class="n">insert_session_record</span><span class="p">()</span>

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

  408. 				<span class="k">raise</span> <span class="n">e</span>

  409. 

  410. 		<span class="c"># update profile</span>

  411. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="s">&quot;UPDATE tabProfile SET last_login = &#39;</span><span class="si">%s</span><span class="s">&#39;, last_ip = &#39;</span><span class="si">%s</span><span class="s">&#39; where name=&#39;</span><span class="si">%s</span><span class="s">&#39;&quot;</span> <span class="o">%</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">now</span><span class="p">(),</span> <span class="n">webnotes</span><span class="o">.</span><span class="n">remote_ip</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">data</span><span class="p">[</span><span class="s">&#39;user&#39;</span><span class="p">]))</span>

  412. 

  413. 		<span class="c"># set cookies to write</span>

  414. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">session</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">data</span>

  415. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">cookie_manager</span><span class="o">.</span><span class="n">set_cookies</span><span class="p">()</span>

  416. 

  417. 

  418. 	<span class="c"># resume session</span>

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

  420. <div class="viewcode-block" id="Session.resume"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.Session.resume">[docs]</a>	<span class="k">def</span> <span class="nf">resume</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>

  421. 		<span class="n">pwd</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;pwd&#39;</span><span class="p">)</span>

  422. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">login_manager</span><span class="o">.</span><span class="n">authenticate</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">data</span><span class="p">[</span><span class="s">&#39;user&#39;</span><span class="p">],</span> <span class="n">pwd</span><span class="p">)</span>

  423. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="s">&quot;update tabSessions set status=&#39;Active&#39; where sid=</span><span class="si">%s</span><span class="s">&quot;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">data</span><span class="p">[</span><span class="s">&#39;sid&#39;</span><span class="p">])</span>

  424. 		<span class="k">return</span> <span class="s">&#39;Logged In&#39;</span>

  425. 	

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

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

  428. <div class="viewcode-block" id="Session.update"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.Session.update">[docs]</a>	<span class="k">def</span> <span class="nf">update</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>

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

  430. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="s">&quot;update tabSessions set sessiondata=</span><span class="si">%s</span><span class="s">, user=</span><span class="si">%s</span><span class="s">, lastupdate=NOW() where sid=</span><span class="si">%s</span><span class="s">&quot;</span> <span class="p">,</span> <span class="p">(</span><span class="nb">str</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">data</span><span class="p">[</span><span class="s">&#39;data&#39;</span><span class="p">]),</span> <span class="bp">self</span><span class="o">.</span><span class="n">data</span><span class="p">[</span><span class="s">&#39;user&#39;</span><span class="p">],</span> <span class="bp">self</span><span class="o">.</span><span class="n">data</span><span class="p">[</span><span class="s">&#39;sid&#39;</span><span class="p">]))</span>	

  431. 

  432. 		<span class="bp">self</span><span class="o">.</span><span class="n">check_expired</span><span class="p">()</span>

  433. 

  434. 	<span class="c"># check expired</span>

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

  436. <div class="viewcode-block" id="Session.check_expired"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.Session.check_expired">[docs]</a>	<span class="k">def</span> <span class="nf">check_expired</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>

  437. 		<span class="c"># in control panel?</span>

  438. 		<span class="n">exp_sec</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">get_value</span><span class="p">(</span><span class="s">&#39;Control Panel&#39;</span><span class="p">,</span> <span class="bp">None</span><span class="p">,</span> <span class="s">&#39;session_expiry&#39;</span><span class="p">)</span> <span class="ow">or</span> <span class="s">&#39;6:00:00&#39;</span>

  439. 		

  440. 		<span class="c"># set sessions as expired</span>

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

  442. 			<span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="s">&quot;update from tabSessions where TIMEDIFF(NOW(), lastupdate) &gt; </span><span class="si">%s</span><span class="s"> SET `status`=&#39;Expired&#39;&quot;</span><span class="p">,</span> <span class="n">exp_sec</span><span class="p">)</span>

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

  444. 			<span class="k">if</span> <span class="n">e</span><span class="o">.</span><span class="n">args</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">==</span><span class="mi">1054</span><span class="p">:</span>

  445. 				<span class="bp">self</span><span class="o">.</span><span class="n">add_status_column</span><span class="p">()</span>

  446. 		

  447. 		<span class="c"># clear out old sessions</span>

  448. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="s">&quot;delete from tabSessions where TIMEDIFF(NOW(), lastupdate) &gt; &#39;72:00:00&#39;&quot;</span><span class="p">)</span>

  449. 

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

  451. <div class="viewcode-block" id="Session.add_status_column"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.Session.add_status_column">[docs]</a>	<span class="k">def</span> <span class="nf">add_status_column</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>

  452. 		<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>

  453. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="s">&quot;alter table tabSessions add column `status` varchar(20)&quot;</span><span class="p">)</span>

  454. 		<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>

  455. 

  456. 

  457. 	<span class="c"># Get IP Info from ipinfodb.com</span>

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

  459. <div class="viewcode-block" id="Session.get_ipinfo"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.Session.get_ipinfo">[docs]</a>	<span class="k">def</span> <span class="nf">get_ipinfo</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>

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

  461. 		

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

  463. 			<span class="kn">import</span> <span class="nn">pygeoip</span>

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

  465. 			<span class="k">return</span>

  466. 		

  467. 		<span class="n">gi</span> <span class="o">=</span> <span class="n">pygeoip</span><span class="o">.</span><span class="n">GeoIP</span><span class="p">(</span><span class="s">&#39;data/GeoIP.dat&#39;</span><span class="p">)</span>

  468. 		<span class="bp">self</span><span class="o">.</span><span class="n">data</span><span class="p">[</span><span class="s">&#39;data&#39;</span><span class="p">][</span><span class="s">&#39;ipinfo&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="p">{</span><span class="s">&#39;countryName&#39;</span><span class="p">:</span> <span class="n">gi</span><span class="o">.</span><span class="n">country_name_by_addr</span><span class="p">(</span><span class="n">os</span><span class="o">.</span><span class="n">environ</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">&#39;REMOTE_ADDR&#39;</span><span class="p">))}</span>

  469. 			

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

  471. <div class="viewcode-block" id="Session.insert_session_record"><a class="viewcode-back" href="../../webnotes.html#webnotes.auth.Session.insert_session_record">[docs]</a>	<span class="k">def</span> <span class="nf">insert_session_record</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>

  472. 		<span class="n">webnotes</span><span class="o">.</span><span class="n">conn</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="s">&quot;insert into tabSessions (sessiondata, user, lastupdate, sid, status) values (</span><span class="si">%s</span><span class="s"> , </span><span class="si">%s</span><span class="s">, NOW(), </span><span class="si">%s</span><span class="s">, &#39;Active&#39;)&quot;</span><span class="p">,</span> <span class="p">(</span><span class="nb">str</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">data</span><span class="p">[</span><span class="s">&#39;data&#39;</span><span class="p">]),</span> <span class="bp">self</span><span class="o">.</span><span class="n">data</span><span class="p">[</span><span class="s">&#39;user&#39;</span><span class="p">],</span> <span class="bp">self</span><span class="o">.</span><span class="n">data</span><span class="p">[</span><span class="s">&#39;sid&#39;</span><span class="p">]))</span>

  473. 		</div></div>

  474. </pre></div>

  475. 

  476.           </div>

  477.         </div>

  478.       </div>

  479.       <div class="sphinxsidebar">

  480.         <div class="sphinxsidebarwrapper">

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

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

  483.             </a></p>

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

  485.   <h3>Quick search</h3>

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

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

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

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

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

  491.     </form>

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

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

  494.     </p>

  495. </div>

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

  497.         </div>

  498.       </div>

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

  500.     </div>

  501.     <div class="related">

  502.       <h3>Navigation</h3>

  503.       <ul>

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

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

  506.              >index</a></li>

  507.         <li class="right" >

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

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

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

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

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

  513.       </ul>

  514.     </div>

  515.     <div class="footer">

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

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

  518.     </div>

  519.   </body>

  520. </html>