cache.html 61 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560
  1. <!DOCTYPE html>
  2. <html lang="en" data-content_root="../../">
  3. <head>
  4. <meta charset="utf-8" />
  5. <meta name="viewport" content="width=device-width, initial-scale=1.0" />
  6. <meta name="viewport" content="width=device-width, initial-scale=1">
  7. <title>searx.cache &#8212; SearXNG Documentation (2025.5.5+0315988f5)</title>
  8. <link rel="stylesheet" type="text/css" href="../../_static/pygments.css?v=6625fa76" />
  9. <link rel="stylesheet" type="text/css" href="../../_static/searxng.css?v=52e4ff28" />
  10. <script src="../../_static/documentation_options.js?v=6bec52eb"></script>
  11. <script src="../../_static/doctools.js?v=9a2dae69"></script>
  12. <script src="../../_static/sphinx_highlight.js?v=dc90522c"></script>
  13. <script data-project="searxng" data-version="2025.5.5+0315988f5" src="../../_static/describe_version.js?v=fa7f30d0"></script>
  14. <link rel="index" title="Index" href="../../genindex.html" />
  15. <link rel="search" title="Search" href="../../search.html" />
  16. </head><body>
  17. <div class="related" role="navigation" aria-label="Related">
  18. <h3>Navigation</h3>
  19. <ul>
  20. <li class="right" style="margin-right: 10px">
  21. <a href="../../genindex.html" title="General Index"
  22. accesskey="I">index</a></li>
  23. <li class="right" >
  24. <a href="../../py-modindex.html" title="Python Module Index"
  25. >modules</a> |</li>
  26. <li class="nav-item nav-item-0"><a href="../../index.html">SearXNG Documentation (2025.5.5+0315988f5)</a> &#187;</li>
  27. <li class="nav-item nav-item-1"><a href="../index.html" accesskey="U">Module code</a> &#187;</li>
  28. <li class="nav-item nav-item-this"><a href="">searx.cache</a></li>
  29. </ul>
  30. </div>
  31. <div class="document">
  32. <div class="documentwrapper">
  33. <div class="bodywrapper">
  34. <div class="body" role="main">
  35. <h1>Source code for searx.cache</h1><div class="highlight"><pre>
  36. <span></span><span class="sd">&quot;&quot;&quot;Implementation of caching solutions.</span>
  37. <span class="sd">- :py:obj:`searx.cache.ExpireCache` and its :py:obj:`searx.cache.ExpireCacheCfg`</span>
  38. <span class="sd">----</span>
  39. <span class="sd">&quot;&quot;&quot;</span>
  40. <span class="kn">from</span><span class="w"> </span><span class="nn">__future__</span><span class="w"> </span><span class="kn">import</span> <span class="n">annotations</span>
  41. <span class="n">__all__</span> <span class="o">=</span> <span class="p">[</span><span class="s2">&quot;ExpireCacheCfg&quot;</span><span class="p">,</span> <span class="s2">&quot;ExpireCacheStats&quot;</span><span class="p">,</span> <span class="s2">&quot;ExpireCache&quot;</span><span class="p">,</span> <span class="s2">&quot;ExpireCacheSQLite&quot;</span><span class="p">]</span>
  42. <span class="kn">import</span><span class="w"> </span><span class="nn">abc</span>
  43. <span class="kn">import</span><span class="w"> </span><span class="nn">dataclasses</span>
  44. <span class="kn">import</span><span class="w"> </span><span class="nn">datetime</span>
  45. <span class="kn">import</span><span class="w"> </span><span class="nn">hashlib</span>
  46. <span class="kn">import</span><span class="w"> </span><span class="nn">hmac</span>
  47. <span class="kn">import</span><span class="w"> </span><span class="nn">os</span>
  48. <span class="kn">import</span><span class="w"> </span><span class="nn">pickle</span>
  49. <span class="kn">import</span><span class="w"> </span><span class="nn">sqlite3</span>
  50. <span class="kn">import</span><span class="w"> </span><span class="nn">string</span>
  51. <span class="kn">import</span><span class="w"> </span><span class="nn">tempfile</span>
  52. <span class="kn">import</span><span class="w"> </span><span class="nn">time</span>
  53. <span class="kn">import</span><span class="w"> </span><span class="nn">typing</span>
  54. <span class="kn">import</span><span class="w"> </span><span class="nn">msgspec</span>
  55. <span class="kn">from</span><span class="w"> </span><span class="nn">searx</span><span class="w"> </span><span class="kn">import</span> <span class="n">sqlitedb</span>
  56. <span class="kn">from</span><span class="w"> </span><span class="nn">searx</span><span class="w"> </span><span class="kn">import</span> <span class="n">logger</span>
  57. <span class="kn">from</span><span class="w"> </span><span class="nn">searx</span><span class="w"> </span><span class="kn">import</span> <span class="n">get_setting</span>
  58. <span class="n">log</span> <span class="o">=</span> <span class="n">logger</span><span class="o">.</span><span class="n">getChild</span><span class="p">(</span><span class="s2">&quot;cache&quot;</span><span class="p">)</span>
  59. <div class="viewcode-block" id="ExpireCacheCfg">
  60. <a class="viewcode-back" href="../../src/searx.cache.html#searx.cache.ExpireCacheCfg">[docs]</a>
  61. <span class="k">class</span><span class="w"> </span><span class="nc">ExpireCacheCfg</span><span class="p">(</span><span class="n">msgspec</span><span class="o">.</span><span class="n">Struct</span><span class="p">):</span> <span class="c1"># pylint: disable=too-few-public-methods</span>
  62. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Configuration of a :py:obj:`ExpireCache` cache.&quot;&quot;&quot;</span>
  63. <span class="n">name</span><span class="p">:</span> <span class="nb">str</span>
  64. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Name of the cache.&quot;&quot;&quot;</span>
  65. <span class="n">db_url</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">&quot;&quot;</span>
  66. <span class="w"> </span><span class="sd">&quot;&quot;&quot;URL of the SQLite DB, the path to the database file. If unset a default</span>
  67. <span class="sd"> DB will be created in `/tmp/sxng_cache_{self.name}.db`&quot;&quot;&quot;</span>
  68. <span class="n">MAX_VALUE_LEN</span><span class="p">:</span> <span class="nb">int</span> <span class="o">=</span> <span class="mi">1024</span> <span class="o">*</span> <span class="mi">10</span>
  69. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Max lenght of a *serialized* value.&quot;&quot;&quot;</span>
  70. <span class="n">MAXHOLD_TIME</span><span class="p">:</span> <span class="nb">int</span> <span class="o">=</span> <span class="mi">60</span> <span class="o">*</span> <span class="mi">60</span> <span class="o">*</span> <span class="mi">24</span> <span class="o">*</span> <span class="mi">7</span> <span class="c1"># 7 days</span>
  71. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Hold time (default in sec.), after which a value is removed from the cache.&quot;&quot;&quot;</span>
  72. <span class="n">MAINTENANCE_PERIOD</span><span class="p">:</span> <span class="nb">int</span> <span class="o">=</span> <span class="mi">60</span> <span class="o">*</span> <span class="mi">60</span> <span class="c1"># 2h</span>
  73. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Maintenance period in seconds / when :py:obj:`MAINTENANCE_MODE` is set to</span>
  74. <span class="sd"> ``auto``.&quot;&quot;&quot;</span>
  75. <span class="n">MAINTENANCE_MODE</span><span class="p">:</span> <span class="n">typing</span><span class="o">.</span><span class="n">Literal</span><span class="p">[</span><span class="s2">&quot;auto&quot;</span><span class="p">,</span> <span class="s2">&quot;off&quot;</span><span class="p">]</span> <span class="o">=</span> <span class="s2">&quot;auto&quot;</span>
  76. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Type of maintenance mode</span>
  77. <span class="sd"> ``auto``:</span>
  78. <span class="sd"> Maintenance is carried out automatically as part of the maintenance</span>
  79. <span class="sd"> intervals (:py:obj:`MAINTENANCE_PERIOD`); no external process is required.</span>
  80. <span class="sd"> ``off``:</span>
  81. <span class="sd"> Maintenance is switched off and must be carried out by an external process</span>
  82. <span class="sd"> if required.</span>
  83. <span class="sd"> &quot;&quot;&quot;</span>
  84. <span class="n">password</span><span class="p">:</span> <span class="nb">bytes</span> <span class="o">=</span> <span class="n">get_setting</span><span class="p">(</span><span class="s2">&quot;server.secret_key&quot;</span><span class="p">)</span><span class="o">.</span><span class="n">encode</span><span class="p">()</span> <span class="c1"># type: ignore</span>
  85. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Password used by :py:obj:`ExpireCache.secret_hash`.</span>
  86. <span class="sd"> The default password is taken from :ref:`secret_key &lt;server.secret_key&gt;`.</span>
  87. <span class="sd"> When the password is changed, the hashed keys in the cache can no longer be</span>
  88. <span class="sd"> used, which is why all values in the cache are deleted when the password is</span>
  89. <span class="sd"> changed.</span>
  90. <span class="sd"> &quot;&quot;&quot;</span>
  91. <span class="k">def</span><span class="w"> </span><span class="nf">__post_init__</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  92. <span class="c1"># if db_url is unset, use a default DB in /tmp/sxng_cache_{name}.db</span>
  93. <span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">db_url</span><span class="p">:</span>
  94. <span class="bp">self</span><span class="o">.</span><span class="n">db_url</span> <span class="o">=</span> <span class="n">tempfile</span><span class="o">.</span><span class="n">gettempdir</span><span class="p">()</span> <span class="o">+</span> <span class="n">os</span><span class="o">.</span><span class="n">sep</span> <span class="o">+</span> <span class="sa">f</span><span class="s2">&quot;sxng_cache_</span><span class="si">{</span><span class="n">ExpireCache</span><span class="o">.</span><span class="n">normalize_name</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">name</span><span class="p">)</span><span class="si">}</span><span class="s2">.db&quot;</span></div>
  95. <div class="viewcode-block" id="ExpireCacheStats">
  96. <a class="viewcode-back" href="../../src/searx.cache.html#searx.cache.ExpireCacheStats">[docs]</a>
  97. <span class="nd">@dataclasses</span><span class="o">.</span><span class="n">dataclass</span>
  98. <span class="k">class</span><span class="w"> </span><span class="nc">ExpireCacheStats</span><span class="p">:</span>
  99. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Dataclass wich provides information on the status of the cache.&quot;&quot;&quot;</span>
  100. <span class="n">cached_items</span><span class="p">:</span> <span class="nb">dict</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="nb">list</span><span class="p">[</span><span class="nb">tuple</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="n">typing</span><span class="o">.</span><span class="n">Any</span><span class="p">,</span> <span class="nb">int</span><span class="p">]]]</span>
  101. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Values in the cache mapped by context name.</span>
  102. <span class="sd"> .. code: python</span>
  103. <span class="sd"> {</span>
  104. <span class="sd"> &quot;context name&quot;: [</span>
  105. <span class="sd"> (&quot;foo key&quot;: &quot;foo value&quot;, &lt;expire&gt;),</span>
  106. <span class="sd"> (&quot;bar key&quot;: &quot;bar value&quot;, &lt;expire&gt;),</span>
  107. <span class="sd"> # ...</span>
  108. <span class="sd"> ],</span>
  109. <span class="sd"> # ...</span>
  110. <span class="sd"> }</span>
  111. <span class="sd"> &quot;&quot;&quot;</span>
  112. <span class="k">def</span><span class="w"> </span><span class="nf">report</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  113. <span class="n">c_ctx</span> <span class="o">=</span> <span class="mi">0</span>
  114. <span class="n">c_kv</span> <span class="o">=</span> <span class="mi">0</span>
  115. <span class="n">lines</span> <span class="o">=</span> <span class="p">[]</span>
  116. <span class="k">for</span> <span class="n">ctx_name</span><span class="p">,</span> <span class="n">kv_list</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">cached_items</span><span class="o">.</span><span class="n">items</span><span class="p">():</span>
  117. <span class="n">c_ctx</span> <span class="o">+=</span> <span class="mi">1</span>
  118. <span class="k">if</span> <span class="ow">not</span> <span class="n">kv_list</span><span class="p">:</span>
  119. <span class="n">lines</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="sa">f</span><span class="s2">&quot;[</span><span class="si">{</span><span class="n">ctx_name</span><span class="si">:</span><span class="s2">20s</span><span class="si">}</span><span class="s2">] empty&quot;</span><span class="p">)</span>
  120. <span class="k">continue</span>
  121. <span class="k">for</span> <span class="n">key</span><span class="p">,</span> <span class="n">value</span><span class="p">,</span> <span class="n">expire</span> <span class="ow">in</span> <span class="n">kv_list</span><span class="p">:</span>
  122. <span class="n">valid_until</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">fromtimestamp</span><span class="p">(</span><span class="n">expire</span><span class="p">)</span><span class="o">.</span><span class="n">strftime</span><span class="p">(</span><span class="s2">&quot;%Y-%m-</span><span class="si">%d</span><span class="s2"> %H:%M:%S&quot;</span><span class="p">)</span>
  123. <span class="n">c_kv</span> <span class="o">+=</span> <span class="mi">1</span>
  124. <span class="n">lines</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="sa">f</span><span class="s2">&quot;[</span><span class="si">{</span><span class="n">ctx_name</span><span class="si">:</span><span class="s2">20s</span><span class="si">}</span><span class="s2">] </span><span class="si">{</span><span class="n">valid_until</span><span class="si">}</span><span class="s2"> </span><span class="si">{</span><span class="n">key</span><span class="si">:</span><span class="s2">12</span><span class="si">}</span><span class="s2">&quot;</span> <span class="sa">f</span><span class="s2">&quot; --&gt; (</span><span class="si">{</span><span class="nb">type</span><span class="p">(</span><span class="n">value</span><span class="p">)</span><span class="o">.</span><span class="vm">__name__</span><span class="si">}</span><span class="s2">) </span><span class="si">{</span><span class="n">value</span><span class="si">}</span><span class="s2"> &quot;</span><span class="p">)</span>
  125. <span class="n">lines</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="sa">f</span><span class="s2">&quot;Number of contexts: </span><span class="si">{</span><span class="n">c_ctx</span><span class="si">}</span><span class="s2">&quot;</span><span class="p">)</span>
  126. <span class="n">lines</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="sa">f</span><span class="s2">&quot;number of key/value pairs: </span><span class="si">{</span><span class="n">c_kv</span><span class="si">}</span><span class="s2">&quot;</span><span class="p">)</span>
  127. <span class="k">return</span> <span class="s2">&quot;</span><span class="se">\n</span><span class="s2">&quot;</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">lines</span><span class="p">)</span></div>
  128. <div class="viewcode-block" id="ExpireCache">
  129. <a class="viewcode-back" href="../../src/searx.cache.html#searx.cache.ExpireCache">[docs]</a>
  130. <span class="k">class</span><span class="w"> </span><span class="nc">ExpireCache</span><span class="p">(</span><span class="n">abc</span><span class="o">.</span><span class="n">ABC</span><span class="p">):</span>
  131. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Abstract base class for the implementation of a key/value cache</span>
  132. <span class="sd"> with expire date.&quot;&quot;&quot;</span>
  133. <span class="n">cfg</span><span class="p">:</span> <span class="n">ExpireCacheCfg</span>
  134. <span class="n">hash_token</span> <span class="o">=</span> <span class="s2">&quot;hash_token&quot;</span>
  135. <div class="viewcode-block" id="ExpireCache.set">
  136. <a class="viewcode-back" href="../../src/searx.cache.html#searx.cache.ExpireCache.set">[docs]</a>
  137. <span class="nd">@abc</span><span class="o">.</span><span class="n">abstractmethod</span>
  138. <span class="k">def</span><span class="w"> </span><span class="nf">set</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">key</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">value</span><span class="p">:</span> <span class="n">typing</span><span class="o">.</span><span class="n">Any</span><span class="p">,</span> <span class="n">expire</span><span class="p">:</span> <span class="nb">int</span> <span class="o">|</span> <span class="kc">None</span><span class="p">,</span> <span class="n">ctx</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">bool</span><span class="p">:</span>
  139. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Set *key* to *value*. To set a timeout on key use argument</span>
  140. <span class="sd"> ``expire`` (in sec.). If expire is unset the default is taken from</span>
  141. <span class="sd"> :py:obj:`ExpireCacheCfg.MAXHOLD_TIME`. After the timeout has expired,</span>
  142. <span class="sd"> the key will automatically be deleted.</span>
  143. <span class="sd"> The ``ctx`` argument specifies the context of the ``key``. A key is</span>
  144. <span class="sd"> only unique in its context.</span>
  145. <span class="sd"> The concrete implementations of this abstraction determine how the</span>
  146. <span class="sd"> context is mapped in the connected database. In SQL databases, for</span>
  147. <span class="sd"> example, the context is a DB table or in a Key/Value DB it could be</span>
  148. <span class="sd"> a prefix for the key.</span>
  149. <span class="sd"> If the context is not specified (the default is ``None``) then a</span>
  150. <span class="sd"> default context should be used, e.g. a default table for SQL databases</span>
  151. <span class="sd"> or a default prefix in a Key/Value DB.</span>
  152. <span class="sd"> &quot;&quot;&quot;</span></div>
  153. <div class="viewcode-block" id="ExpireCache.get">
  154. <a class="viewcode-back" href="../../src/searx.cache.html#searx.cache.ExpireCache.get">[docs]</a>
  155. <span class="nd">@abc</span><span class="o">.</span><span class="n">abstractmethod</span>
  156. <span class="k">def</span><span class="w"> </span><span class="nf">get</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">key</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">default</span><span class="o">=</span><span class="kc">None</span><span class="p">,</span> <span class="n">ctx</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">typing</span><span class="o">.</span><span class="n">Any</span><span class="p">:</span>
  157. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Return *value* of *key*. If key is unset, ``None`` is returned.&quot;&quot;&quot;</span></div>
  158. <div class="viewcode-block" id="ExpireCache.maintenance">
  159. <a class="viewcode-back" href="../../src/searx.cache.html#searx.cache.ExpireCache.maintenance">[docs]</a>
  160. <span class="nd">@abc</span><span class="o">.</span><span class="n">abstractmethod</span>
  161. <span class="k">def</span><span class="w"> </span><span class="nf">maintenance</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">force</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span> <span class="n">truncate</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">bool</span><span class="p">:</span>
  162. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Performs maintenance on the cache.</span>
  163. <span class="sd"> ``force``:</span>
  164. <span class="sd"> Maintenance should be carried out even if the maintenance interval has</span>
  165. <span class="sd"> not yet been reached.</span>
  166. <span class="sd"> ``truncate``:</span>
  167. <span class="sd"> Truncate the entire cache, which is necessary, for example, if the</span>
  168. <span class="sd"> password has changed.</span>
  169. <span class="sd"> &quot;&quot;&quot;</span></div>
  170. <div class="viewcode-block" id="ExpireCache.state">
  171. <a class="viewcode-back" href="../../src/searx.cache.html#searx.cache.ExpireCache.state">[docs]</a>
  172. <span class="nd">@abc</span><span class="o">.</span><span class="n">abstractmethod</span>
  173. <span class="k">def</span><span class="w"> </span><span class="nf">state</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">ExpireCacheStats</span><span class="p">:</span>
  174. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Returns a :py:obj:`ExpireCacheStats`, which provides information</span>
  175. <span class="sd"> about the status of the cache.&quot;&quot;&quot;</span></div>
  176. <div class="viewcode-block" id="ExpireCache.build_cache">
  177. <a class="viewcode-back" href="../../src/searx.cache.html#searx.cache.ExpireCache.build_cache">[docs]</a>
  178. <span class="nd">@staticmethod</span>
  179. <span class="k">def</span><span class="w"> </span><span class="nf">build_cache</span><span class="p">(</span><span class="n">cfg</span><span class="p">:</span> <span class="n">ExpireCacheCfg</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">ExpireCache</span><span class="p">:</span>
  180. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Factory to build a caching instance.</span>
  181. <span class="sd"> .. note::</span>
  182. <span class="sd"> Currently, only the SQLite adapter is available, but other database</span>
  183. <span class="sd"> types could be implemented in the future, e.g. a Valkey (Redis)</span>
  184. <span class="sd"> adapter.</span>
  185. <span class="sd"> &quot;&quot;&quot;</span>
  186. <span class="k">return</span> <span class="n">ExpireCacheSQLite</span><span class="p">(</span><span class="n">cfg</span><span class="p">)</span></div>
  187. <div class="viewcode-block" id="ExpireCache.normalize_name">
  188. <a class="viewcode-back" href="../../src/searx.cache.html#searx.cache.ExpireCache.normalize_name">[docs]</a>
  189. <span class="nd">@staticmethod</span>
  190. <span class="k">def</span><span class="w"> </span><span class="nf">normalize_name</span><span class="p">(</span><span class="n">name</span><span class="p">:</span> <span class="nb">str</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">str</span><span class="p">:</span>
  191. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Returns a normalized name that can be used as a file name or as a SQL</span>
  192. <span class="sd"> table name (is used, for example, to normalize the context name).&quot;&quot;&quot;</span>
  193. <span class="n">_valid</span> <span class="o">=</span> <span class="s2">&quot;-_.&quot;</span> <span class="o">+</span> <span class="n">string</span><span class="o">.</span><span class="n">ascii_letters</span> <span class="o">+</span> <span class="n">string</span><span class="o">.</span><span class="n">digits</span>
  194. <span class="k">return</span> <span class="s2">&quot;&quot;</span><span class="o">.</span><span class="n">join</span><span class="p">([</span><span class="n">c</span> <span class="k">for</span> <span class="n">c</span> <span class="ow">in</span> <span class="n">name</span> <span class="k">if</span> <span class="n">c</span> <span class="ow">in</span> <span class="n">_valid</span><span class="p">])</span></div>
  195. <span class="k">def</span><span class="w"> </span><span class="nf">serialize</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="p">:</span> <span class="n">typing</span><span class="o">.</span><span class="n">Any</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">bytes</span><span class="p">:</span>
  196. <span class="n">dump</span><span class="p">:</span> <span class="nb">bytes</span> <span class="o">=</span> <span class="n">pickle</span><span class="o">.</span><span class="n">dumps</span><span class="p">(</span><span class="n">value</span><span class="p">)</span>
  197. <span class="k">return</span> <span class="n">dump</span>
  198. <span class="k">def</span><span class="w"> </span><span class="nf">deserialize</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="p">:</span> <span class="nb">bytes</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">typing</span><span class="o">.</span><span class="n">Any</span><span class="p">:</span>
  199. <span class="n">obj</span> <span class="o">=</span> <span class="n">pickle</span><span class="o">.</span><span class="n">loads</span><span class="p">(</span><span class="n">value</span><span class="p">)</span>
  200. <span class="k">return</span> <span class="n">obj</span>
  201. <div class="viewcode-block" id="ExpireCache.secret_hash">
  202. <a class="viewcode-back" href="../../src/searx.cache.html#searx.cache.ExpireCache.secret_hash">[docs]</a>
  203. <span class="k">def</span><span class="w"> </span><span class="nf">secret_hash</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">name</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="nb">bytes</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">str</span><span class="p">:</span>
  204. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Creates a hash of the argument ``name``. The hash value is formed</span>
  205. <span class="sd"> from the ``name`` combined with the :py:obj:`password</span>
  206. <span class="sd"> &lt;ExpireCacheCfg.password&gt;`. Can be used, for example, to make the</span>
  207. <span class="sd"> ``key`` stored in the DB unreadable for third parties.&quot;&quot;&quot;</span>
  208. <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">name</span><span class="p">,</span> <span class="nb">str</span><span class="p">):</span>
  209. <span class="n">name</span> <span class="o">=</span> <span class="nb">bytes</span><span class="p">(</span><span class="n">name</span><span class="p">,</span> <span class="n">encoding</span><span class="o">=</span><span class="s1">&#39;utf-8&#39;</span><span class="p">)</span>
  210. <span class="n">m</span> <span class="o">=</span> <span class="n">hmac</span><span class="o">.</span><span class="n">new</span><span class="p">(</span><span class="n">name</span> <span class="o">+</span> <span class="bp">self</span><span class="o">.</span><span class="n">cfg</span><span class="o">.</span><span class="n">password</span><span class="p">,</span> <span class="n">digestmod</span><span class="o">=</span><span class="s1">&#39;sha256&#39;</span><span class="p">)</span>
  211. <span class="k">return</span> <span class="n">m</span><span class="o">.</span><span class="n">hexdigest</span><span class="p">()</span></div>
  212. </div>
  213. <div class="viewcode-block" id="ExpireCacheSQLite">
  214. <a class="viewcode-back" href="../../src/searx.cache.html#searx.cache.ExpireCacheSQLite">[docs]</a>
  215. <span class="k">class</span><span class="w"> </span><span class="nc">ExpireCacheSQLite</span><span class="p">(</span><span class="n">sqlitedb</span><span class="o">.</span><span class="n">SQLiteAppl</span><span class="p">,</span> <span class="n">ExpireCache</span><span class="p">):</span>
  216. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Cache that manages key/value pairs in a SQLite DB. The DB model in the</span>
  217. <span class="sd"> SQLite DB is implemented in abstract class :py:obj:`SQLiteAppl</span>
  218. <span class="sd"> &lt;searx.sqlitedb.SQLiteAppl&gt;`.</span>
  219. <span class="sd"> The following configurations are required / supported:</span>
  220. <span class="sd"> - :py:obj:`ExpireCacheCfg.db_url`</span>
  221. <span class="sd"> - :py:obj:`ExpireCacheCfg.MAXHOLD_TIME`</span>
  222. <span class="sd"> - :py:obj:`ExpireCacheCfg.MAINTENANCE_PERIOD`</span>
  223. <span class="sd"> - :py:obj:`ExpireCacheCfg.MAINTENANCE_MODE`</span>
  224. <span class="sd"> &quot;&quot;&quot;</span>
  225. <span class="n">DB_SCHEMA</span> <span class="o">=</span> <span class="mi">1</span>
  226. <span class="c1"># The key/value tables will be created on demand by self.create_table</span>
  227. <span class="n">DDL_CREATE_TABLES</span> <span class="o">=</span> <span class="p">{}</span>
  228. <span class="n">CACHE_TABLE_PREFIX</span> <span class="o">=</span> <span class="s2">&quot;CACHE-TABLE-&quot;</span>
  229. <span class="k">def</span><span class="w"> </span><span class="fm">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">cfg</span><span class="p">:</span> <span class="n">ExpireCacheCfg</span><span class="p">):</span>
  230. <span class="w"> </span><span class="sd">&quot;&quot;&quot;An instance of the SQLite expire cache is build up from a</span>
  231. <span class="sd"> :py:obj:`config &lt;ExpireCacheCfg&gt;`.&quot;&quot;&quot;</span>
  232. <span class="bp">self</span><span class="o">.</span><span class="n">cfg</span> <span class="o">=</span> <span class="n">cfg</span>
  233. <span class="k">if</span> <span class="n">cfg</span><span class="o">.</span><span class="n">db_url</span> <span class="o">==</span> <span class="s2">&quot;:memory:&quot;</span><span class="p">:</span>
  234. <span class="n">log</span><span class="o">.</span><span class="n">critical</span><span class="p">(</span><span class="s2">&quot;don&#39;t use SQLite DB in :memory: in production!!&quot;</span><span class="p">)</span>
  235. <span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="n">cfg</span><span class="o">.</span><span class="n">db_url</span><span class="p">)</span>
  236. <div class="viewcode-block" id="ExpireCacheSQLite.init">
  237. <a class="viewcode-back" href="../../src/searx.cache.html#searx.cache.ExpireCacheSQLite.init">[docs]</a>
  238. <span class="k">def</span><span class="w"> </span><span class="nf">init</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">conn</span><span class="p">:</span> <span class="n">sqlite3</span><span class="o">.</span><span class="n">Connection</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">bool</span><span class="p">:</span>
  239. <span class="n">ret_val</span> <span class="o">=</span> <span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="n">init</span><span class="p">(</span><span class="n">conn</span><span class="p">)</span>
  240. <span class="k">if</span> <span class="ow">not</span> <span class="n">ret_val</span><span class="p">:</span>
  241. <span class="k">return</span> <span class="kc">False</span>
  242. <span class="n">new</span> <span class="o">=</span> <span class="n">hashlib</span><span class="o">.</span><span class="n">sha256</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">cfg</span><span class="o">.</span><span class="n">password</span><span class="p">)</span><span class="o">.</span><span class="n">hexdigest</span><span class="p">()</span>
  243. <span class="n">old</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">properties</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">hash_token</span><span class="p">)</span>
  244. <span class="k">if</span> <span class="n">old</span> <span class="o">!=</span> <span class="n">new</span><span class="p">:</span>
  245. <span class="k">if</span> <span class="n">old</span> <span class="ow">is</span> <span class="ow">not</span> <span class="kc">None</span><span class="p">:</span>
  246. <span class="n">log</span><span class="o">.</span><span class="n">warning</span><span class="p">(</span><span class="s2">&quot;[</span><span class="si">%s</span><span class="s2">] hash token changed: truncate all cache tables&quot;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">cfg</span><span class="o">.</span><span class="n">name</span><span class="p">)</span>
  247. <span class="bp">self</span><span class="o">.</span><span class="n">maintenance</span><span class="p">(</span><span class="n">force</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> <span class="n">truncate</span><span class="o">=</span><span class="kc">True</span><span class="p">)</span>
  248. <span class="bp">self</span><span class="o">.</span><span class="n">properties</span><span class="o">.</span><span class="n">set</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">hash_token</span><span class="p">,</span> <span class="n">new</span><span class="p">)</span>
  249. <span class="k">return</span> <span class="kc">True</span></div>
  250. <div class="viewcode-block" id="ExpireCacheSQLite.maintenance">
  251. <a class="viewcode-back" href="../../src/searx.cache.html#searx.cache.ExpireCacheSQLite.maintenance">[docs]</a>
  252. <span class="k">def</span><span class="w"> </span><span class="nf">maintenance</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">force</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span> <span class="n">truncate</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">bool</span><span class="p">:</span>
  253. <span class="k">if</span> <span class="ow">not</span> <span class="n">force</span> <span class="ow">and</span> <span class="nb">int</span><span class="p">(</span><span class="n">time</span><span class="o">.</span><span class="n">time</span><span class="p">())</span> <span class="o">&lt;</span> <span class="bp">self</span><span class="o">.</span><span class="n">next_maintenance_time</span><span class="p">:</span>
  254. <span class="c1"># log.debug(&quot;no maintenance required yet, next maintenance interval is in the future&quot;)</span>
  255. <span class="k">return</span> <span class="kc">False</span>
  256. <span class="c1"># Prevent parallel DB maintenance cycles from other DB connections</span>
  257. <span class="c1"># (e.g. in multi thread or process environments).</span>
  258. <span class="bp">self</span><span class="o">.</span><span class="n">properties</span><span class="o">.</span><span class="n">set</span><span class="p">(</span><span class="s2">&quot;LAST_MAINTENANCE&quot;</span><span class="p">,</span> <span class="s2">&quot;&quot;</span><span class="p">)</span> <span class="c1"># hint: this (also) sets the m_time of the property!</span>
  259. <span class="k">if</span> <span class="n">truncate</span><span class="p">:</span>
  260. <span class="bp">self</span><span class="o">.</span><span class="n">truncate_tables</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">table_names</span><span class="p">)</span>
  261. <span class="k">return</span> <span class="kc">True</span>
  262. <span class="c1"># drop items by expire time stamp ..</span>
  263. <span class="n">expire</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">time</span><span class="o">.</span><span class="n">time</span><span class="p">())</span>
  264. <span class="k">with</span> <span class="bp">self</span><span class="o">.</span><span class="n">connect</span><span class="p">()</span> <span class="k">as</span> <span class="n">conn</span><span class="p">:</span>
  265. <span class="k">for</span> <span class="n">table</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">table_names</span><span class="p">:</span>
  266. <span class="n">res</span> <span class="o">=</span> <span class="n">conn</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="sa">f</span><span class="s2">&quot;DELETE FROM </span><span class="si">{</span><span class="n">table</span><span class="si">}</span><span class="s2"> WHERE expire &lt; ?&quot;</span><span class="p">,</span> <span class="p">(</span><span class="n">expire</span><span class="p">,))</span>
  267. <span class="n">log</span><span class="o">.</span><span class="n">debug</span><span class="p">(</span><span class="s2">&quot;deleted </span><span class="si">%s</span><span class="s2"> keys from table </span><span class="si">%s</span><span class="s2"> (expire date reached)&quot;</span><span class="p">,</span> <span class="n">res</span><span class="o">.</span><span class="n">rowcount</span><span class="p">,</span> <span class="n">table</span><span class="p">)</span>
  268. <span class="c1"># Vacuuming the WALs</span>
  269. <span class="c1"># https://www.theunterminatedstring.com/sqlite-vacuuming/</span>
  270. <span class="n">conn</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="s2">&quot;PRAGMA wal_checkpoint(TRUNCATE)&quot;</span><span class="p">)</span>
  271. <span class="n">conn</span><span class="o">.</span><span class="n">close</span><span class="p">()</span>
  272. <span class="k">return</span> <span class="kc">True</span></div>
  273. <div class="viewcode-block" id="ExpireCacheSQLite.create_table">
  274. <a class="viewcode-back" href="../../src/searx.cache.html#searx.cache.ExpireCacheSQLite.create_table">[docs]</a>
  275. <span class="k">def</span><span class="w"> </span><span class="nf">create_table</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">table</span><span class="p">:</span> <span class="nb">str</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">bool</span><span class="p">:</span>
  276. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Create DB ``table`` if it has not yet been created, no recreates are</span>
  277. <span class="sd"> initiated if the table already exists.</span>
  278. <span class="sd"> &quot;&quot;&quot;</span>
  279. <span class="k">if</span> <span class="n">table</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">table_names</span><span class="p">:</span>
  280. <span class="c1"># log.debug(&quot;key/value table %s exists in DB (no need to recreate)&quot;, table)</span>
  281. <span class="k">return</span> <span class="kc">False</span>
  282. <span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;key/value table &#39;</span><span class="si">%s</span><span class="s2">&#39; NOT exists in DB -&gt; create DB table ..&quot;</span><span class="p">,</span> <span class="n">table</span><span class="p">)</span>
  283. <span class="n">sql_table</span> <span class="o">=</span> <span class="s2">&quot;</span><span class="se">\n</span><span class="s2">&quot;</span><span class="o">.</span><span class="n">join</span><span class="p">(</span>
  284. <span class="p">[</span>
  285. <span class="sa">f</span><span class="s2">&quot;CREATE TABLE IF NOT EXISTS </span><span class="si">{</span><span class="n">table</span><span class="si">}</span><span class="s2"> (&quot;</span><span class="p">,</span>
  286. <span class="s2">&quot; key TEXT,&quot;</span><span class="p">,</span>
  287. <span class="s2">&quot; value BLOB,&quot;</span><span class="p">,</span>
  288. <span class="sa">f</span><span class="s2">&quot; expire INTEGER DEFAULT (strftime(&#39;%s&#39;, &#39;now&#39;) + </span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">cfg</span><span class="o">.</span><span class="n">MAXHOLD_TIME</span><span class="si">}</span><span class="s2">),&quot;</span><span class="p">,</span>
  289. <span class="s2">&quot;PRIMARY KEY (key))&quot;</span><span class="p">,</span>
  290. <span class="p">]</span>
  291. <span class="p">)</span>
  292. <span class="n">sql_index</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">&quot;CREATE INDEX IF NOT EXISTS index_expire_</span><span class="si">{</span><span class="n">table</span><span class="si">}</span><span class="s2"> ON </span><span class="si">{</span><span class="n">table</span><span class="si">}</span><span class="s2">(expire);&quot;</span>
  293. <span class="k">with</span> <span class="bp">self</span><span class="o">.</span><span class="n">connect</span><span class="p">()</span> <span class="k">as</span> <span class="n">conn</span><span class="p">:</span>
  294. <span class="n">conn</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="n">sql_table</span><span class="p">)</span>
  295. <span class="n">conn</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="n">sql_index</span><span class="p">)</span>
  296. <span class="n">conn</span><span class="o">.</span><span class="n">close</span><span class="p">()</span>
  297. <span class="bp">self</span><span class="o">.</span><span class="n">properties</span><span class="o">.</span><span class="n">set</span><span class="p">(</span><span class="sa">f</span><span class="s2">&quot;</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">CACHE_TABLE_PREFIX</span><span class="si">}</span><span class="s2">-</span><span class="si">{</span><span class="n">table</span><span class="si">}</span><span class="s2">&quot;</span><span class="p">,</span> <span class="n">table</span><span class="p">)</span>
  298. <span class="k">return</span> <span class="kc">True</span></div>
  299. <span class="nd">@property</span>
  300. <span class="k">def</span><span class="w"> </span><span class="nf">table_names</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">list</span><span class="p">[</span><span class="nb">str</span><span class="p">]:</span>
  301. <span class="w"> </span><span class="sd">&quot;&quot;&quot;List of key/value tables already created in the DB.&quot;&quot;&quot;</span>
  302. <span class="n">sql</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">&quot;SELECT value FROM properties WHERE name LIKE &#39;</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">CACHE_TABLE_PREFIX</span><span class="si">}</span><span class="s2">%%&#39;&quot;</span>
  303. <span class="n">rows</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">DB</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="n">sql</span><span class="p">)</span><span class="o">.</span><span class="n">fetchall</span><span class="p">()</span> <span class="ow">or</span> <span class="p">[]</span>
  304. <span class="k">return</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="k">for</span> <span class="n">r</span> <span class="ow">in</span> <span class="n">rows</span><span class="p">]</span>
  305. <span class="k">def</span><span class="w"> </span><span class="nf">truncate_tables</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">table_names</span><span class="p">:</span> <span class="nb">list</span><span class="p">[</span><span class="nb">str</span><span class="p">]):</span>
  306. <span class="n">log</span><span class="o">.</span><span class="n">debug</span><span class="p">(</span><span class="s2">&quot;truncate table: </span><span class="si">%s</span><span class="s2">&quot;</span><span class="p">,</span> <span class="s2">&quot;,&quot;</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">table_names</span><span class="p">))</span>
  307. <span class="k">with</span> <span class="bp">self</span><span class="o">.</span><span class="n">connect</span><span class="p">()</span> <span class="k">as</span> <span class="n">conn</span><span class="p">:</span>
  308. <span class="k">for</span> <span class="n">table</span> <span class="ow">in</span> <span class="n">table_names</span><span class="p">:</span>
  309. <span class="n">conn</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="sa">f</span><span class="s2">&quot;DELETE FROM </span><span class="si">{</span><span class="n">table</span><span class="si">}</span><span class="s2">&quot;</span><span class="p">)</span>
  310. <span class="n">conn</span><span class="o">.</span><span class="n">close</span><span class="p">()</span>
  311. <span class="k">return</span> <span class="kc">True</span>
  312. <span class="nd">@property</span>
  313. <span class="k">def</span><span class="w"> </span><span class="nf">next_maintenance_time</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">int</span><span class="p">:</span>
  314. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Returns (unix epoch) time of the next maintenance.&quot;&quot;&quot;</span>
  315. <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">cfg</span><span class="o">.</span><span class="n">MAINTENANCE_PERIOD</span> <span class="o">+</span> <span class="bp">self</span><span class="o">.</span><span class="n">properties</span><span class="o">.</span><span class="n">m_time</span><span class="p">(</span><span class="s2">&quot;LAST_MAINTENANCE&quot;</span><span class="p">,</span> <span class="nb">int</span><span class="p">(</span><span class="n">time</span><span class="o">.</span><span class="n">time</span><span class="p">()))</span>
  316. <span class="c1"># implement ABC methods of ExpireCache</span>
  317. <div class="viewcode-block" id="ExpireCacheSQLite.set">
  318. <a class="viewcode-back" href="../../src/searx.cache.html#searx.cache.ExpireCacheSQLite.set">[docs]</a>
  319. <span class="k">def</span><span class="w"> </span><span class="nf">set</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">key</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">value</span><span class="p">:</span> <span class="n">typing</span><span class="o">.</span><span class="n">Any</span><span class="p">,</span> <span class="n">expire</span><span class="p">:</span> <span class="nb">int</span> <span class="o">|</span> <span class="kc">None</span><span class="p">,</span> <span class="n">ctx</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">bool</span><span class="p">:</span>
  320. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Set key/value in DB table given by argument ``ctx``. If expire is</span>
  321. <span class="sd"> unset the default is taken from :py:obj:`ExpireCacheCfg.MAXHOLD_TIME`.</span>
  322. <span class="sd"> If ``ctx`` argument is ``None`` (the default), a table name is</span>
  323. <span class="sd"> generated from the :py:obj:`ExpireCacheCfg.name`. If DB table does not</span>
  324. <span class="sd"> exists, it will be created (on demand) by :py:obj:`self.create_table</span>
  325. <span class="sd"> &lt;ExpireCacheSQLite.create_table&gt;`.</span>
  326. <span class="sd"> &quot;&quot;&quot;</span>
  327. <span class="n">table</span> <span class="o">=</span> <span class="n">ctx</span>
  328. <span class="bp">self</span><span class="o">.</span><span class="n">maintenance</span><span class="p">()</span>
  329. <span class="n">value</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">serialize</span><span class="p">(</span><span class="n">value</span><span class="o">=</span><span class="n">value</span><span class="p">)</span>
  330. <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">value</span><span class="p">)</span> <span class="o">&gt;</span> <span class="bp">self</span><span class="o">.</span><span class="n">cfg</span><span class="o">.</span><span class="n">MAX_VALUE_LEN</span><span class="p">:</span>
  331. <span class="n">log</span><span class="o">.</span><span class="n">warning</span><span class="p">(</span><span class="s2">&quot;ExpireCache.set(): </span><span class="si">%s</span><span class="s2">.key=&#39;</span><span class="si">%s</span><span class="s2">&#39; - value too big to cache (len: </span><span class="si">%s</span><span class="s2">) &quot;</span><span class="p">,</span> <span class="n">table</span><span class="p">,</span> <span class="n">value</span><span class="p">,</span> <span class="nb">len</span><span class="p">(</span><span class="n">value</span><span class="p">))</span>
  332. <span class="k">return</span> <span class="kc">False</span>
  333. <span class="k">if</span> <span class="ow">not</span> <span class="n">expire</span><span class="p">:</span>
  334. <span class="n">expire</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">cfg</span><span class="o">.</span><span class="n">MAXHOLD_TIME</span>
  335. <span class="n">expire</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">time</span><span class="o">.</span><span class="n">time</span><span class="p">())</span> <span class="o">+</span> <span class="n">expire</span>
  336. <span class="n">table_name</span> <span class="o">=</span> <span class="n">table</span>
  337. <span class="k">if</span> <span class="ow">not</span> <span class="n">table_name</span><span class="p">:</span>
  338. <span class="n">table_name</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">normalize_name</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">cfg</span><span class="o">.</span><span class="n">name</span><span class="p">)</span>
  339. <span class="bp">self</span><span class="o">.</span><span class="n">create_table</span><span class="p">(</span><span class="n">table_name</span><span class="p">)</span>
  340. <span class="n">sql</span> <span class="o">=</span> <span class="p">(</span>
  341. <span class="sa">f</span><span class="s2">&quot;INSERT INTO </span><span class="si">{</span><span class="n">table_name</span><span class="si">}</span><span class="s2"> (key, value, expire) VALUES (?, ?, ?)&quot;</span>
  342. <span class="sa">f</span><span class="s2">&quot; ON CONFLICT DO &quot;</span>
  343. <span class="sa">f</span><span class="s2">&quot;UPDATE SET value=?, expire=?&quot;</span>
  344. <span class="p">)</span>
  345. <span class="k">if</span> <span class="n">table</span><span class="p">:</span>
  346. <span class="k">with</span> <span class="bp">self</span><span class="o">.</span><span class="n">DB</span><span class="p">:</span>
  347. <span class="bp">self</span><span class="o">.</span><span class="n">DB</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="n">sql</span><span class="p">,</span> <span class="p">(</span><span class="n">key</span><span class="p">,</span> <span class="n">value</span><span class="p">,</span> <span class="n">expire</span><span class="p">,</span> <span class="n">value</span><span class="p">,</span> <span class="n">expire</span><span class="p">))</span>
  348. <span class="k">else</span><span class="p">:</span>
  349. <span class="k">with</span> <span class="bp">self</span><span class="o">.</span><span class="n">connect</span><span class="p">()</span> <span class="k">as</span> <span class="n">conn</span><span class="p">:</span>
  350. <span class="n">conn</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="n">sql</span><span class="p">,</span> <span class="p">(</span><span class="n">key</span><span class="p">,</span> <span class="n">value</span><span class="p">,</span> <span class="n">expire</span><span class="p">,</span> <span class="n">value</span><span class="p">,</span> <span class="n">expire</span><span class="p">))</span>
  351. <span class="n">conn</span><span class="o">.</span><span class="n">close</span><span class="p">()</span>
  352. <span class="k">return</span> <span class="kc">True</span></div>
  353. <div class="viewcode-block" id="ExpireCacheSQLite.get">
  354. <a class="viewcode-back" href="../../src/searx.cache.html#searx.cache.ExpireCacheSQLite.get">[docs]</a>
  355. <span class="k">def</span><span class="w"> </span><span class="nf">get</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">key</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">default</span><span class="o">=</span><span class="kc">None</span><span class="p">,</span> <span class="n">ctx</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">typing</span><span class="o">.</span><span class="n">Any</span><span class="p">:</span>
  356. <span class="w"> </span><span class="sd">&quot;&quot;&quot;Get value of ``key`` from table given by argument ``ctx``. If</span>
  357. <span class="sd"> ``ctx`` argument is ``None`` (the default), a table name is generated</span>
  358. <span class="sd"> from the :py:obj:`ExpireCacheCfg.name`. If ``key`` not exists (in</span>
  359. <span class="sd"> table), the ``default`` value is returned.</span>
  360. <span class="sd"> &quot;&quot;&quot;</span>
  361. <span class="n">table</span> <span class="o">=</span> <span class="n">ctx</span>
  362. <span class="bp">self</span><span class="o">.</span><span class="n">maintenance</span><span class="p">()</span>
  363. <span class="k">if</span> <span class="ow">not</span> <span class="n">table</span><span class="p">:</span>
  364. <span class="n">table</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">normalize_name</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">cfg</span><span class="o">.</span><span class="n">name</span><span class="p">)</span>
  365. <span class="k">if</span> <span class="n">table</span> <span class="ow">not</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">table_names</span><span class="p">:</span>
  366. <span class="k">return</span> <span class="n">default</span>
  367. <span class="n">sql</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">&quot;SELECT value FROM </span><span class="si">{</span><span class="n">table</span><span class="si">}</span><span class="s2"> WHERE key = ?&quot;</span>
  368. <span class="n">row</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">DB</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="n">sql</span><span class="p">,</span> <span class="p">(</span><span class="n">key</span><span class="p">,))</span><span class="o">.</span><span class="n">fetchone</span><span class="p">()</span>
  369. <span class="k">if</span> <span class="n">row</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
  370. <span class="k">return</span> <span class="n">default</span>
  371. <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">deserialize</span><span class="p">(</span><span class="n">row</span><span class="p">[</span><span class="mi">0</span><span class="p">])</span></div>
  372. <div class="viewcode-block" id="ExpireCacheSQLite.state">
  373. <a class="viewcode-back" href="../../src/searx.cache.html#searx.cache.ExpireCacheSQLite.state">[docs]</a>
  374. <span class="k">def</span><span class="w"> </span><span class="nf">state</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">ExpireCacheStats</span><span class="p">:</span>
  375. <span class="n">cached_items</span> <span class="o">=</span> <span class="p">{}</span>
  376. <span class="k">for</span> <span class="n">table</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">table_names</span><span class="p">:</span>
  377. <span class="n">cached_items</span><span class="p">[</span><span class="n">table</span><span class="p">]</span> <span class="o">=</span> <span class="p">[]</span>
  378. <span class="k">for</span> <span class="n">row</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">DB</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="sa">f</span><span class="s2">&quot;SELECT key, value, expire FROM </span><span class="si">{</span><span class="n">table</span><span class="si">}</span><span class="s2">&quot;</span><span class="p">):</span>
  379. <span class="n">cached_items</span><span class="p">[</span><span class="n">table</span><span class="p">]</span><span class="o">.</span><span class="n">append</span><span class="p">((</span><span class="n">row</span><span class="p">[</span><span class="mi">0</span><span class="p">],</span> <span class="bp">self</span><span class="o">.</span><span class="n">deserialize</span><span class="p">(</span><span class="n">row</span><span class="p">[</span><span class="mi">1</span><span class="p">]),</span> <span class="n">row</span><span class="p">[</span><span class="mi">2</span><span class="p">]))</span>
  380. <span class="k">return</span> <span class="n">ExpireCacheStats</span><span class="p">(</span><span class="n">cached_items</span><span class="o">=</span><span class="n">cached_items</span><span class="p">)</span></div>
  381. </div>
  382. </pre></div>
  383. <div class="clearer"></div>
  384. </div>
  385. </div>
  386. </div>
  387. <span id="sidebar-top"></span>
  388. <div class="sphinxsidebar" role="navigation" aria-label="Main">
  389. <div class="sphinxsidebarwrapper">
  390. <p class="logo"><a href="../../index.html">
  391. <img class="logo" src="../../_static/searxng-wordmark.svg" alt="Logo of SearXNG"/>
  392. </a></p>
  393. <h3><a href="../../index.html">Table of Contents</a></h3>
  394. <ul>
  395. <li class="toctree-l1"><a class="reference internal" href="../../user/index.html">User information</a></li>
  396. <li class="toctree-l1"><a class="reference internal" href="../../own-instance.html">Why use a private instance?</a></li>
  397. <li class="toctree-l1"><a class="reference internal" href="../../admin/index.html">Administrator documentation</a></li>
  398. <li class="toctree-l1"><a class="reference internal" href="../../dev/index.html">Developer documentation</a></li>
  399. <li class="toctree-l1"><a class="reference internal" href="../../utils/index.html">DevOps tooling box</a></li>
  400. <li class="toctree-l1"><a class="reference internal" href="../../src/index.html">Source-Code</a></li>
  401. </ul>
  402. <h3>Project Links</h3>
  403. <ul>
  404. <li><a href="https://github.com/searxng/searxng/tree/master">Source</a>
  405. <li><a href="https://github.com/searxng/searxng/wiki">Wiki</a>
  406. <li><a href="https://searx.space">Public instances</a>
  407. <li><a href="https://github.com/searxng/searxng/issues">Issue Tracker</a>
  408. </ul><h3>Navigation</h3>
  409. <ul>
  410. <li><a href="../../index.html">Overview</a>
  411. <ul>
  412. <li><a href="../index.html">Module code</a>
  413. </ul>
  414. </li>
  415. </ul>
  416. </li>
  417. </ul>
  418. <search id="searchbox" style="display: none" role="search">
  419. <h3 id="searchlabel">Quick search</h3>
  420. <div class="searchformwrapper">
  421. <form class="search" action="../../search.html" method="get">
  422. <input type="text" name="q" aria-labelledby="searchlabel" autocomplete="off" autocorrect="off" autocapitalize="off" spellcheck="false"/>
  423. <input type="submit" value="Go" />
  424. </form>
  425. </div>
  426. </search>
  427. <script>document.getElementById('searchbox').style.display = "block"</script>
  428. </div>
  429. </div>
  430. <div class="clearer"></div>
  431. </div>
  432. <div class="footer" role="contentinfo">
  433. &#169; Copyright SearXNG team.
  434. </div>
  435. </body>
  436. </html>