完成对接数据库查询

2026-01-19 17:44:51 +08:00
commit 502cf71367
13 changed files with 2069 additions and 0 deletions
--- a/.gitignore
+++ b/.gitignore
@@ -0,0 +1,141 @@
+.idea/**/workspace.xml
+.idea/**/tasks.xml
+.idea/**/usage.statistics.xml
+.idea/**/dictionaries
+.idea/**/shelf
+.idea/**/aws.xml
+.idea/**/contentModel.xml
+.idea/**/dataSources/
+.idea/**/dataSources.ids
+.idea/**/dataSources.local.xml
+.idea/**/sqlDataSources.xml
+.idea/**/dynamic.xml
+.idea/**/uiDesigner.xml
+.idea/**/dbnavigator.xml
+.idea/**/gradle.xml
+.idea/**/libraries
+cmake-build-*/
+.idea/**/mongoSettings.xml
+*.iws
+out/
+.idea_modules/
+atlassian-ide-plugin.xml
+.idea/replstate.xml
+.idea/sonarlint/
+com_crashlytics_export_strings.xml
+crashlytics.properties
+crashlytics-build.properties
+fabric.properties
+.idea/httpRequests
+.idea/caches/build_file_checksums.ser
+*~
+.fuse_hidden*
+.directory
+.Trash-*
+.nfs*
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+pip-log.txt
+pip-delete-this-directory.txt
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+*.mo
+*.pot
+__pypackages__/
+Thumbs.db
+Thumbs.db:encryptable
+ehthumbs.db
+ehthumbs_vista.db
+*.stackdump
+[Dd]esktop.ini
+$RECYCLE.BIN/
+*.cab
+*.msi
+*.msix
+*.msm
+*.msp
+*.lnk
+.DS_Store
+.AppleDouble
+.LSOverride
+Icon
+._*
+.DocumentRevisions-V100
+.fseventsd
+.Spotlight-V100
+.TemporaryItems
+.Trashes
+.VolumeIcon.icns
+.com.apple.timemachine.donotpresent
+.AppleDB
+.AppleDesktop
+Network Trash Folder
+Temporary Items
+.apdisk
+*.manifest
+*.spec
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+instance/
+.webassets-cache
+.scrapy
+docs/_build/
+.pybuilder/
+target/
+.ipynb_checkpoints
+profile_default/
+ipython_config.py
+.pdm.toml
+celerybeat-schedule
+celerybeat.pid
+*.sage.py
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+.spyderproject
+.spyproject
+.ropeproject
+/site
+.mypy_cache/
+.dmypy.json
+dmypy.json
+.pyre/
+.pytype/
+cython_debug/
--- a/.idea/.gitignore
+++ b/.idea/.gitignore
@@ -0,0 +1,10 @@
+# 默认忽略的文件
+/shelf/
+/workspace.xml
+# 已忽略包含查询文件的默认文件夹
+/queries/
+# Datasource local storage ignored files
+/dataSources/
+/dataSources.local.xml
+# 基于编辑器的 HTTP 客户端请求
+/httpRequests/
--- a/.idea/dataSources.xml
+++ b/.idea/dataSources.xml
@@ -0,0 +1,12 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="DataSourceManagerImpl" format="xml" multifile-model="true">
+    <data-source source="LOCAL" name="leopard_dev@81.71.3.24" uuid="bd7b5f2a-eb99-4aad-81ec-1fec76b3d7fc">
+      <driver-ref>postgresql</driver-ref>
+      <synchronize>true</synchronize>
+      <jdbc-driver>org.postgresql.Driver</jdbc-driver>
+      <jdbc-url>jdbc:postgresql://81.71.3.24:6785/leopard_dev</jdbc-url>
+      <working-dir>$ProjectFileDir$</working-dir>
+    </data-source>
+  </component>
+</project>
--- a/.idea/inspectionProfiles/Project_Default.xml
+++ b/.idea/inspectionProfiles/Project_Default.xml
@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <profile version="1.0">
+    <option name="myName" value="Project Default" />
+    <inspection_tool class="MarkdownIncorrectTableFormatting" enabled="false" level="WEAK WARNING" enabled_by_default="false" />
+  </profile>
+</component>
--- a/.idea/inspectionProfiles/profiles_settings.xml
+++ b/.idea/inspectionProfiles/profiles_settings.xml
@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>
--- a/.idea/leopard_analysis.iml
+++ b/.idea/leopard_analysis.iml
@@ -0,0 +1,10 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$">
+      <excludeFolder url="file://$MODULE_DIR$/.venv" />
+    </content>
+    <orderEntry type="jdk" jdkName="uv (leopard_analysis)" jdkType="Python SDK" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>
--- a/.idea/misc.xml
+++ b/.idea/misc.xml
@@ -0,0 +1,7 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="Black">
+    <option name="sdkName" value="uv (leopard_analysis)" />
+  </component>
+  <component name="ProjectRootManager" version="2" project-jdk-name="uv (leopard_analysis)" project-jdk-type="Python SDK" />
+</project>
--- a/.idea/modules.xml
+++ b/.idea/modules.xml
@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/leopard_analysis.iml" filepath="$PROJECT_DIR$/.idea/leopard_analysis.iml" />
+    </modules>
+  </component>
+</project>
--- a/.idea/sqldialects.xml
+++ b/.idea/sqldialects.xml
@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="SqlDialectMappings">
+    <file url="PROJECT" dialect="PostgreSQL" />
+  </component>
+</project>
--- a/.idea/vcs.xml
+++ b/.idea/vcs.xml
@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="$PROJECT_DIR$" vcs="Git" />
+  </component>
+</project>
--- a/backtest.ipynb
+++ b/backtest.ipynb
@@ -0,0 +1,237 @@
+{
+ "cells": [
+  {
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2026-01-19T09:22:09.640369Z",
+     "start_time": "2026-01-19T09:22:07.517191Z"
+    }
+   },
+   "cell_type": "code",
+   "source": [
+    "import pandas as pd\n",
+    "\n",
+    "host = '81.71.3.24'\n",
+    "port = 6785\n",
+    "database = 'leopard_dev'\n",
+    "username = 'leopard'\n",
+    "password = '9NEzFzovnddf@PyEP?e*AYAWnCyd7UhYwQK$pJf>7?ccFiN^x4$eKEZ5~E<7<+~X'"
+   ],
+   "id": "d815a3591c4f9463",
+   "outputs": [],
+   "execution_count": 7
+  },
+  {
+   "cell_type": "code",
+   "id": "initial_id",
+   "metadata": {
+    "collapsed": true,
+    "ExecuteTime": {
+     "end_time": "2026-01-19T09:42:25.068816Z",
+     "start_time": "2026-01-19T09:42:18.026869Z"
+    }
+   },
+   "source": [
+    "import psycopg2\n",
+    "\n",
+    "dailies_df = pd.DataFrame()\n",
+    "\n",
+    "with psycopg2.connect(host=host, port=port, database=database, user=username, password=password) as connection:\n",
+    "    with connection.cursor() as cursor:\n",
+    "        # language=PostgreSQL\n",
+    "        cursor.execute(\n",
+    "            \"\"\"select trade_date, open, close, high, low, factor\n",
+    "from leopard_daily daily\n",
+    "         left join leopard_stock stock on stock.id = daily.stock_id\n",
+    "where stock.code = '000001.SZ'\n",
+    "  and daily.trade_date between '2025-01-01 00:00:00' and '2025-12-31 23:59:59'\n",
+    "order by daily.trade_date\"\"\"\n",
+    "            )\n",
+    "        rows = cursor.fetchall()\n",
+    "\n",
+    "        df = pd.DataFrame.from_records(rows, columns=['trade_date', 'open', 'close', 'high', 'low', 'factor'])\n",
+    "\n",
+    "df"
+   ],
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "     trade_date   open  close   high    low    factor\n",
+       "0    2025-01-02  11.73  11.43  11.77  11.39  127.7841\n",
+       "1    2025-01-03  11.44  11.38  11.54  11.36  127.7841\n",
+       "2    2025-01-06  11.38  11.44  11.48  11.22  127.7841\n",
+       "3    2025-01-07  11.42  11.51  11.53  11.37  127.7841\n",
+       "4    2025-01-08  11.50  11.50  11.63  11.40  127.7841\n",
+       "..          ...    ...    ...    ...    ...       ...\n",
+       "183  2025-10-09  11.33  11.40  11.41  11.27  131.7878\n",
+       "184  2025-10-10  11.37  11.43  11.49  11.36  131.7878\n",
+       "185  2025-10-13  11.32  11.40  11.46  11.28  131.7878\n",
+       "186  2025-10-14  11.39  11.57  11.60  11.36  131.7878\n",
+       "187  2025-10-15  11.34  11.40  11.42  11.26  134.5794\n",
+       "\n",
+       "[188 rows x 6 columns]"
+      ],
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
+       "      <th></th>\n",
+       "      <th>trade_date</th>\n",
+       "      <th>open</th>\n",
+       "      <th>close</th>\n",
+       "      <th>high</th>\n",
+       "      <th>low</th>\n",
+       "      <th>factor</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>0</th>\n",
+       "      <td>2025-01-02</td>\n",
+       "      <td>11.73</td>\n",
+       "      <td>11.43</td>\n",
+       "      <td>11.77</td>\n",
+       "      <td>11.39</td>\n",
+       "      <td>127.7841</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>1</th>\n",
+       "      <td>2025-01-03</td>\n",
+       "      <td>11.44</td>\n",
+       "      <td>11.38</td>\n",
+       "      <td>11.54</td>\n",
+       "      <td>11.36</td>\n",
+       "      <td>127.7841</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>2</th>\n",
+       "      <td>2025-01-06</td>\n",
+       "      <td>11.38</td>\n",
+       "      <td>11.44</td>\n",
+       "      <td>11.48</td>\n",
+       "      <td>11.22</td>\n",
+       "      <td>127.7841</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>3</th>\n",
+       "      <td>2025-01-07</td>\n",
+       "      <td>11.42</td>\n",
+       "      <td>11.51</td>\n",
+       "      <td>11.53</td>\n",
+       "      <td>11.37</td>\n",
+       "      <td>127.7841</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>4</th>\n",
+       "      <td>2025-01-08</td>\n",
+       "      <td>11.50</td>\n",
+       "      <td>11.50</td>\n",
+       "      <td>11.63</td>\n",
+       "      <td>11.40</td>\n",
+       "      <td>127.7841</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>...</th>\n",
+       "      <td>...</td>\n",
+       "      <td>...</td>\n",
+       "      <td>...</td>\n",
+       "      <td>...</td>\n",
+       "      <td>...</td>\n",
+       "      <td>...</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>183</th>\n",
+       "      <td>2025-10-09</td>\n",
+       "      <td>11.33</td>\n",
+       "      <td>11.40</td>\n",
+       "      <td>11.41</td>\n",
+       "      <td>11.27</td>\n",
+       "      <td>131.7878</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>184</th>\n",
+       "      <td>2025-10-10</td>\n",
+       "      <td>11.37</td>\n",
+       "      <td>11.43</td>\n",
+       "      <td>11.49</td>\n",
+       "      <td>11.36</td>\n",
+       "      <td>131.7878</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>185</th>\n",
+       "      <td>2025-10-13</td>\n",
+       "      <td>11.32</td>\n",
+       "      <td>11.40</td>\n",
+       "      <td>11.46</td>\n",
+       "      <td>11.28</td>\n",
+       "      <td>131.7878</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>186</th>\n",
+       "      <td>2025-10-14</td>\n",
+       "      <td>11.39</td>\n",
+       "      <td>11.57</td>\n",
+       "      <td>11.60</td>\n",
+       "      <td>11.36</td>\n",
+       "      <td>131.7878</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>187</th>\n",
+       "      <td>2025-10-15</td>\n",
+       "      <td>11.34</td>\n",
+       "      <td>11.40</td>\n",
+       "      <td>11.42</td>\n",
+       "      <td>11.26</td>\n",
+       "      <td>134.5794</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "<p>188 rows × 6 columns</p>\n",
+       "</div>"
+      ]
+     },
+     "execution_count": 17,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "execution_count": 17
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 2
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython2",
+   "version": "2.7.6"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -0,0 +1,13 @@
+[project]
+name = "leopard-analysis"
+version = "0.1.0"
+description = "Stock analysis"
+requires-python = ">=3.14"
+dependencies = [
+    "backtesting>=0.6.5",
+    "jupyter>=1.1.1",
+    "pandas>=2.3.3",
+    "pandas-stubs~=2.3.3",
+    "peewee>=3.19.0",
+    "psycopg2-binary>=2.9.11",
+]
--- a/uv.lock
+++ b/uv.lock