修改页面

This commit is contained in:
崔芝斌 2025-08-03 21:35:56 +08:00
parent b4b45eeb14
commit 9c7bb9e568
129 changed files with 81091 additions and 76012 deletions

View File

@ -59,20 +59,22 @@ android {
chaquopy {
defaultConfig {
version = "3.11"
buildPython("C:\\Users\\xuyvqi\\AppData\\Local\\Programs\\Python\\Python311\\python.exe")
buildPython("C:\\Users\\cuizhibin\\AppData\\Local\\Programs\\Python\\Python311\\python.exe")
pip {
install("typing_extensions")
install("lxml==5.3.0") // python-docx 的依赖
install("python-docx") // 使用兼容性较好的版本
install("Pillow") // 可选,处理图片
}
}
}
dependencies {
implementation(libs.room.runtime)
implementation(libs.firebase.crashlytics.buildtools)
implementation(libs.activity)
annotationProcessor(libs.room.compiler)
implementation(libs.appcompat)
implementation(libs.material)
@ -83,7 +85,9 @@ dependencies {
androidTestImplementation(libs.ext.junit)
androidTestImplementation(libs.espresso.core)
implementation(libs.okhttp)
implementation ("androidx.core:core-ktx:1.12.0")
implementation ("androidx.lifecycle:lifecycle-livedata-ktx:2.7.0")
implementation ("androidx.lifecycle:lifecycle-viewmodel-ktx:2.7.0")
implementation("com.google.android.gms:play-services-location:21.0.1")
implementation("androidx.exifinterface:exifinterface:1.3.3")
implementation ("commons-io:commons-io:2.11.0")

View File

@ -2,11 +2,11 @@
"formatVersion": 1,
"database": {
"version": 4,
"identityHash": "ab7260384e2d58c0d78dc0dae5130632",
"identityHash": "6390dc3df9684c38dfe932d1189a35a8",
"entities": [
{
"tableName": "images",
"createSql": "CREATE TABLE IF NOT EXISTS `${TABLE_NAME}` (`id` INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL, `path` TEXT, `time` INTEGER NOT NULL, `latitude` REAL NOT NULL, `longitude` REAL NOT NULL, `altitude` REAL NOT NULL, `user` TEXT, `audioPath` TEXT, `blade` INTEGER NOT NULL, `unit` TEXT, `unitName` TEXT, `project` TEXT, `b` INTEGER NOT NULL, `temperature` TEXT, `humidity` TEXT, `weather` TEXT)",
"createSql": "CREATE TABLE IF NOT EXISTS `${TABLE_NAME}` (`id` INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL, `path` TEXT, `time` INTEGER NOT NULL, `latitude` REAL NOT NULL, `longitude` REAL NOT NULL, `altitude` REAL NOT NULL, `user` TEXT, `audioPath` TEXT, `turbineId` TEXT, `turbineName` TEXT, `partId` TEXT, `partName` TEXT, `project` TEXT, `b` INTEGER NOT NULL, `temperature` TEXT, `humidity` TEXT, `weather` TEXT, `imageSource` TEXT)",
"fields": [
{
"fieldPath": "id",
@ -57,20 +57,26 @@
"notNull": false
},
{
"fieldPath": "blade",
"columnName": "blade",
"affinity": "INTEGER",
"notNull": true
},
{
"fieldPath": "unit",
"columnName": "unit",
"fieldPath": "turbineId",
"columnName": "turbineId",
"affinity": "TEXT",
"notNull": false
},
{
"fieldPath": "unitName",
"columnName": "unitName",
"fieldPath": "turbineName",
"columnName": "turbineName",
"affinity": "TEXT",
"notNull": false
},
{
"fieldPath": "partId",
"columnName": "partId",
"affinity": "TEXT",
"notNull": false
},
{
"fieldPath": "partName",
"columnName": "partName",
"affinity": "TEXT",
"notNull": false
},
@ -103,6 +109,12 @@
"columnName": "weather",
"affinity": "TEXT",
"notNull": false
},
{
"fieldPath": "imageSource",
"columnName": "imageSource",
"affinity": "TEXT",
"notNull": false
}
],
"primaryKey": {
@ -148,7 +160,7 @@
},
{
"tableName": "audio_entities",
"createSql": "CREATE TABLE IF NOT EXISTS `${TABLE_NAME}` (`id` INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL, `fileName` TEXT, `AudioPath` TEXT, `ImagePath` TEXT, `fileSize` INTEGER NOT NULL, `time` TEXT, `duration` INTEGER NOT NULL, `createdAt` INTEGER NOT NULL)",
"createSql": "CREATE TABLE IF NOT EXISTS `${TABLE_NAME}` (`id` INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL, `isupdated` INTEGER NOT NULL, `fileName` TEXT, `AudioPath` TEXT, `ImagePath` TEXT, `fileSize` INTEGER NOT NULL, `time` TEXT, `duration` INTEGER NOT NULL, `createdAt` INTEGER NOT NULL)",
"fields": [
{
"fieldPath": "id",
@ -156,6 +168,12 @@
"affinity": "INTEGER",
"notNull": true
},
{
"fieldPath": "isupdated",
"columnName": "isupdated",
"affinity": "INTEGER",
"notNull": true
},
{
"fieldPath": "fileName",
"columnName": "fileName",
@ -212,7 +230,7 @@
"views": [],
"setupQueries": [
"CREATE TABLE IF NOT EXISTS room_master_table (id INTEGER PRIMARY KEY,identity_hash TEXT)",
"INSERT OR REPLACE INTO room_master_table (id,identity_hash) VALUES(42, 'ab7260384e2d58c0d78dc0dae5130632')"
"INSERT OR REPLACE INTO room_master_table (id,identity_hash) VALUES(42, '6390dc3df9684c38dfe932d1189a35a8')"
]
}
}

View File

@ -1,101 +1,132 @@
<?xml version="1.0" encoding="utf-8"?>
<manifest xmlns:android="http://schemas.android.com/apk/res/android"
xmlns:tools="http://schemas.android.com/tools">
xmlns:tools="http://schemas.android.com/tools" >
<!-- 基础权限 -->
<!-- Android 14+ 新增权限 -->
<uses-feature
android:name="android.hardware.camera"
android:required="false" />
<uses-permission android:name="android.permission.FOREGROUND_SERVICE_CAMERA" />
<uses-permission android:name="android.permission.CAMERA" />
<uses-permission android:name="android.permission.READ_MEDIA_AUDIO" />
<uses-permission android:name="android.permission.MANAGE_EXTERNAL_STORAGE"
<uses-permission
android:name="android.permission.MANAGE_EXTERNAL_STORAGE"
tools:ignore="ScopedStorage" />
<uses-permission android:name="android.permission.POST_NOTIFICATIONS" />
<uses-permission android:name="android.permission.READ_MEDIA_VISUAL_USER_SELECTED" />
<uses-permission android:name="android.permission.REQUEST_INSTALL_PACKAGES" />
<uses-permission android:name="android.permission.FOREGROUND_SERVICE_SPECIAL_USE" />
<uses-permission android:name="android.permission.READ_MEDIA_VIDEO" />
<uses-permission android:name="android.permission.READ_MEDIA_IMAGES"
<uses-permission
android:name="android.permission.READ_MEDIA_IMAGES"
tools:ignore="SelectedPhotoAccess" />
<uses-permission android:name="android.permission.FOREGROUND_SERVICE"/>
<uses-permission android:name="android.permission.FOREGROUND_SERVICE" />
<uses-permission android:name="android.permission.INTERNET" />
<uses-permission android:name="android.permission.ACCESS_NETWORK_STATE" />
<uses-permission android:name="android.permission.ACCESS_WIFI_STATE" />
<!--位置权限-->
<uses-permission android:name="android.permission.ACCESS_WIFI_STATE" /> <!-- 位置权限 -->
<uses-permission android:name="android.permission.ACCESS_FINE_LOCATION" />
<uses-permission android:name="android.permission.ACCESS_COARSE_LOCATION" />
<uses-permission android:name="android.permission.ACCESS_BACKGROUND_LOCATION" />
<uses-permission android:name="android.permission.RECORD_AUDIO"/>
<uses-permission android:name="android.permission.READ_EXTERNAL_STORAGE"
<uses-permission android:name="android.permission.RECORD_AUDIO" />
<uses-permission
android:name="android.permission.READ_EXTERNAL_STORAGE"
android:maxSdkVersion="32" />
<uses-permission android:name="android.permission.WRITE_EXTERNAL_STORAGE"
<uses-permission android:name="android.permission.VIBRATE" />
<uses-permission
android:name="android.permission.WRITE_EXTERNAL_STORAGE"
android:maxSdkVersion="32"
tools:ignore="ScopedStorage" />
<uses-permission android:name="android.permission.SYSTEM_ALERT_WINDOW" />
<uses-permission android:name="android.permission.FOREGROUND_SERVICE_MICROPHONE" /> <!-- Android 14+ 必须 -->
<uses-permission android:name="android.permission.WAKE_LOCK" />
<queries>
<!-- 允许查询所有能处理图片选择的 Intent -->
<intent>
<action android:name="android.intent.action.PICK" />
<data android:mimeType="image/*" />
</intent>
<intent>
<action android:name="android.provider.action.PICK_IMAGES" />
<data android:mimeType="image/*" />
</intent>
<intent>
<action android:name="android.intent.action.GET_CONTENT" />
<data android:mimeType="image/*" />
</intent>
</queries>
<application
android:name="com.chaquo.python.android.PyApplication"
android:allowBackup="true"
android:dataExtractionRules="@xml/data_extraction_rules"
android:fullBackupContent="@xml/backup_rules"
android:icon="@mipmap/ic_launcher"
android:label="@string/app_name"
android:roundIcon="@mipmap/ic_launcher_round"
android:usesCleartextTraffic="true"
android:supportsRtl="true"
android:requestLegacyExternalStorage="true"
android:windowSoftInputMode="stateVisible|adjustResize"
android:roundIcon="@mipmap/ic_launcher_round"
android:supportsRtl="true"
android:theme="@style/Theme.MyApplication"
android:name="com.chaquo.python.android.PyApplication"
tools:targetApi="31"
tools:ignore="ForegroundServicePermission">
<activity
android:name=".MainActivity">
android:usesCleartextTraffic="true"
android:windowSoftInputMode="stateVisible|adjustResize"
tools:ignore="ForegroundServicePermission"
tools:targetApi="31" >
<activity android:name=".MainActivity" >
</activity>
<activity android:name=".AdminActivity" />
<activity android:name=".LoginActivity"
android:exported="true">
<activity
android:name=".LoginActivity"
android:exported="true" >
<intent-filter>
<action android:name="android.intent.action.MAIN" />
<category android:name="android.intent.category.LAUNCHER" />
</intent-filter>
</activity>
<service
android:name=".Service.PhotoMonitoringService"
android:enabled="true"
android:exported="false"
android:foregroundServiceType="camera" />
<service
android:name=".Service.FloatingWindowService"
android:foregroundServiceType="specialUse"
android:enabled="true"
android:exported="false" />
<service android:name=".Service.RecordingService"
android:foregroundServiceType="microphone"
android:exported="false"
android:foregroundServiceType="specialUse" />
<service
android:name=".Service.RecordingService"
android:enabled="true"
android:exported="false"/>
android:exported="false"
android:foregroundServiceType="microphone" />
<service
android:name=".Service.ForegroundService"
android:foregroundServiceType="microphone"
android:enabled="true"
android:exported="false" />
android:exported="false"
android:foregroundServiceType="microphone" />
<provider
android:name="androidx.core.content.FileProvider"
android:authorities="${applicationId}.provider"
android:exported="false"
android:grantUriPermissions="true">
android:grantUriPermissions="true" >
<meta-data
android:name="android.support.FILE_PROVIDER_PATHS"
android:resource="@xml/file_paths" />
</provider>
<receiver
android:name=".Tool.NotificationReceiver"
android:exported="false" > <!-- 安全限制:只接收本应用广播 -->
<intent-filter>
<action android:name="ACTION_STOP_SERVICE" />
</intent-filter>
</receiver>
</application>
<queries>
<!-- 允许查询所有能处理图片选择的 Intent -->
<intent>
<action android:name="android.intent.action.PICK" />
<data android:mimeType="image/*" />
</intent>
<intent>
<action android:name="android.provider.action.PICK_IMAGES" />
<data android:mimeType="image/*" />
</intent>
<intent>
<action android:name="android.intent.action.GET_CONTENT" />
<data android:mimeType="image/*" />
</intent>
</queries>
</manifest>

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

After

Width:  |  Height:  |  Size: 38 KiB

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

View File

@ -22,12 +22,7 @@ public abstract class AppDatabase extends RoomDatabase {
private static final Migration MIGRATION_4_5 = new Migration(4, 5) {
@Override
public void migrate(@NonNull SupportSQLiteDatabase database) {
}
};
public static AppDatabase getDatabase(Context context) {
if (INSTANCE == null) {
synchronized (AppDatabase.class) {

View File

@ -35,4 +35,5 @@ public interface AudioDao {
@Query("DELETE FROM audio_entities WHERE AudioPath = :path")
void deleteByPath(String path);
}

View File

@ -10,6 +10,7 @@ import android.database.sqlite.SQLiteOpenHelper;
import com.example.myapplication.model.ImageSourceItem;
import com.example.myapplication.model.PartResponse;
import com.example.myapplication.model.Project;
import com.example.myapplication.model.Turbine;
import com.google.firebase.crashlytics.buildtools.reloc.com.google.common.reflect.TypeToken;
import com.google.gson.Gson;
@ -20,10 +21,10 @@ import java.util.Map;
public class DatabaseHelper extends SQLiteOpenHelper {
private static final String DATABASE_NAME = "AppDB";
private static final int DATABASE_VERSION = 6;
private static final int DATABASE_VERSION = 7;
private static final String TABLE_USERS = "users";
private static final String TABLE_PROJECTS = "projects";
private static final String TABLE_TURBINES = "turbines";
// 用户表列名
public static final String COLUMN_ID = "_id";
public static final String COLUMN_USERNAME = "username";
@ -77,6 +78,12 @@ public class DatabaseHelper extends SQLiteOpenHelper {
+ COLUMN_PROJECT_NAME + " TEXT,"
+ COLUMN_LAST_UPDATED + " INTEGER" + ")";
db.execSQL(CREATE_PROJECTS_TABLE);
String CREATE_TURBINE_TABLE = "CREATE TABLE " + TABLE_TURBINES + "("
+ COLUMN_TURBINE_ID + " TEXT PRIMARY KEY,"
+ COLUMN_TURBINE_NAME + " TEXT,"
+ COLUMN_PROJECT_NAME + " TEXT,"
+ COLUMN_LAST_UPDATED + " INTEGER" + ")";
db.execSQL(CREATE_TURBINE_TABLE);
String CREATE_PARTS_TABLE = "CREATE TABLE " + TABLE_PARTS + "("
+ COLUMN_PART_ID + " TEXT PRIMARY KEY,"
+ COLUMN_PART_NAME + " TEXT,"
@ -157,6 +164,15 @@ public class DatabaseHelper extends SQLiteOpenHelper {
+ COLUMN_LAST_SYNC + " INTEGER DEFAULT 0" + ")";
db.execSQL(CREATE_DYNAMIC_DATA_TABLE);
}
if (oldVersion < 7) {
// 版本5到版本6的迁移添加机组表
String CREATE_TURBINE_TABLE = "CREATE TABLE " + TABLE_TURBINES + "("
+ COLUMN_TURBINE_ID + " TEXT PRIMARY KEY,"
+ COLUMN_TURBINE_NAME + " TEXT,"
+ COLUMN_PROJECT_NAME + " TEXT,"
+ COLUMN_LAST_UPDATED + " INTEGER" + ")";
db.execSQL(CREATE_TURBINE_TABLE);
}
}
/**
* 保存动态数据列表
@ -279,6 +295,110 @@ public class DatabaseHelper extends SQLiteOpenHelper {
cursor.close();
return sources;
}
/**
* 保存部件列表到数据库
*/
public void saveTurbines(List<Turbine> turbines) {
SQLiteDatabase db = this.getWritableDatabase();
db.beginTransaction();
try {
// 先清空表
db.delete(TABLE_TURBINES, null, null);
// 插入新数据
for (Turbine turbine : turbines) {
ContentValues values = new ContentValues();
values.put(COLUMN_PROJECT_ID, turbine.getProjectId());
values.put(COLUMN_TURBINE_ID, turbine.getTurbineId());
values.put(COLUMN_TURBINE_NAME, turbine.getTurbineName());
values.put(COLUMN_LAST_SYNC, System.currentTimeMillis());
db.insert(TABLE_TURBINES, null, values);
}
db.setTransactionSuccessful();
} finally {
db.endTransaction();
}
}
@SuppressLint("Range")
public List<Turbine> getTurbinesByProjectId(String projectId) {
List<Turbine> turbines = new ArrayList<>();
SQLiteDatabase db = this.getReadableDatabase();
// 定义查询条件
String selection = COLUMN_PROJECT_ID + " = ?";
String[] selectionArgs = { projectId };
Cursor cursor = db.query(TABLE_TURBINES,
null,
selection, selectionArgs, null, null, COLUMN_TURBINE_NAME + " ASC");
if (cursor.moveToFirst()) {
do {
Turbine turbine = new Turbine();
turbine.setProjectId(cursor.getString(cursor.getColumnIndex(COLUMN_Part_PROJECT_ID)));
turbine.setTurbineId(cursor.getString(cursor.getColumnIndex(COLUMN_TURBINE_ID)));
turbine.setTurbineName(cursor.getString(cursor.getColumnIndex(COLUMN_TURBINE_NAME)));
turbines.add(turbine);
} while (cursor.moveToNext());
}
cursor.close();
// 关闭游标
cursor.close();
return turbines;
}
/**
* 获取所有部件
*/
@SuppressLint("Range")
public List<Turbine> getAllTurbines() {
List<Turbine> turbines = new ArrayList<>();
SQLiteDatabase db = this.getReadableDatabase();
Cursor cursor = db.query(TABLE_TURBINES,
null, null, null, null, null, null);
if (cursor.moveToFirst()) {
do {
Turbine turbine = new Turbine();
turbine.setProjectId(cursor.getString(cursor.getColumnIndex(COLUMN_Part_PROJECT_ID)));
turbine.setTurbineId(cursor.getString(cursor.getColumnIndex(COLUMN_TURBINE_ID)));
turbine.setTurbineName(cursor.getString(cursor.getColumnIndex(COLUMN_TURBINE_NAME)));
turbines.add(turbine);
} while (cursor.moveToNext());
}
cursor.close();
return turbines;
}
@SuppressLint("Range")
public List<Turbine> searchTurbines(String query) {
List<Turbine> turbines = new ArrayList<>();
SQLiteDatabase db = this.getReadableDatabase();
String selection = COLUMN_TURBINE_NAME + " LIKE ? OR " + COLUMN_TURBINE_ID + " LIKE ?";
String[] selectionArgs = new String[]{"%" + query + "%", "%" + query + "%"};
Cursor cursor = db.query(TABLE_TURBINES,
null, selection, selectionArgs, null, null, COLUMN_TURBINE_NAME + " ASC");
if (cursor.moveToFirst()) {
do {
Turbine part = new Turbine();
part.setProjectId(cursor.getString(cursor.getColumnIndex(COLUMN_Part_PROJECT_ID)));
part.setTurbineId(cursor.getString(cursor.getColumnIndex(COLUMN_TURBINE_ID)));
part.setTurbineName(cursor.getString(cursor.getColumnIndex(COLUMN_TURBINE_NAME)));
turbines.add(part);
} while (cursor.moveToNext());
}
cursor.close();
return turbines;
}
/**
* 保存部件列表到数据库
*/
@ -361,7 +481,88 @@ public class DatabaseHelper extends SQLiteOpenHelper {
cursor.close();
return parts;
}
@SuppressLint("Range")
public List<PartResponse> getPartsByTurbineId(String turbineId) {
List<PartResponse> parts = new ArrayList<>();
SQLiteDatabase db = this.getReadableDatabase();
// 定义查询条件
String selection = COLUMN_TURBINE_ID + " = ?";
String[] selectionArgs = { turbineId };
// 执行查询
Cursor cursor = db.query(
TABLE_PARTS, // 表名
null, // 查询的列null 表示查询所有列
selection, // WHERE 条件
selectionArgs, // WHERE 条件的参数
null, // GROUP BY
null, // HAVING
null // ORDER BY
);
// 遍历查询结果
if (cursor.moveToFirst()) {
do {
PartResponse part = new PartResponse();
part.setPartId(cursor.getString(cursor.getColumnIndex(COLUMN_PART_ID)));
part.setPartName(cursor.getString(cursor.getColumnIndex(COLUMN_PART_NAME)));
part.setPartCode(cursor.getString(cursor.getColumnIndex(COLUMN_PART_CODE)));
part.setPartType(cursor.getString(cursor.getColumnIndex(COLUMN_PART_TYPE)));
part.setPartTypeLabel(cursor.getString(cursor.getColumnIndex(COLUMN_PART_TYPE_LABEL)));
part.setProjectId(cursor.getString(cursor.getColumnIndex(COLUMN_Part_PROJECT_ID)));
part.setProjectName(cursor.getString(cursor.getColumnIndex(COLUMN_Part_PROJECT_NAME)));
part.setTurbineId(cursor.getString(cursor.getColumnIndex(COLUMN_TURBINE_ID)));
part.setTurbineName(cursor.getString(cursor.getColumnIndex(COLUMN_TURBINE_NAME)));
parts.add(part);
} while (cursor.moveToNext());
}
// 关闭游标
cursor.close();
return parts;
}
@SuppressLint("Range")
public List<PartResponse> searchParts(String query) {
List<PartResponse> parts = new ArrayList<>();
SQLiteDatabase db = this.getReadableDatabase();
String selection = COLUMN_PART_NAME + " LIKE ? OR " + COLUMN_PART_ID + " LIKE ?";
String[] selectionArgs = new String[]{"%" + query + "%", "%" + query + "%"};
// 执行查询
Cursor cursor = db.query(
TABLE_PARTS, // 表名
null, // 查询的列null 表示查询所有列
selection, // WHERE 条件
selectionArgs, // WHERE 条件的参数
null, // GROUP BY
null, // HAVING
COLUMN_PART_NAME + " ASC" // ORDER BY
);
// 遍历查询结果
if (cursor.moveToFirst()) {
do {
PartResponse part = new PartResponse();
part.setPartId(cursor.getString(cursor.getColumnIndex(COLUMN_PART_ID)));
part.setPartName(cursor.getString(cursor.getColumnIndex(COLUMN_PART_NAME)));
part.setPartCode(cursor.getString(cursor.getColumnIndex(COLUMN_PART_CODE)));
part.setPartType(cursor.getString(cursor.getColumnIndex(COLUMN_PART_TYPE)));
part.setPartTypeLabel(cursor.getString(cursor.getColumnIndex(COLUMN_PART_TYPE_LABEL)));
part.setProjectId(cursor.getString(cursor.getColumnIndex(COLUMN_Part_PROJECT_ID)));
part.setProjectName(cursor.getString(cursor.getColumnIndex(COLUMN_Part_PROJECT_NAME)));
part.setTurbineId(cursor.getString(cursor.getColumnIndex(COLUMN_TURBINE_ID)));
part.setTurbineName(cursor.getString(cursor.getColumnIndex(COLUMN_TURBINE_NAME)));
parts.add(part);
} while (cursor.moveToNext());
}
cursor.close();
return parts;
}
/**
* 获取最后同步时间
*/
@ -380,13 +581,11 @@ public class DatabaseHelper extends SQLiteOpenHelper {
return lastSync;
}
// 用户操作方法
public boolean addUser(String username, String password, String projectId, String projectName) {
public boolean addUser(String username, String password) {
SQLiteDatabase db = this.getWritableDatabase();
ContentValues values = new ContentValues();
values.put(COLUMN_USERNAME, username);
values.put(COLUMN_PASSWORD, password);
values.put(COLUMN_USER_PROJECT_ID, projectId);
values.put(COLUMN_USER_PROJECT_NAME, projectName);
long result = db.insert(TABLE_USERS, null, values);
db.close();
@ -512,6 +711,7 @@ public class DatabaseHelper extends SQLiteOpenHelper {
cursor.close();
return projects;
}
public void updateLastUpdateTime(long time) {
SQLiteDatabase db = this.getWritableDatabase();
ContentValues values = new ContentValues();

View File

@ -1,98 +1,49 @@
package com.example.myapplication;
import static android.content.ContentValues.TAG;
import android.annotation.SuppressLint;
import android.app.ProgressDialog;
import android.content.ClipData;
import android.content.ClipboardManager;
import android.content.Context;
import android.content.Intent;
import android.content.SharedPreferences;
import android.graphics.Color;
import android.net.ConnectivityManager;
import android.net.NetworkCapabilities;
import android.net.NetworkInfo;
import android.os.Build;
import android.os.Bundle;
import android.os.Handler;
import android.os.Looper;
import android.util.Base64;
import android.util.Log;
import android.util.Pair;
import android.view.View;
import android.view.ViewGroup;
import android.widget.AdapterView;
import android.widget.ArrayAdapter;
import android.widget.AutoCompleteTextView;
import android.widget.Button;
import android.widget.CheckBox;
import android.widget.EditText;
import android.widget.Filter;
import android.widget.LinearLayout;
import android.widget.TextView;
import android.widget.Toast;
import androidx.annotation.NonNull;
import androidx.annotation.Nullable;
import androidx.appcompat.app.AlertDialog;
import androidx.appcompat.app.AppCompatActivity;
import com.example.myapplication.DataBase.DatabaseHelper;
import com.example.myapplication.Tool.AuthInterceptor;
import com.example.myapplication.Tool.RetrofitClient;
import com.example.myapplication.Tool.ShowError;
import com.example.myapplication.api.AuthApi;
import com.example.myapplication.api.ProjectApi;
import com.example.myapplication.api.UserApi;
import com.example.myapplication.model.ApiResponse;
import com.example.myapplication.model.ChangePasswordRequest;
import com.example.myapplication.model.LoginEntity;
import com.example.myapplication.model.LoginRequest;
import com.example.myapplication.model.PageResult;
import com.example.myapplication.model.Project;
import com.example.myapplication.model.UserInfo;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.lang.reflect.Method;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.net.SocketTimeoutException;
import java.net.UnknownHostException;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import javax.crypto.Cipher;
import javax.crypto.spec.SecretKeySpec;
import okhttp3.HttpUrl;
import okhttp3.OkHttpClient;
import okhttp3.Request;
import okhttp3.ResponseBody;
import okhttp3.logging.HttpLoggingInterceptor;
import retrofit2.Call;
import retrofit2.Callback;
import retrofit2.Response;
import retrofit2.Retrofit;
import retrofit2.adapter.rxjava2.RxJava2CallAdapterFactory;
import retrofit2.converter.gson.GsonConverterFactory;
import retrofit2.http.Body;
import retrofit2.http.Header;
import retrofit2.http.Headers;
import retrofit2.http.POST;
import retrofit2.http.PUT;
public class LoginActivity extends AppCompatActivity {
@ -101,25 +52,19 @@ public class LoginActivity extends AppCompatActivity {
private final Handler mainHandler = new Handler(Looper.getMainLooper());
private final ExecutorService executor = Executors.newFixedThreadPool(7);
private Button btnClearProject;
private Button btnLogin;
private DatabaseHelper dbHelper;
private SharedPreferences sharedPreferences;
private final String AUTH_TOKEN_KEY = "auth_token";
private static final String Name = "name";
private static final String Userid = "userid";
private static final String BASE_URL = "http://pms.dtyx.net:9158/";
private String authToken = "";
private static final String PREFS_NAME = "LoginPrefs";
private static final String PREF_USERNAME = "username";
private static final String PREF_PASSWORD = "password";
private static final String PREF_REMEMBER = "remember";
private static final String PREF_PROJECT = "project"; // 新增项目键名
private AutoCompleteTextView actvProject;
private ArrayAdapter<Project> projectAdapter;
private TextView tvSelectedProjectName;
private TextView tvSelectedProjectId;
private List<Project> projectList = new ArrayList<>();
ShowError errorShower = new ShowError(this);
@ -137,34 +82,16 @@ public class LoginActivity extends AppCompatActivity {
initViews();
setupListeners();
loadSavedPreferences();
loadProjects();
}
private void initViews() {
etUsername = findViewById(R.id.etUsername);
etPassword = findViewById(R.id.etPassword);
actvProject = findViewById(R.id.etProject);
cbRemember = findViewById(R.id.cbRemember);
btnLogin = findViewById(R.id.btnLogin);
btnClearProject = findViewById(R.id.btnClearProject);
tvSelectedProjectName = findViewById(R.id.tvSelectedProjectName);
tvSelectedProjectId = findViewById(R.id.tvSelectedProjectId);
// 初始化项目下拉框
initProjectDropdown();
}
private void setupListeners() {
// 清空项目按钮点击事件
btnClearProject.setOnClickListener(v -> {
actvProject.setText("");
tvSelectedProjectName.setText("项目名称:未选择");
tvSelectedProjectId.setText("项目ID未选择");
loadProjects();
projectAdapter.getFilter().filter("");
actvProject.showDropDown();
});
btnLogin.setOnClickListener(v -> attemptLogin());
}
private boolean isNetworkAvailable() {
@ -182,30 +109,9 @@ public class LoginActivity extends AppCompatActivity {
if (remember) {
String savedUsername = sharedPreferences.getString(PREF_USERNAME, "");
String savedPassword = sharedPreferences.getString(PREF_PASSWORD, "");
String savedProject = sharedPreferences.getString(PREF_PROJECT, "");
etUsername.setText(savedUsername);
etPassword.setText(savedPassword);
actvProject.setText(savedProject);
if (!savedProject.isEmpty()) {
try {
int lastBracketIndex = savedProject.lastIndexOf("(");
int closingBracketIndex = savedProject.lastIndexOf(")");
if (lastBracketIndex != -1 && closingBracketIndex != -1 && closingBracketIndex > lastBracketIndex) {
String projectName = savedProject.substring(0, lastBracketIndex).trim();
String projectId = savedProject.substring(lastBracketIndex + 1, closingBracketIndex).trim();
tvSelectedProjectName.setText("项目名称:" + projectName);
tvSelectedProjectId.setText("项目ID:" + projectId);
}
} catch (Exception e) {
e.printStackTrace();
tvSelectedProjectName.setText("解析错误");
tvSelectedProjectId.setText("");
}
}
cbRemember.setChecked(true);
}
}
@ -213,9 +119,8 @@ public class LoginActivity extends AppCompatActivity {
private void attemptLogin() {
String username = etUsername.getText().toString().trim();
String password = etPassword.getText().toString().trim();
String projectInput = actvProject.getText().toString().trim();
if (username.isEmpty() || password.isEmpty() || projectInput.isEmpty()) {
if (username.isEmpty() || password.isEmpty()) {
Toast.makeText(LoginActivity.this, "请填写所有字段", Toast.LENGTH_SHORT).show();
return;
}
@ -223,10 +128,10 @@ public class LoginActivity extends AppCompatActivity {
// 在后台线程执行数据库操作
// 先进行认证
if(isNetworkAvailable())
authenticateUser(username, password, projectInput);
else continueLoginProcess(username,password,projectInput);
authenticateUser(username, password);
else continueLoginProcess(username,password);
}
private void authenticateUser(String username, String password, String projectInput) {
private void authenticateUser(String username, String password) {
executor.execute(() -> {
try {
@ -281,10 +186,12 @@ public class LoginActivity extends AppCompatActivity {
&& !userResponse.body().getRows().isEmpty()) {
// 获取第一个匹配用户的name
String name = userResponse.body().getRows().get(0).getName();
String userid=userResponse.body().getRows().get(0).getUserId();
sharedPreferences.edit().putString(Name, name).apply();
sharedPreferences.edit().putString(Userid, userid).apply();
executor.execute(()-> {
continueLoginProcess(username, password, projectInput);
continueLoginProcess(username, password);
});
} else {
@ -294,7 +201,7 @@ public class LoginActivity extends AppCompatActivity {
sharedPreferences.edit().putString(Name, "").apply();
executor.execute(()-> {
continueLoginProcess(username, password, projectInput);
continueLoginProcess(username, password);
});
}
} catch (IOException e) {
@ -303,7 +210,7 @@ public class LoginActivity extends AppCompatActivity {
errorShower.showErrorDialog("出现异常:", e.getMessage());
executor.execute(()-> {
continueLoginProcess(username, password, projectInput);
continueLoginProcess(username, password);
});
}
@ -383,14 +290,11 @@ public class LoginActivity extends AppCompatActivity {
// 初始化Retrofit
Retrofit retrofit = RetrofitClient.getClient(token);
// 创建API服务
AuthApi authApi = retrofit.create(AuthApi.class);
// 执行修改密码请求
Call<ApiResponse<Void>> call = authApi.modifyPassword(request);
Response<ApiResponse<Void>> response = call.execute();
runOnUiThread(() -> {
if (response.isSuccessful() && response.body() != null) {
if (response.body().isSuccess()) {
@ -418,54 +322,15 @@ public class LoginActivity extends AppCompatActivity {
}
// 新增修改密码请求体
public static class ChangePasswordRequest {
private String account;
private String newPassword;
private String oldPassword;
public ChangePasswordRequest(String account, String newPassword, String oldPassword) {
this.account = account;
this.newPassword = newPassword;
this.oldPassword = oldPassword;
}
// getters and setters
public String getAccount() {
return account;
}
public void setAccount(String account) {
this.account = account;
}
public String getNewPassword() {
return newPassword;
}
public void setNewPassword(String newPassword) {
this.newPassword = newPassword;
}
public String getOldPassword() {
return oldPassword;
}
public void setOldPassword(String oldPassword) {
this.oldPassword = oldPassword;
}
}
private void continueLoginProcess(String username, String password, String projectInput) {
private void continueLoginProcess(String username, String password) {
// 在后台线程执行数据库操作
ExecutorService executor = Executors.newSingleThreadExecutor();
executor.execute(() -> {
try {
// 解析项目ID和名称
Pair<String, String> projectInfo = parseProjectInfo(projectInput);
String projectId = projectInfo.first;
String projectName = projectInfo.second;
// 保存登录信息
saveLoginPreferences(username, password, projectName, projectId);
saveLoginPreferences(username, password);
// 检查用户是否存在
boolean userExists = dbHelper.userExists(username);
@ -474,14 +339,14 @@ public class LoginActivity extends AppCompatActivity {
if (userExists) {
// 用户存在验证密码
if (dbHelper.checkUser(username, password)) {
handleLoginSuccess(username, projectId, projectName,password);
handleLoginSuccess(username,password);
} else {
Toast.makeText(LoginActivity.this, "密码错误", Toast.LENGTH_SHORT).show();
}
} else {
// 用户不存在自动注册
if (dbHelper.addUser(username, password, projectId, projectName)) {
handleLoginSuccess(username, projectId, projectName,password);
if (dbHelper.addUser(username, password)) {
handleLoginSuccess(username,password);
} else {
Toast.makeText(LoginActivity.this, "注册失败", Toast.LENGTH_SHORT).show();
}
@ -535,39 +400,11 @@ public class LoginActivity extends AppCompatActivity {
private Pair<String, String> parseProjectInfo(String projectInput) {
String projectId = "";
String projectName = "";
Matcher matcher = Pattern.compile(".*\\((.*)\\)").matcher(projectInput);
if (matcher.find()) {
projectId = matcher.group(1).trim();
projectName = projectInput.replace("(" + projectId + ")", "").trim();
} else {
for (Project project : projectList) {
if (project.getProjectName().equals(projectInput) ||
project.getProjectId().equals(projectInput)) {
projectId = project.getProjectId();
projectName = project.getProjectName();
break;
}
}
if (projectId.isEmpty()) {
projectName = projectInput;
}
}
return new Pair<>(projectId, projectName);
}
private void saveLoginPreferences(String username, String password, String projectName, String projectId) {
private void saveLoginPreferences(String username, String password) {
SharedPreferences.Editor editor = sharedPreferences.edit();
if (cbRemember.isChecked()) {
editor.putString(PREF_USERNAME, username);
editor.putString(PREF_PASSWORD, password);
String projectDisplayText = projectId.isEmpty() ? projectName :
(projectName + " (" + projectId + ")");
editor.putString(PREF_PROJECT, projectDisplayText);
editor.putBoolean(PREF_REMEMBER, true);
} else {
editor.clear();
@ -575,279 +412,15 @@ public class LoginActivity extends AppCompatActivity {
editor.apply();
}
private void handleLoginSuccess(String username, String projectId, String projectName,String password) {
dbHelper.updateUserProject(username, projectId, projectName);
private void handleLoginSuccess(String username, String password) {
// todo 切换时保存个人项目信息
// dbHelper.updateUserProject(username, projectId, projectName);
Toast.makeText(LoginActivity.this, "登录成功", Toast.LENGTH_SHORT).show();
Intent intent = new Intent(LoginActivity.this, MainActivity.class);
intent.putExtra("username", username);
intent.putExtra("password",password);
intent.putExtra("projectId", projectId);
intent.putExtra("projectName", projectName);
startActivity(intent);
finish();
}
private void initProjectDropdown() {
projectAdapter = new ArrayAdapter<Project>(this,
R.layout.custom_dropdown_item,
projectList) {
@NonNull
@Override
public View getView(int position, @Nullable View convertView, @NonNull ViewGroup parent) {
View view = super.getView(position, convertView, parent);
TextView textView = view.findViewById(android.R.id.text1);
Project project = getItem(position);
if (project != null) {
textView.setText(project.getProjectName() + " (" + project.getProjectId() + ")");
textView.setSingleLine(false);
textView.setMaxLines(3);
textView.setEllipsize(null);
}
return view;
}
@Override
public View getDropDownView(int position, @Nullable View convertView, @NonNull ViewGroup parent) {
return getView(position, convertView, parent);
}
@Override
public Filter getFilter() {
return new Filter() {
@Override
protected FilterResults performFiltering(CharSequence constraint) {
FilterResults results = new FilterResults();
List<Project> filteredList = new ArrayList<>();
if (constraint != null && constraint.length() > 0) {
String filterPattern = constraint.toString().toLowerCase().trim();
DatabaseHelper dbHelper = new DatabaseHelper(LoginActivity.this);
filteredList = dbHelper.searchProjects(filterPattern);
} else {
filteredList.addAll(projectList);
}
results.values = filteredList;
results.count = filteredList.size();
return results;
}
@Override
protected void publishResults(CharSequence constraint, FilterResults results) {
clear();
if (results.values != null) {
addAll((List<Project>) results.values);
}
notifyDataSetChanged();
}
};
}
};
actvProject.setAdapter(projectAdapter);
actvProject.setThreshold(0);
actvProject.setDropDownBackgroundResource(android.R.color.white);
actvProject.setOnItemClickListener((parent, view, position, id) -> {
Object item = parent.getItemAtPosition(position);
if (item instanceof Project) {
Project selectedProject = (Project) item;
tvSelectedProjectName.setText("项目名称:" + selectedProject.getProjectName());
tvSelectedProjectId.setText("项目ID" + selectedProject.getProjectId());
actvProject.setText(selectedProject.getProjectName() + " (" + selectedProject.getProjectId() + ")");
} else {
Log.e("TYPE_ERROR", "Expected Project but got: " + item.getClass());
Toast.makeText(this, "数据格式错误", Toast.LENGTH_SHORT).show();
}
});
}
private void loadProjects() {
try {
// 1. 验证接口定义
if (!validateApiDefinition()) {
return;
}
OkHttpClient.Builder okHttpClientBuilder = new OkHttpClient.Builder()
.connectTimeout(30, TimeUnit.SECONDS)
.readTimeout(30, TimeUnit.SECONDS)
.writeTimeout(30, TimeUnit.SECONDS);
okHttpClientBuilder.addInterceptor(new AuthInterceptor(authToken));
// 2. 初始化Retrofit
Retrofit retrofit = RetrofitClient.getClient(authToken);
// 3. 创建API服务
ProjectApi projectApi = retrofit.create(ProjectApi.class);
// 4. 执行网络请求
executeProjectRequest(projectApi);
} catch (Exception e) {
showErrorOnUI("初始化失败", getErrorMessage(e));
logErrorToFile(e);
}
}
private boolean validateApiDefinition() {
try {
Method method = ProjectApi.class.getMethod("getProjectList");
Type returnType = method.getGenericReturnType();
// 预期的完整类型签名
String expectedType = "retrofit2.Call<com.example.myapplication.model.ApiResponse<java.util.List<com.example.myapplication.model.Project>>>";
if (!returnType.toString().equals(expectedType)) {
String errorMsg = "接口定义错误!\n\n当前定义:\n" + returnType +
"\n\n应修改为:\n@GET(\"project/list\")\nCall<ApiResponse<List<Project>>> getProjectList();";
new Handler(Looper.getMainLooper()).post(() -> {
AlertDialog.Builder builder = new AlertDialog.Builder(this);
builder.setTitle("API接口定义不合法")
.setMessage(errorMsg)
.setPositiveButton("复制定义", (d, w) -> copyToClipboard(errorMsg))
.setNegativeButton("关闭", null)
.show();
});
return false;
}
return true;
} catch (Exception e) {
showErrorOnUI("接口验证失败", e.getMessage());
return false;
}
}
private void executeProjectRequest(ProjectApi projectApi) {
Call<ApiResponse<List<Project>>> call = projectApi.getProjectList();
call.enqueue(new Callback<ApiResponse<List<Project>>>() {
@Override
public void onResponse(Call<ApiResponse<List<Project>>> call, Response<ApiResponse<List<Project>>> response) {
if (response.isSuccessful() && response.body() != null) {
handleSuccessResponse(response.body().getData());
} else {
handleApiError(response);
}
}
@Override
public void onFailure(Call<ApiResponse<List<Project>>> call, Throwable t) {
handleNetworkError(t);
}
});
}
// === 错误处理方法 ===
private void handleSuccessResponse(List<Project> projects) {
runOnUiThread(() -> {
try {
dbHelper.saveProjects(projects);
projectList.clear();
projectList.addAll(projects);
projectAdapter.notifyDataSetChanged();
} catch (Exception e) {
showErrorOnUI("数据处理错误", e.getMessage());
}
});
}
private void handleApiError(Response<?> response) {
String errorMsg = "服务器响应错误: " + response.code();
try {
if (response.errorBody() != null) {
errorMsg += "\n" + response.errorBody().string();
}
} catch (IOException e) {
errorMsg += "\n无法读取错误详情";
}
showErrorOnUI("API请求失败", errorMsg);
loadCachedProjects();
}
private void handleNetworkError(Throwable t) {
String errorMsg = "网络错误: " + t.getMessage();
if (t instanceof SocketTimeoutException) {
errorMsg = "连接超时,请检查网络";
} else if (t instanceof UnknownHostException) {
errorMsg = "无法解析主机请检查URL";
}
showErrorOnUI("网络连接失败", errorMsg);
loadCachedProjects();
}
// === 工具方法 ===
private void showErrorOnUI(String title, String message) {
runOnUiThread(() -> {
// 在界面底部显示错误面板
ViewGroup rootView = findViewById(android.R.id.content);
LinearLayout errorPanel = new LinearLayout(this);
errorPanel.setOrientation(LinearLayout.VERTICAL);
errorPanel.setBackgroundColor(0x33FF0000);
errorPanel.setPadding(32, 16, 32, 16);
TextView tvError = new TextView(this);
tvError.setTextColor(Color.RED);
tvError.setText(title);
TextView tvMsg = new TextView(this);
tvMsg.setText(message);
errorPanel.addView(tvError);
errorPanel.addView(tvMsg);
rootView.addView(errorPanel);
// 5秒后自动消失
new Handler().postDelayed(() -> rootView.removeView(errorPanel), 5000);
});
}
private void copyToClipboard(String text) {
ClipboardManager clipboard = (ClipboardManager) getSystemService(CLIPBOARD_SERVICE);
clipboard.setPrimaryClip(ClipData.newPlainText("错误定义", text));
Toast.makeText(this, "定义已复制", Toast.LENGTH_SHORT).show();
}
private String getErrorMessage(Exception e) {
if (e instanceof IllegalArgumentException) {
return "参数错误: " + e.getMessage();
} else if (e instanceof NullPointerException) {
return "空指针异常: " + e.getMessage();
} else {
return "未知错误: " + e.getClass().getSimpleName();
}
}
private void logErrorToFile(Exception e) {
String log = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(new Date()) +
"\nError: " + e.getClass().getName() +
"\nMessage: " + e.getMessage() +
"\nStack Trace:\n" + Log.getStackTraceString(e);
try {
File file = new File(getExternalFilesDir(null), "error_log.txt");
FileWriter writer = new FileWriter(file, true);
writer.append(log).append("\n\n");
writer.close();
} catch (IOException ioException) {
Log.e("FileLog", "无法写入错误日志", ioException);
}
}
private void loadCachedProjects() {
new Thread(() -> {
List<Project> cachedProjects = dbHelper.getAllProjects();
runOnUiThread(() -> {
projectList.clear();
projectList.addAll(cachedProjects);
projectAdapter.notifyDataSetChanged();
Toast.makeText(this, "已加载本地缓存数据", Toast.LENGTH_LONG).show();
});
}).start();
}
}

File diff suppressed because it is too large Load Diff

View File

@ -1,4 +1,3 @@
package com.example.myapplication.Service;
import android.annotation.SuppressLint;
@ -18,6 +17,8 @@ import android.os.Build;
import android.os.Handler;
import android.os.IBinder;
import android.os.Looper;
import android.os.VibrationEffect;
import android.os.Vibrator;
import android.text.Editable;
import android.text.TextUtils;
import android.text.TextWatcher;
@ -75,9 +76,11 @@ public class FloatingWindowService extends Service {
private String lastUnitName = "";
private AppDatabase database;
private TurbineDao turbineDao;
private AutoCompleteTextView actvUnit; // 替换原来的EditText
private AutoCompleteTextView actvUnit;
private ArrayAdapter<Turbine> turbineAdapter;
private String projectID;
public static boolean isFloatingWindowShowing = false;
@SuppressLint("SetTextI18n")
@Override
public void onCreate() {
@ -85,19 +88,17 @@ public class FloatingWindowService extends Service {
createNotificationChannel();
database = AppDatabase.getDatabase(this);
turbineDao = database.turbineDao();
createNotificationChannel();
showFloatingWindow();
}
@Override
public int onStartCommand(Intent intent, int flags, int startId) {
if (intent != null && intent.hasExtra(EXTRA_PROJECT_ID)) {
projectID = intent.getStringExtra(EXTRA_PROJECT_ID);
}
return super.onStartCommand(intent, flags, startId);
}
private void createNotificationChannel() {
if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.O) {
NotificationChannel channel = new NotificationChannel(
@ -112,6 +113,7 @@ public class FloatingWindowService extends Service {
private void showFloatingWindow() {
// 创建前台服务通知
isFloatingWindowShowing = true;
Notification notification = new NotificationCompat.Builder(this, CHANNEL_ID)
.setContentTitle("悬浮窗服务运行中")
.setContentText("正在显示悬浮提示框")
@ -120,11 +122,9 @@ public class FloatingWindowService extends Service {
.build();
if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.Q) {
// Android 10+ 需要指定服务类型
startForeground(NOTIFICATION_ID, notification,
ServiceInfo.FOREGROUND_SERVICE_TYPE_SPECIAL_USE);
} else {
// Android 8.0-9.0
startForeground(NOTIFICATION_ID, notification);
}
@ -154,14 +154,57 @@ public class FloatingWindowService extends Service {
setupTurbineDropdown();
radioGroup = floatingView.findViewById(R.id.radioGroup);
// 设置关闭按钮点击事件
View btnClose = floatingView.findViewById(R.id.btnClose);
btnClose.setOnClickListener(v -> {
// 震动反馈
Vibrator vibrator = (Vibrator) getSystemService(VIBRATOR_SERVICE);
if (vibrator != null && vibrator.hasVibrator()) {
if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.O) {
vibrator.vibrate(VibrationEffect.createOneShot(50, VibrationEffect.DEFAULT_AMPLITUDE));
} else {
vibrator.vibrate(50);
}
}
closeFloatingWindow();
});
setupDragListener(params);
setupValueChangeListeners();
setupAutoCompleteBehavior();
}
private void closeFloatingWindow() {
if (floatingView != null && floatingView.getWindowToken() != null) {
// 添加淡出动画
floatingView.animate()
.alpha(0f)
.setDuration(200)
.withEndAction(() -> {
removeFloatingView();
stopSelf();
})
.start();
} else {
removeFloatingView();
stopSelf();
}
isFloatingWindowShowing = false;
}
private synchronized void removeFloatingView() {
try {
if (windowManager != null && floatingView != null && floatingView.getWindowToken() != null) {
windowManager.removeView(floatingView);
}
} catch (IllegalArgumentException e) {
Log.e("FloatingWindow", "View already removed", e);
} finally {
floatingView = null;
}
}
private void setupAutoCompleteBehavior() {
actvUnit.setFocusableInTouchMode(true); // 允许触摸获取焦点
actvUnit.setFocusableInTouchMode(true);
actvUnit.setFocusable(true);
actvUnit.setOnClickListener(v -> {
if (turbineAdapter.getCount() > 0) {
@ -190,6 +233,7 @@ public class FloatingWindowService extends Service {
return false;
});
}
private void showSoftInputDelayed(View view) {
view.postDelayed(() -> {
InputMethodManager imm = (InputMethodManager) getSystemService(INPUT_METHOD_SERVICE);
@ -205,52 +249,55 @@ public class FloatingWindowService extends Service {
imm.hideSoftInputFromWindow(view.getWindowToken(), 0);
}
}
// 移除 setupInputMethod() 方法因为功能已整合到上面的方法中
private void setupDragListener(WindowManager.LayoutParams params) {
floatingView.findViewById(R.id.floatingContainer).setOnTouchListener(new View.OnTouchListener() {
View dragHandle = floatingView.findViewById(R.id.floatingContainer);
dragHandle.setOnTouchListener(new View.OnTouchListener() {
private int initialX, initialY;
private float initialTouchX, initialTouchY;
@Override
public boolean onTouch(View v, MotionEvent event) {
switch (event.getAction()) {
case MotionEvent.ACTION_DOWN:
// 检查是否点击在输入区域
Rect rect = new Rect();
actvUnit.getGlobalVisibleRect(rect);
if (rect.contains((int)event.getRawX(), (int)event.getRawY())) {
// 如果是点击输入区域不处理拖动让输入框获取焦点
// 检查是否点击在关闭按钮上
if (!isViewAttached()) {
return false;
}
Rect closeButtonRect = new Rect();
floatingView.findViewById(R.id.btnClose).getGlobalVisibleRect(closeButtonRect);
if (closeButtonRect.contains((int)event.getRawX(), (int)event.getRawY())) {
return false;
}
// 记录初始位置
// 检查是否点击在输入区域
Rect inputRect = new Rect();
actvUnit.getGlobalVisibleRect(inputRect);
if (inputRect.contains((int)event.getRawX(), (int)event.getRawY())) {
return false;
}
switch (event.getAction()) {
case MotionEvent.ACTION_DOWN:
initialX = params.x;
initialY = params.y;
initialTouchX = event.getRawX();
initialTouchY = event.getRawY();
// 隐藏键盘和下拉菜单
hideSoftInput(actvUnit);
actvUnit.dismissDropDown();
return true;
case MotionEvent.ACTION_MOVE:
// 更新悬浮窗位置
params.x = initialX + (int)(event.getRawX() - initialTouchX);
params.y = initialY + (int)(event.getRawY() - initialTouchY);
windowManager.updateViewLayout(floatingView, params);
return true;
case MotionEvent.ACTION_UP:
// 拖动结束不处理焦点
return true;
}
return false;
}
});
}
private void setupTurbineDropdown() {
turbineAdapter = new ArrayAdapter<Turbine>(this,
R.layout.dropdown_item, R.id.text1) {
@ -261,7 +308,6 @@ public class FloatingWindowService extends Service {
TextView textView = view.findViewById(R.id.text1);
Turbine turbine = getItem(position);
if (turbine != null) {
// 显示更详细的信息
textView.setText(String.format(turbine.turbineName));
textView.setSingleLine(false);
textView.setMaxLines(3);
@ -277,17 +323,14 @@ public class FloatingWindowService extends Service {
FilterResults results = new FilterResults();
List<Turbine> filtered;
if (TextUtils.isEmpty(constraint)) {
// 无输入条件时显示全部数据
filtered = turbineDao.getTurbinesByProject(projectID);
} else {
// 有输入条件时执行搜索
filtered = turbineDao.searchTurbines(projectID, "%" + constraint + "%");
}
results.values = filtered;
results.count = filtered.size();
return results;
}
@Override
@ -308,11 +351,10 @@ public class FloatingWindowService extends Service {
};
actvUnit.setAdapter(turbineAdapter);
actvUnit.setThreshold(0); // 输入1个字符后开始搜索
actvUnit.setThreshold(0);
actvUnit.setDropDownHeight(ViewGroup.LayoutParams.WRAP_CONTENT);
actvUnit.setDropDownVerticalOffset(10); // 下拉框与输入框的垂直偏移
actvUnit.setDropDownVerticalOffset(10);
// 设置输入监听
actvUnit.addTextChangedListener(new TextWatcher() {
@Override public void beforeTextChanged(CharSequence s, int start, int count, int after) {}
@Override public void afterTextChanged(Editable s) {}
@ -321,9 +363,8 @@ public class FloatingWindowService extends Service {
public void onTextChanged(CharSequence s, int start, int before, int count) {
String input=s.toString().trim();
lastUnit = input;
lastUnitName = input; // 如果没有匹配项使用原始输入作为名称
lastUnitName = input;
if (turbineAdapter.getCount() > 0 && !TextUtils.isEmpty(input)) {
// 尝试查找完全匹配的项
for (int i = 0; i < turbineAdapter.getCount(); i++) {
Turbine turbine = turbineAdapter.getItem(i);
if (turbine != null && turbine.turbineName.equalsIgnoreCase(input)) {
@ -333,7 +374,7 @@ public class FloatingWindowService extends Service {
}
}
}
checkAndSendUpdate(); // 实时触发检查
checkAndSendUpdate();
}
});
@ -344,14 +385,11 @@ public class FloatingWindowService extends Service {
ExecutorService executor = Executors.newSingleThreadExecutor();
executor.execute(() -> {
try {
// 先检查本地是否有数据
List<Turbine> localTurbines = turbineDao.getTurbinesByProject(projectID);
if (localTurbines.isEmpty() || isNetworkAvailable()) {
// 如果本地无数据或有网络尝试从API获取
fetchTurbinesFromApi();
} else {
// 使用本地数据更新UI
mainHandler.post(() -> updateTurbineList(localTurbines));
}
} catch (Exception e) {
@ -359,13 +397,13 @@ public class FloatingWindowService extends Service {
}
});
}
private void updateTurbineList(List<Turbine> turbines) {
turbineAdapter.clear();
if (!turbines.isEmpty()) {
turbineAdapter.addAll(turbines);
}
// 如果输入框有内容重新过滤
if (!TextUtils.isEmpty(actvUnit.getText())) {
turbineAdapter.getFilter().filter(actvUnit.getText());
}
@ -401,16 +439,14 @@ public class FloatingWindowService extends Service {
List<Turbine> turbines = apiResponse.getData();
Log.d("API_SUCCESS", "Fetched " + turbines.size() + " turbines");
// 保存到数据库子线程
new Thread(() -> {
try {
turbineDao.deleteByProjectId(projectID);
for (Turbine turbine : turbines) {
turbine.projectId = projectID; // 确保 projectId 正确
turbine.projectId = projectID;
}
turbineDao.insertAll(turbines);
// 使用 mainHandler 更新 UI主线程
mainHandler.post(() -> {
updateTurbineList(turbines);
showToast("数据加载成功");
@ -420,19 +456,14 @@ public class FloatingWindowService extends Service {
mainHandler.post(() -> showToast("保存数据失败: " + e.getMessage()));
}
}).start();
} else {
// API 返回错误 code != 200
String errorMsg = "服务器错误: " + apiResponse.getCode() + " - " + apiResponse.getMsg();
Log.e("API_ERROR", errorMsg);
mainHandler.post(() -> showToast(errorMsg));
}
} else {
// HTTP 状态码非 200 404500
String errorMsg = "请求失败: HTTP " + response.code();
if (response.errorBody() != null) {
try {
errorMsg += "\n" + response.errorBody().string();
} catch (IOException e) {
@ -447,7 +478,6 @@ public class FloatingWindowService extends Service {
@Override
public void onFailure(Call<ApiResponse<List<Turbine>>> call, Throwable t) {
// 网络请求失败如无网络超时
String errorMsg = "网络错误: " + t.getMessage();
Log.e("NETWORK_ERROR", errorMsg, t);
mainHandler.post(() -> showToast(errorMsg));
@ -455,12 +485,11 @@ public class FloatingWindowService extends Service {
});
}
// 显示 Toast 的辅助方法
private void showToast(String message) {
Toast.makeText(this, message, Toast.LENGTH_LONG).show();
}
private void setupValueChangeListeners() {
// 监听机组号输入框变化
actvUnit.addTextChangedListener(new TextWatcher() {
@Override
public void beforeTextChanged(CharSequence s, int start, int count, int after) {}
@ -469,21 +498,19 @@ public class FloatingWindowService extends Service {
public void onTextChanged(CharSequence s, int start, int before, int count) {}
@Override
public void afterTextChanged(Editable s) {
}
public void afterTextChanged(Editable s) {}
});
// 添加下拉框项点击监听
actvUnit.setOnItemClickListener((parent, view, position, id) -> {
Turbine selected = turbineAdapter.getItem(position);
if (selected != null) {
lastUnit = selected.turbineId;
lastUnitName = selected.turbineName;
actvUnit.setText(selected.turbineName); // 显示名称
actvUnit.setText(selected.turbineName);
checkAndSendUpdate();
}
});
// 监听叶片号选择变化
radioGroup.setOnCheckedChangeListener((group, checkedId) -> {
int currentBlade = 0;
if (checkedId != -1) {
@ -499,7 +526,6 @@ public class FloatingWindowService extends Service {
}
private void checkAndSendUpdate() {
// 只有当两个值都有有效变化时才发送广播
if (!lastUnit.isEmpty() && lastBlade != -1) {
sendUpdateBroadcast();
highlightChanges();
@ -510,13 +536,12 @@ public class FloatingWindowService extends Service {
Intent intent = new Intent("com.example.myapplication.UPDATE_VALUES");
intent.putExtra("unit", lastUnit);
intent.putExtra("blade", lastBlade);
intent.putExtra("unitName", lastUnitName); // 发送机组名称
intent.putExtra("unitName", lastUnitName);
intent.setPackage(getPackageName());
sendBroadcast(intent);
}
private void highlightChanges() {
// 高亮效果 - 改变背景色然后恢复
int originalColor = actvUnit.getSolidColor();
actvUnit.setBackgroundColor(Color.YELLOW);
radioGroup.setBackgroundColor(Color.YELLOW);
@ -524,7 +549,7 @@ public class FloatingWindowService extends Service {
new Handler().postDelayed(() -> {
actvUnit.setBackgroundColor(originalColor);
radioGroup.setBackgroundColor(Color.TRANSPARENT);
}, 300); // 0.3秒后恢复
}, 300);
}
@Override
@ -532,14 +557,26 @@ public class FloatingWindowService extends Service {
super.onDestroy();
hideSoftInput(actvUnit);
actvUnit.dismissDropDown();
if (floatingView != null && windowManager != null) {
windowManager.removeView(floatingView);
}
isFloatingWindowShowing = false;
// 直接调用移除方法不再重复执行stopSelf()
removeFloatingView();
mainHandler.removeCallbacksAndMessages(null);
}
private boolean isViewAttached() {
return floatingView != null
&& floatingView.getWindowToken() != null
&& windowManager != null;
}
@Override
public IBinder onBind(Intent intent) {
return null;
}
public static void closeFloatingWindow(Context context) {
isFloatingWindowShowing = false;
Intent serviceIntent = new Intent(context, FloatingWindowService.class);
context.stopService(serviceIntent);
}
}

View File

@ -0,0 +1,558 @@
package com.example.myapplication.Service;
import android.app.Notification;
import android.app.NotificationChannel;
import android.app.NotificationManager;
import android.app.PendingIntent;
import android.app.Service;
import android.content.Context;
import android.content.Intent;
import android.content.SharedPreferences;
import android.database.ContentObserver;
import android.database.Cursor;
import android.location.Location;
import android.location.LocationManager;
import android.media.ExifInterface;
import android.net.ConnectivityManager;
import android.net.NetworkCapabilities;
import android.net.Uri;
import android.os.Build;
import android.os.Handler;
import android.os.IBinder;
import android.os.Looper;
import android.provider.MediaStore;
import android.util.Log;
import androidx.annotation.Nullable;
import androidx.core.app.NotificationCompat;
import androidx.room.Room;
import com.example.myapplication.DataBase.AppDatabase;
import com.example.myapplication.R;
import com.example.myapplication.Tool.BackgroundToast;
import com.example.myapplication.Tool.NotificationReceiver;
import com.example.myapplication.model.ImageEntity;
import com.example.myapplication.model.SharedDataManager;
import org.json.JSONObject;
import java.io.File;
import java.util.List;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import okhttp3.HttpUrl;
import okhttp3.MediaType;
import okhttp3.MultipartBody;
import okhttp3.OkHttpClient;
import okhttp3.Request;
import okhttp3.RequestBody;
import okhttp3.Response;
public class PhotoMonitoringService extends Service implements SharedDataManager.DataChangeListener {
private static final String CHANNEL_ID = "PhotoMonitoringChannel";
private static final int NOTIFICATION_ID = 101;
private static final int THUMBNAIL_SIZE = 100;
// 锁对象
private final Object monitoringLock = new Object();
private final Object uploadLock = new Object();
private long lastSuccessToastTime = 0;
private static final long TOAST_THROTTLE_INTERVAL = 1500;
// 状态标志
private boolean isMonitoring = false;
private boolean isUploading = false;
private boolean isRecording = false;
private int successCount = 0;
private int totalUploadedCount = 0; // 新增总上传成功计数
// 组件
private ContentObserver contentObserver;
private final ExecutorService executor = Executors.newFixedThreadPool(5);
private final OkHttpClient httpClient = new OkHttpClient.Builder()
.connectTimeout(30, TimeUnit.SECONDS) // 连接超时
.readTimeout(60, TimeUnit.SECONDS) // 读取超时
.writeTimeout(60, TimeUnit.SECONDS) // 写入超时
.build();
// 数据库
private AppDatabase db;
private AppDatabase db2;
// 位置相关
private Location lastKnownLocation;
private LocationManager locationManager;
// 配置参数
private String token;
private String user;
private String projectId;
private String partId;
private String partName;
private String turbineId;
private String turbineName;
private String Temperator;
private String Humidity;
private String Weather;
private String ChooseImageSource;
// 其他状态
private long lastPhotoId = 0;
private String currentAudioPath="-1";
private String imgpath="-1";
private String UserId;
@Override
public void onImageIdChanged(String newId) {
}
@Override
public void onTokenChanged(String newToken) {
}
@Override
public void onUserChanged(String newUser) {
user=newUser;
}
@Override
public void onAudioPathChanged(String newAudioPath) {
currentAudioPath=newAudioPath;
}
@Override
public void onProjectIdChanged(String newProjectId) {
projectId=newProjectId;
}
@Override
public void onPartIdChanged(String newPartId) {
partId =newPartId;
}
@Override
public void onPartNameChanged(String newPartName) {
partName = newPartName;
}
@Override
public void onTurbineIdChanged(String newTurbineId) {
turbineId =newTurbineId;
}
@Override
public void onTurbineNameChanged(String newTurbineName) {
turbineName = newTurbineName;
}
@Override
public void onChooseImageSourceChanged(String newChooseImageSource) {
ChooseImageSource=newChooseImageSource;
}
@Override
public void onIsRecordingChanged(boolean newisRecording) {
isRecording=newisRecording;
}
@Override
public void onUploadingStatusChanged(boolean isUploading) {
this.isUploading=isUploading;
}
@Override
public void onLocationChanged(Location newLocation) {
lastKnownLocation=newLocation;
}
@Override
public void onOtherMsgChanged(String msg) {
}
SharedDataManager dataManager;
@Override
public void onCreate() {
super.onCreate();
createNotificationChannel();
SharedPreferences prefs = getSharedPreferences("WeatherPrefs", MODE_PRIVATE);
Weather = prefs.getString("weather", "0");
Temperator = prefs.getString("temperature", "0");
Humidity = prefs.getString("humidity", "0");
SharedPreferences sharedPreferences = getSharedPreferences("LoginPrefs", Context.MODE_PRIVATE);
// 初始化数据库
UserId=sharedPreferences.getString("userid","-1");
db = Room.databaseBuilder(getApplicationContext(),
AppDatabase.class, "image-database").build();
db2 = Room.databaseBuilder(getApplicationContext(),
AppDatabase.class, "image-database2").build();
// 初始化位置管理器
locationManager = (LocationManager) getSystemService(Context.LOCATION_SERVICE);
dataManager=SharedDataManager.getInstance();
dataManager.addListener(this);
updateFromSharedData();
// 初始化时从数据库获取已上传成功的数量
totalUploadedCount = 0;
updateNotification("partId"+partId);
}
private void updateFromSharedData() {
dataManager = SharedDataManager.getInstance();
this.token = dataManager.getToken();
this.user = dataManager.getUser();
this.currentAudioPath = dataManager.getAudioPath();
this.projectId = dataManager.getProjectId();
this.partId = dataManager.getPartId();
this.turbineId = dataManager.getTurbineId();
this.ChooseImageSource = dataManager.getChooseImageSource();
this.isUploading=dataManager.getisUploading();
this.lastKnownLocation=dataManager.getLocation();
}
@Override
public int onStartCommand(Intent intent, int flags, int startId) {
if (intent != null && "start_monitoring".equals(intent.getStringExtra("command"))) {
startMonitoring();
}
startForeground(NOTIFICATION_ID, buildNotification("服务正在运行", "监听照片变化中"));
return START_STICKY;
}
@Nullable
@Override
public IBinder onBind(Intent intent) {
return null;
}
@Override
public void onDestroy() {
stopMonitoring();
super.onDestroy();
}
public void startMonitoring() {
synchronized (monitoringLock) {
if (isMonitoring) return;
lastPhotoId = getLastPhotoId();
contentObserver = new ContentObserver(new Handler(Looper.getMainLooper())) {
@Override
public void onChange(boolean selfChange, Uri uri) {
super.onChange(selfChange, uri);
if (!isRecording && uri != null && uri.toString().contains("images/media")) {
checkNewPhotos();
}
}
};
getContentResolver().registerContentObserver(
MediaStore.Images.Media.EXTERNAL_CONTENT_URI,
true,
contentObserver
);
isMonitoring = true;
executor.execute(() -> {
List<ImageEntity> history = db2.imageDao().getAll();
if (history != null && !history.isEmpty()) {
synchronized (uploadLock) {
for (ImageEntity image : history) {
uploadPhoto(image.path, image.time, lastKnownLocation, image.user,
image.audioPath, image.project, image.partId, image.turbineId,
image.imageSource);
}
}
}
db2.imageDao().deleteAll();
});
}
}
public void stopMonitoring() {
synchronized (monitoringLock) {
if (!isMonitoring) return;
if (contentObserver != null) {
getContentResolver().unregisterContentObserver(contentObserver);
contentObserver = null;
}
isMonitoring=false;
}
}
private void checkNewPhotos() {
String[] projection = {
MediaStore.Images.Media._ID,
MediaStore.Images.Media.DATA,
MediaStore.Images.Media.DATE_ADDED,
MediaStore.Images.Media.DATE_TAKEN,
MediaStore.Images.Media.LATITUDE,
MediaStore.Images.Media.LONGITUDE
};
String selection = MediaStore.Images.Media._ID + " > ? AND " +
MediaStore.Images.Media.DATE_TAKEN + " > 0 AND " +
"(" + MediaStore.Images.Media.DATA + " LIKE '%/DCIM/%' OR " +
MediaStore.Images.Media.DATA + " LIKE '%/Camera/%')";
String[] selectionArgs = new String[]{String.valueOf(lastPhotoId)};
try (Cursor cursor = getContentResolver().query(
MediaStore.Images.Media.EXTERNAL_CONTENT_URI,
projection,
selection,
selectionArgs,
MediaStore.Images.Media._ID + " DESC"
)) {
if (cursor != null) {
while (cursor.moveToNext()) {
processNewPhoto(cursor);
}
}
} catch (Exception e) {
Log.e("PhotoMonitoringService", "Error checking new photos", e);
}
}
private void processNewPhoto(Cursor cursor) {
long id = cursor.getLong(0);
String path = cursor.getString(1);
imgpath = path;
long time = cursor.getLong(2) * 1000;
double latitude = cursor.getDouble(3);
double longitude = cursor.getDouble(4);
double altitude = 0.0;
if (lastKnownLocation != null) {
latitude = lastKnownLocation.getLatitude();
longitude = lastKnownLocation.getLongitude();
altitude = lastKnownLocation.getAltitude();
}
try {
ExifInterface exif = new ExifInterface(path);
String altitudeStr = exif.getAttribute(ExifInterface.TAG_GPS_ALTITUDE);
if (altitudeStr != null) {
altitude = Double.parseDouble(altitudeStr);
}
} catch (Exception e) {
Log.e("PhotoMonitoringService", "Error reading EXIF data", e);
}
String audioPath = currentAudioPath;
lastPhotoId = Math.max(lastPhotoId, id);
if (isNetworkAvailable() && !isRecording) {
uploadNewPhoto(path, time, audioPath);
} else {
queuePhotoForLater(path, time, audioPath, latitude, longitude, altitude);
}
}
private void uploadNewPhoto(String path, long time, String audioPath) {
executor.execute(() -> {
List<ImageEntity> history = db2.imageDao().getAll();
if (history != null && !history.isEmpty()) {
synchronized (uploadLock) {
for (ImageEntity image : history) {
uploadPhoto(image.path, image.time, lastKnownLocation,
image.user, image.audioPath, image.project,
image.turbineId, image.partId, image.imageSource);
}
}
}
db2.imageDao().deleteAll();
uploadPhoto(path, time, lastKnownLocation, user,
audioPath, projectId, partId, turbineId, partId);
});
}
private void queuePhotoForLater(String path, long time, String audioPath,
double latitude, double longitude, double altitude) {
executor.execute(() -> {
db2.imageDao().insert(new ImageEntity(path, time, latitude, longitude, altitude,
user, audioPath, projectId, partId, partName, turbineId, turbineName, false,
Temperator, Humidity, Weather, ChooseImageSource));
});
}
private void uploadPhoto(String filePath, long time, Location location, String user,
String audioPath, String project, String turbineId,
String partId, String imageSource) {
synchronized (uploadLock) {
if (isUploading) {
queuePhotoForLater(filePath, time, audioPath,
location != null ? location.getLatitude() : 0,
location != null ? location.getLongitude() : 0,
location != null ? location.getAltitude() : 0);
return;
}
}
executor.execute(() -> {
try {
File imageFile = new File(filePath);
if (!imageFile.exists()) {
return;
}
double latitude = location != null ? location.getLatitude() : 0;
double longitude = location != null ? location.getLongitude() : 0;
double altitude = location != null ? location.getAltitude() : 0;
// 构建请求URL
HttpUrl.Builder urlBuilder = HttpUrl.parse("http://pms.dtyx.net:9158/image/" + imageSource + "/upload/" + partId)
.newBuilder()
.addQueryParameter("collectorId",UserId)
.addQueryParameter("collectorName", user)
.addQueryParameter("humidness", String.valueOf(Humidity))
.addQueryParameter("shootingDistance", String.valueOf(0)) // 根据需要设置实际值
.addQueryParameter("temperatureMax", String.valueOf(Temperator))
.addQueryParameter("temperatureMin", String.valueOf(Temperator))
.addQueryParameter("weather", Weather)
.addQueryParameter("windLevel", String.valueOf(0)); // 根据需要设置实际值
// 创建请求体
RequestBody requestBody = new MultipartBody.Builder()
.setType(MultipartBody.FORM)
.addFormDataPart("file", imageFile.getName(),
RequestBody.create(imageFile, MediaType.parse("image/*")))
.build();
Request request = new Request.Builder()
.url(urlBuilder.build())
.post(requestBody)
.addHeader("Authorization", token)
.addHeader("Content-Type", "application/x-www-form-urlencoded")
.build();
try (Response response = httpClient.newCall(request).execute()) {
if (response.isSuccessful()) {
// 解析响应数据
String responseData = response.body().string();
JSONObject jsonResponse = new JSONObject(responseData);
JSONObject data = jsonResponse.optJSONObject("data");
System.out.println(responseData);
String imageId = data != null ? data.optString("imageId") : "";
dataManager.setImageId(imageId);
dataManager.setOthermsg("partid:"+partId+"\n"+"imageid:"+imageId+"\n"+"response:"+responseData);
db.imageDao().insert(new ImageEntity(filePath, time, latitude, longitude, altitude,
user, audioPath, project, partId, partName, turbineId, turbineName, true,
Temperator, Humidity, Weather, ChooseImageSource));
totalUploadedCount++;
updateNotification(imageId);
Intent intent = new Intent("RESPONSE_ACTION");
intent.putExtra("response", response.toString());
sendBroadcast(intent);
long currentTime = System.currentTimeMillis();
if (currentTime - lastSuccessToastTime > TOAST_THROTTLE_INTERVAL) {
BackgroundToast.show(this, "图片上传成功,已上传" + totalUploadedCount + "");
lastSuccessToastTime = currentTime;
}
} else {
db2.imageDao().insert(new ImageEntity(filePath, time, latitude, longitude, altitude,
user, audioPath, project, partId, partName, turbineId, turbineName, true,
Temperator, Humidity, Weather, ChooseImageSource));
dataManager.setImageId("-1");
}
}
} catch (Exception e) {
db2.imageDao().insert(new ImageEntity(filePath, time, 0, 0, 0,
user, audioPath, project, partId, partName, turbineId, turbineName, true,
Temperator, Humidity, Weather, ChooseImageSource));
dataManager.setImageId("-1");
}
});
}
private void updateNotification(String msg) {
String content = "已成功上传 " + totalUploadedCount + " 张照片"+msg;
Notification notification = buildNotification("照片上传服务运行中", content);
NotificationManager manager = (NotificationManager) getSystemService(Context.NOTIFICATION_SERVICE);
if (manager != null) {
manager.notify(NOTIFICATION_ID, notification);
}
}
private boolean isNetworkAvailable() {
ConnectivityManager cm = (ConnectivityManager) getSystemService(Context.CONNECTIVITY_SERVICE);
if (cm == null) return false;
NetworkCapabilities capabilities = cm.getNetworkCapabilities(cm.getActiveNetwork());
return capabilities != null &&
(capabilities.hasTransport(NetworkCapabilities.TRANSPORT_WIFI) ||
capabilities.hasTransport(NetworkCapabilities.TRANSPORT_CELLULAR) ||
capabilities.hasTransport(NetworkCapabilities.TRANSPORT_ETHERNET));
}
private void createNotificationChannel() {
if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.O) {
NotificationChannel channel = new NotificationChannel(
CHANNEL_ID,
"照片上传服务",
NotificationManager.IMPORTANCE_HIGH
);
channel.setDescription("显示照片上传状态和进度");
NotificationManager manager = getSystemService(NotificationManager.class);
manager.createNotificationChannel(channel);
}
}
private Notification buildNotification(String title, String content) {
return new NotificationCompat.Builder(this, CHANNEL_ID)
.setContentTitle(title)
.setContentText(content)
.setSmallIcon(R.drawable.ic_upload)
.setPriority(NotificationCompat.PRIORITY_HIGH)
.setOngoing(true)
.setOnlyAlertOnce(true)
.addAction(createStopAction())
.build();
}
private NotificationCompat.Action createStopAction() {
Intent stopIntent = new Intent(this, NotificationReceiver.class);
stopIntent.setAction("ACTION_STOP_SERVICE");
PendingIntent stopPendingIntent = PendingIntent.getBroadcast(
this,
0,
stopIntent,
PendingIntent.FLAG_IMMUTABLE | PendingIntent.FLAG_UPDATE_CURRENT);
return new NotificationCompat.Action(
R.drawable.ic_stop, "停止", stopPendingIntent);
}
private long getLastPhotoId() {
long id = -1;
Cursor cursor = getContentResolver().query(
MediaStore.Images.Media.EXTERNAL_CONTENT_URI,
new String[]{"MAX(" + MediaStore.Images.Media._ID + ")"},
null,
null,
null
);
if (cursor != null && cursor.moveToFirst()) {
id = cursor.getLong(0);
cursor.close();
}
return id;
}
}

View File

@ -7,8 +7,13 @@ import android.app.Service;
import android.content.Intent;
import android.content.pm.PackageManager;
import android.content.pm.ServiceInfo;
import android.location.Location;
import android.media.MediaRecorder;
import android.os.Build;
import android.os.Handler;
import android.os.IBinder;
import android.os.Looper;
import android.util.Log;
import android.widget.Toast;
import androidx.annotation.RequiresApi;
@ -16,12 +21,60 @@ import androidx.core.app.ActivityCompat;
import androidx.core.app.NotificationCompat;
import androidx.core.app.NotificationManagerCompat;
import androidx.lifecycle.Observer;
import androidx.lifecycle.ViewModelProvider;
import androidx.room.Room;
import com.example.myapplication.DataBase.AppDatabase;
import com.example.myapplication.R;
import com.example.myapplication.Tool.BackgroundToast;
import com.example.myapplication.model.AudioEntity;
import com.example.myapplication.model.SharedDataManager;
public class RecordingService extends Service {
import java.io.File;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Locale;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import io.reactivex.disposables.Disposable;
import okhttp3.MediaType;
import okhttp3.MultipartBody;
import okhttp3.OkHttpClient;
import okhttp3.Request;
import okhttp3.RequestBody;
import okhttp3.Response;
public class RecordingService extends Service implements SharedDataManager.DataChangeListener {
private static final String CHANNEL_ID = "recording_channel";
private static final int NOTIFICATION_ID = 1;
private static final int NOTIFICATION_ID = 2;
private MediaRecorder mediaRecorder;
private String currentAudioPath;
private Handler handler;
private Runnable updateTimerRunnable;
private AppDatabase db;
private AppDatabase db2;
private final ExecutorService executor = Executors.newFixedThreadPool(4);
private String ImageId="-1";
private String currentToken="-1";
private SharedDataManager dataManager;
@Override
public IBinder onBind(Intent intent) {
return null;
}
private final OkHttpClient httpClient = new OkHttpClient.Builder()
.connectTimeout(30, TimeUnit.SECONDS) // 连接超时
.readTimeout(60, TimeUnit.SECONDS) // 读取超时
.writeTimeout(60, TimeUnit.SECONDS) // 写入超时
.build();
private int recordingSeconds = 0;
@RequiresApi(api = Build.VERSION_CODES.R)
@Override
public void onCreate() {
@ -43,19 +96,271 @@ public class RecordingService extends Service {
Toast.makeText(this,"错误:"+e,Toast.LENGTH_LONG).show();
stopSelf();
}
handler = new Handler(Looper.getMainLooper());
db = Room.databaseBuilder(getApplicationContext(),
AppDatabase.class, "image-database").build();
db2 = Room.databaseBuilder(getApplicationContext(),
AppDatabase.class, "image-database2").build();
createNotificationChannel();
// onCreate() 中改进 ViewModel 初始化
// 观察LiveData变化
// 在类成员变量中添加
dataManager=SharedDataManager.getInstance();
dataManager.addListener(this);
currentToken = dataManager.getToken();
ImageId=dataManager.getImageId();
// 在onCreate()中改为使用成员变量观察
}
@Override
public void onImageIdChanged(String newId) {
ImageId=newId;
}
@Override
public void onTokenChanged(String newToken) {
currentToken = newToken;
Log.d("Service", "Token updated: " + newToken);
}
@Override
public void onUserChanged(String newUser) {
}
@Override
public void onAudioPathChanged(String newAudioPath) {
}
@Override
public void onProjectIdChanged(String newProjectId) {
}
@Override
public void onPartIdChanged(String newUnit) {
}
@Override
public void onTurbineIdChanged(String newTurbineId) {
}
@Override
public void onPartNameChanged(String newPartName) {
}
@Override
public void onTurbineNameChanged(String newTurbineName) {
}
@Override
public void onChooseImageSourceChanged(String newChooseImageSource) {
}
@Override
public void onIsRecordingChanged(boolean newisRecording) {
}
@Override
public void onUploadingStatusChanged(boolean isUploading) {
}
@Override
public void onLocationChanged(Location newLocation) {
}
@Override
public void onOtherMsgChanged(String msg) {
}
// 获取当前数据可选如果 Activity 需要读取
@Override
public int onStartCommand(Intent intent, int flags, int startId) {
return START_STICKY;
if (intent != null && "start_recording".equals(intent.getAction())) {
startRecording();
} else if (intent != null && "stop_recording".equals(intent.getAction())) {
stopRecording();
}
return START_REDELIVER_INTENT;
}
private void startRecording() {
try {
dataManager.setIsRecording(true);
// 1. 设置录音文件路径
currentAudioPath = getExternalCacheDir().getAbsolutePath()
+ "/" + System.currentTimeMillis() + ".3gp";
dataManager.setAudioPath(currentAudioPath);
// 2. 初始化MediaRecorder
mediaRecorder = new MediaRecorder();
mediaRecorder.setAudioSource(MediaRecorder.AudioSource.MIC);
mediaRecorder.setOutputFormat(MediaRecorder.OutputFormat.THREE_GPP);
mediaRecorder.setAudioEncoder(MediaRecorder.AudioEncoder.AMR_NB);
mediaRecorder.setOutputFile(currentAudioPath);
mediaRecorder.prepare();
mediaRecorder.start();
// 3. 启动前台服务
startForegroundService();
// 4. 开始计时更新通知
startUpdatingTimer();
} catch (Exception e) {
Toast.makeText(this, "录音失败: " + e.getMessage(), Toast.LENGTH_SHORT).show();
stopSelf();
}
}
@Override
public IBinder onBind(Intent intent) {
return null;
private void startForegroundService() {
Notification notification = buildNotification("录音中... 00:00");
if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.Q) {
startForeground(NOTIFICATION_ID, notification,
ServiceInfo.FOREGROUND_SERVICE_TYPE_MICROPHONE);
} else {
startForeground(NOTIFICATION_ID, notification);
}
}
private void startUpdatingTimer() {
updateTimerRunnable = new Runnable() {
@Override
public void run() {
recordingSeconds++;
String time = String.format("%02d:%02d",
recordingSeconds / 60, recordingSeconds % 60);
updateNotification("录音中... " + time);
handler.postDelayed(this, 1000);
}
};
handler.post(updateTimerRunnable);
}
private void uploadAudio(String imageId, String audioPath, String token) {
// 1. 参数校验在调用线程执行避免不必要的后台任务
if (audioPath == null || audioPath.equals("0")) {
BackgroundToast.show(this, "文件路径为空");
return;
}
File audioFile = new File(audioPath);
if (!audioFile.exists()) {
BackgroundToast.show(this, "文件不存在");
return;
}
// 2. 在后台线程执行上传逻辑
executor.execute(() -> {
// 3. 构建请求体已在后台线程
RequestBody requestBody = new MultipartBody.Builder()
.setType(MultipartBody.FORM)
.addFormDataPart("file", audioFile.getName(),
RequestBody.create(audioFile, MediaType.parse("audio/*")))
.build();
// 4. 构建请求
Request.Builder requestBuilder = new Request.Builder()
.url("http://pms.dtyx.net:9158/audio/upload/" + imageId)
.post(requestBody);
if (token != null && !token.isEmpty()) {
requestBuilder.addHeader("Authorization", token);
}
// 5. 同步执行请求try-with-resources 自动关闭 Response
try (Response response = httpClient.newCall(requestBuilder.build()).execute()) {
String currentTime = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss", Locale.getDefault())
.format(new Date());
if (response.isSuccessful()) {
// 上传成功
executor.execute(()->{ db.AudioDao().deleteByPath(audioPath);
db2.AudioDao().insert(new AudioEntity(audioPath, imageId, currentTime,true));});
BackgroundToast.show(this, "录音上传成功: " + imageId);
stopSelf();
} else {
// 上传失败HTTP 错误
executor.execute(()->{ db.AudioDao().insert(new AudioEntity(audioPath, imageId, currentTime,false));});
BackgroundToast.show(this,
"录音上传失败: " + response.code() + " " + response.message());
stopSelf();
}
} catch (IOException e) {
// 网络异常
String currentTime = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss", Locale.getDefault())
.format(new Date());
executor.execute(() -> {
db.AudioDao().insert(new AudioEntity(currentAudioPath, ImageId, currentToken, false));
});
BackgroundToast.show(this, "录音上传失败: " + e.getMessage());
Log.e("录音上传失败:",e.toString());
stopSelf();
}
});
}
private void stopRecording() {
try {
// 1. 停止录音
if (mediaRecorder != null) {
mediaRecorder.stop();
mediaRecorder.release();
mediaRecorder = null;
}
dataManager.setIsRecording(false);
// 2. 停止计时器
if (handler != null && updateTimerRunnable != null) {
handler.removeCallbacks(updateTimerRunnable);
}
// 3. 更新通知
updateNotification("录音已保存");
// 4. 上传录音文件如果有需要
if (currentAudioPath != null && ImageId != "-1") {
BackgroundToast.show(this,"开始上传录音了");
uploadAudio(ImageId, currentAudioPath, currentToken);
} else
{
BackgroundToast.show(this,"Imageid不存在上传录音失败");
executor.execute(() -> {
db.AudioDao().insert(new AudioEntity(currentAudioPath, ImageId, currentToken, false));
});
}
// 5. 不立即停止服务等待上传完成
// 可以通过其他机制如广播或LiveData在上传完成后通知服务停止
} catch (Exception e) {
Toast.makeText(this, "停止录音失败: " + e.getMessage(), Toast.LENGTH_SHORT).show();
dataManager.setIsRecording(false);
stopSelf();
}
}
private void createNotificationChannel() {
if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.O) {
@ -70,7 +375,7 @@ public class RecordingService extends Service {
}
}
private Notification buildNotification(String text) {
public Notification buildNotification(String text) {
// 确保图标有效R.drawable.ic_mic_on必须存在
return new NotificationCompat.Builder(this, CHANNEL_ID)
.setSmallIcon(R.drawable.ic_mic_on)
@ -92,4 +397,35 @@ public class RecordingService extends Service {
NotificationManagerCompat.from(this)
.notify(NOTIFICATION_ID, buildNotification(text));
}
@Override
public void onDestroy() {
super.onDestroy();
// 2. 停止并释放MediaRecorder
if (mediaRecorder != null) {
try {
mediaRecorder.stop();
mediaRecorder.release();
} catch (IllegalStateException e) {
Log.e("Service", "MediaRecorder释放异常", e);
}
mediaRecorder = null;
}
// 3. 移除Handler回调
if (handler != null && updateTimerRunnable != null) {
handler.removeCallbacks(updateTimerRunnable);
}
dataManager.removeListener(this);
// 5. 关闭数据库如果Room支持
if (db != null) {
db.close();
}
if (db2 != null) {
db2.close();
}
}
}

View File

@ -3,24 +3,31 @@ package com.example.myapplication.Tool;
import android.content.Context;
import android.os.Handler;
import android.os.Looper;
import android.view.Gravity;
import android.widget.Toast;
public class BackgroundToast {
private static Handler handler;
private static Toast currentToast;
private static long lastShowTime = 0;
public static void show(final Context context, final String message) {
if (handler == null) {
handler = new Handler(Looper.getMainLooper());
}
long currentTime = System.currentTimeMillis();
handler.post(() -> {
if (currentToast != null) {
currentToast.cancel();
}
currentToast = Toast.makeText(context.getApplicationContext(), message, Toast.LENGTH_SHORT);
currentToast = Toast.makeText(context.getApplicationContext(), message, Toast.LENGTH_LONG);
currentToast.setGravity(Gravity.CENTER, 0, 0);
currentToast.show();
lastShowTime = currentTime;
});
}
}

View File

@ -1,75 +0,0 @@
package com.example.myapplication.Tool;
import android.content.Context;
import android.util.Log;
import android.widget.Toast;
import com.chaquo.python.PyObject;
import com.chaquo.python.Python;
import java.io.BufferedReader;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
public class DocxHelper {
private static final String TAG = "DocxHelper";
public static File convertJsonToDocx(Context context, String jsonAssetName, String outputFileName) {
// 1. 从assets读取JSON数据
String jsonData = readJsonFromAssets(context, jsonAssetName);
if (jsonData == null) {
Toast.makeText(context, "Failed to read JSON from assets",Toast.LENGTH_SHORT).show();
return null;
}
// 2. 准备输出文件路径
File outputFile = new File(context.getExternalFilesDir(null), outputFileName);
String outputPath = outputFile.getAbsolutePath();
// 3. 调用Python处理
try {
Python py = Python.getInstance();
PyObject module = py.getModule("test");
// 调用Python函数
module.callAttr("json_to_docx", jsonData, outputPath);
Toast.makeText(context, "DOCX file generated at:"+outputPath, Toast.LENGTH_SHORT).show();
} catch (Exception e) {
Log.e(TAG, "Python processing failed", e);
}
finally {
return outputFile;
}
}
private static String readJsonFromAssets(Context context, String filename) {
try {
// 1. 打开assets中的文件输入流
InputStream is = context.getAssets().open(filename);
// 2. 创建缓冲读取器
BufferedReader reader = new BufferedReader(new InputStreamReader(is));
// 3. 使用StringBuilder高效拼接字符串
StringBuilder sb = new StringBuilder();
String line;
// 4. 逐行读取文件内容
while ((line = reader.readLine()) != null) {
sb.append(line);
}
// 5. 关闭资源
reader.close();
// 6. 返回完整的JSON字符串
return sb.toString();
} catch (IOException e) {
Toast.makeText(context, "Error reading JSON file"+e.getMessage(), Toast.LENGTH_SHORT).show();
return null;
}
}
}

View File

@ -79,9 +79,9 @@ public class DynamicDataFetcher {
public void onResponse(Call<ApiResponse<List<Map<String, String>>>> call,
Response<ApiResponse<List<Map<String, String>>>> response) {
if (response.isSuccessful() && response.body() != null && response.body().isSuccess()) {
Log.e("图片源列表:",response.body().toString());
List<Map<String, String>> result = response.body().getData();
Log.e("result",result.toString());
// 保存到数据库带数据类型标识
dbHelper.saveDynamicData(dataType, result);
callback.onSuccess(result);

View File

@ -1,87 +0,0 @@
package com.example.myapplication.Tool;
import android.content.Context;
import android.graphics.Color;
import android.util.SparseBooleanArray;
import android.view.View;
import android.view.ViewGroup;
import android.widget.BaseAdapter;
import android.widget.GridView;
import android.widget.ImageView;
import com.example.myapplication.model.ImageInfo;
import java.util.ArrayList;
import java.util.List;
public class ImageAdapter extends BaseAdapter {
private final Context context;
private final List<ImageInfo> imageList;
private final SparseBooleanArray selectedItems;
public ImageAdapter(Context context, List<ImageInfo> imageList) {
this.context = context;
this.imageList = imageList;
this.selectedItems = new SparseBooleanArray();
}
@Override
public int getCount() {
return imageList.size();
}
@Override
public Object getItem(int position) {
return imageList.get(position);
}
@Override
public long getItemId(int position) {
return position;
}
@Override
public View getView(int position, View convertView, ViewGroup parent) {
ImageView imageView;
if (convertView == null) {
imageView = new ImageView(context);
imageView.setLayoutParams(new GridView.LayoutParams(200, 200));
imageView.setScaleType(ImageView.ScaleType.CENTER_CROP);
imageView.setPadding(5, 5, 5, 5);
} else {
imageView = (ImageView) convertView;
}
ImageInfo imageInfo = imageList.get(position);
imageView.setImageBitmap(imageInfo.thumbnail);
// 设置选中状态
if (selectedItems.get(position, false)) {
imageView.setBackgroundColor(Color.BLUE);
} else {
imageView.setBackgroundColor(Color.TRANSPARENT);
}
imageView.setOnClickListener(v -> {
if (selectedItems.get(position, false)) {
selectedItems.delete(position);
imageView.setBackgroundColor(Color.TRANSPARENT);
} else {
selectedItems.put(position, true);
imageView.setBackgroundColor(Color.BLUE);
}
});
return imageView;
}
public List<ImageInfo> getSelectedImages() {
List<ImageInfo> selected = new ArrayList<>();
for (int i = 0; i < imageList.size(); i++) {
if (selectedItems.get(i, false)) {
selected.add(imageList.get(i));
}
}
return selected;
}
}

View File

@ -0,0 +1,50 @@
package com.example.myapplication.Tool;
import static android.content.Context.ACTIVITY_SERVICE;
import android.app.ActivityManager;
import android.app.NotificationManager;
import android.content.BroadcastReceiver;
import android.content.Context;
import android.content.Intent;
import android.util.Log;
import com.example.myapplication.Service.PhotoMonitoringService;
// 更健壮的广播接收器实现示例
public class NotificationReceiver extends BroadcastReceiver {
@Override
public void onReceive(Context context, Intent intent) {
if (intent == null) return;
try {
if ("ACTION_STOP_SERVICE".equals(intent.getAction())) {
// 防止重复操作
if (isServiceRunning(context)) {
context.stopService(new Intent(context, PhotoMonitoringService.class));
}
// 清除所有相关通知
NotificationManager manager = context.getSystemService(NotificationManager.class);
manager.cancelAll(); // 或指定ID
// 可选发送结果广播
Intent doneIntent = new Intent("ACTION_SERVICE_STOPPED");
context.sendBroadcast(doneIntent);
}
} catch (Exception e) {
Log.e("NotificationReceiver", "Error handling action", e);
}
}
private boolean isServiceRunning(Context context) {
ActivityManager manager = (ActivityManager) context.getSystemService(ACTIVITY_SERVICE);
for (ActivityManager.RunningServiceInfo service :
manager.getRunningServices(Integer.MAX_VALUE)) {
if (PhotoMonitoringService.class.getName().equals(service.service.getClassName())) {
return true;
}
}
return false;
}
}

View File

@ -64,6 +64,7 @@ public class PartListFetcher {
}
}
/**
* 异步获取部件列表
* @param callback 回调接口

View File

@ -0,0 +1,183 @@
package com.example.myapplication.Tool;
import android.content.Context;
import android.util.Log;
import com.chaquo.python.Python;
import com.chaquo.python.android.AndroidPlatform;
import org.json.JSONException;
import org.json.JSONObject;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Locale;
public class ReportGenerator {
private static final String TAG = "ReportGenerator";
private static final String TEMPLATE_DIR = "templates";
private final Context context;
public ReportGenerator(Context context) {
this.context = context;
initPython();
prepareTemplates();
}
private void initPython() {
if (!Python.isStarted()) {
Python.start(new AndroidPlatform(context));
}
}
private void prepareTemplates() {
File templateDir = new File(context.getFilesDir(), TEMPLATE_DIR);
if (!templateDir.exists() || isTemplateUpdateNeeded(templateDir)) {
copyAssetsTemplates(templateDir);
}
}
private boolean isTemplateUpdateNeeded(File templateDir) {
return templateDir.listFiles() == null || templateDir.listFiles().length == 0;
}
private void copyAssetsTemplates(File targetDir) {
try {
if (!targetDir.exists()) {
targetDir.mkdirs();
}
String[] files = context.getAssets().list(TEMPLATE_DIR);
if (files != null) {
for (String filename : files) {
File outFile = new File(targetDir, filename);
try (InputStream in = context.getAssets().open(TEMPLATE_DIR + "/" + filename);
OutputStream out = new FileOutputStream(outFile)) {
byte[] buffer = new byte[1024];
int read;
while ((read = in.read(buffer)) != -1) {
out.write(buffer, 0, read);
}
}
}
}
} catch (IOException e) {
Log.e(TAG, "Failed to copy templates", e);
}
}
public String generateReport() {
try {
// 创建模拟数据
JSONObject jsonData1 = createMockJsonData1();
JSONObject jsonData2 = createMockJsonData2();
Python py = Python.getInstance();
py.getModule("report_generate_server.Generate_Report")
.callAttr("main",
jsonData1.toString(),
jsonData2.toString());
return jsonData2.getString("shengcheng_dir");
} catch (Exception e) {
Log.e(TAG, "Report generation failed", e);
return "Error: " + e.toString();
}
}
private JSONObject createMockJsonData1() throws JSONException {
JSONObject jsonData1 = new JSONObject();
// 项目基本信息
JSONObject jizuData = new JSONObject();
jizuData.put("turbineName", "风力发电机001");
JSONObject projectData = new JSONObject();
projectData.put("fengmian_dir", "storage/emulated/0/DCIM/Camera/IMG_20250721_173921.jpg");
projectData.put("farmName", "华北风电场");
projectData.put("inspectionUnit", "华北检测公司");
projectData.put("inspectionContact", "李检测员");
projectData.put("inspectionPhone", "13800138000");
projectData.put("farmAddress", "河北省张家口市");
projectData.put("client", "国家电网");
projectData.put("clientContact", "王经理");
projectData.put("clientPhone", "13900139000");
projectData.put("scale", "50MW");
projectData.put("turbineModel", "GW-1500");
projectData.put("projectName", "华北风电场年度检测");
projectData.put("startDate", "2025-07-01");
projectData.put("endDate", "2025-07-31");
JSONObject shigongData = new JSONObject();
shigongData.put("startTime", "2025-07-21 08:00");
shigongData.put("endTime", "2025-07-21 17:00");
shigongData.put("weatherCode", "");
shigongData.put("windSpeed", "5.2m/s");
shigongData.put("imageCount", "120");
shigongData.put("temperature", "28℃");
jsonData1.put("jizu_data", jizuData);
jsonData1.put("project_data", projectData);
jsonData1.put("shigong_data", shigongData);
jsonData1.put("partManufacturer", "金风科技");
// Y叶片数据
String imagePath = "storage/emulated/0/DCIM/Camera/IMG_20250721_173921.jpg";
JSONObject y1 = new JSONObject();
y1.put("Code", "1");
JSONObject y1List = new JSONObject();
y1List.put("叶片前缘", imagePath);
y1List.put("叶片后缘", imagePath);
y1List.put("叶片表面", imagePath);
y1.put("list_dict", y1List);
JSONObject y2 = new JSONObject();
y2.put("Code", "2");
JSONObject y2List = new JSONObject();
y2List.put("叶片前缘", imagePath);
y2List.put("叶片后缘", imagePath);
y2List.put("叶片表面", imagePath);
y2.put("list_dict", y2List);
JSONObject y3 = new JSONObject();
y3.put("Code", "3");
JSONObject y3List = new JSONObject();
y3List.put("叶片前缘", imagePath);
y3List.put("叶片后缘", imagePath);
y3List.put("叶片表面", imagePath);
y3.put("list_dict", y3List);
jsonData1.put("Y1", y1);
jsonData1.put("Y2", y2);
jsonData1.put("Y3", y3);
return jsonData1;
}
private JSONObject createMockJsonData2() throws JSONException {
JSONObject jsonData2 = new JSONObject();
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd_HH-mm-ss", Locale.getDefault());
String templatePath = new File(context.getFilesDir(), TEMPLATE_DIR).getAbsolutePath();
String outputPath = context.getExternalFilesDir(null).getAbsolutePath();
jsonData2.put("shengcheng_dir", outputPath);
jsonData2.put("muban_dir", templatePath);
jsonData2.put("if_waibu", false);
jsonData2.put("if_neibu", true);
jsonData2.put("if_fanglei", true);
jsonData2.put("userName", "admin");
jsonData2.put("baogaoCheck", "未审核");
jsonData2.put("key_words", "缺,损,裂,脱,污");
jsonData2.put("shigong_fangan", "标准检测方案");
jsonData2.put("jiancha_renyuan", "张三,李四");
jsonData2.put("baogao_zongjie", "本次检测共发现3处缺陷需要及时处理");
jsonData2.put("total_pic_num", "120");
return jsonData2;
}
}

View File

@ -2,6 +2,7 @@ package com.example.myapplication.api;
import com.example.myapplication.LoginActivity;
import com.example.myapplication.model.ApiResponse;
import com.example.myapplication.model.ChangePasswordRequest;
import com.example.myapplication.model.LoginEntity;
import com.example.myapplication.model.LoginRequest;
@ -16,6 +17,6 @@ public interface AuthApi {
@PUT("/auth/modify-password")
Call<ApiResponse<Void>> modifyPassword(
@Body LoginActivity.ChangePasswordRequest request
@Body ChangePasswordRequest request
);
}

View File

@ -7,7 +7,6 @@ import java.util.List;
import retrofit2.Call;
import retrofit2.http.GET;
import retrofit2.http.Headers;
// ApiService.java
@ -15,5 +14,5 @@ import retrofit2.http.Headers;
public interface ProjectApi {
@GET("project/list")
Call<ApiResponse<List<Project>>> getProjectList();
Call<ApiResponse<List<Project>>> getMyProjectList();
}

View File

@ -7,7 +7,7 @@ import androidx.room.PrimaryKey;
public class AudioEntity {
@PrimaryKey(autoGenerate = true)
private int id;
public boolean isupdated;
private String fileName;
public String AudioPath;
@ -27,10 +27,11 @@ public class AudioEntity {
private long fileSize;
public String time;
public AudioEntity(String audioPath, String imagePath,String time) {
public AudioEntity(String audioPath, String imagePath,String time,boolean isupdated) {
this.AudioPath = audioPath;
this.ImagePath = imagePath;
this.time=time;
this.isupdated=isupdated;
}
public int getId() {

View File

@ -0,0 +1,38 @@
package com.example.myapplication.model;
public class ChangePasswordRequest {
private String account;
private String newPassword;
private String oldPassword;
public ChangePasswordRequest(String account, String newPassword, String oldPassword) {
this.account = account;
this.newPassword = newPassword;
this.oldPassword = oldPassword;
}
// getters and setters
public String getAccount() {
return account;
}
public void setAccount(String account) {
this.account = account;
}
public String getNewPassword() {
return newPassword;
}
public void setNewPassword(String newPassword) {
this.newPassword = newPassword;
}
public String getOldPassword() {
return oldPassword;
}
public void setOldPassword(String oldPassword) {
this.oldPassword = oldPassword;
}
}

View File

@ -15,17 +15,17 @@ import androidx.room.PrimaryKey;
public double altitude;
public String user;
public String audioPath;
public int blade;
public String unit;
public String unitName;
public String turbineId;
public String turbineName;
public String partId;
public String partName;
public String project;
public boolean b;
public String temperature="0"; // 温度
public String humidity="0"; // 湿度
public String weather="0"; // 天气状况
public String imageSource="-1";
public ImageEntity(String path, long time, double latitude, double longitude, double altitude, String user, String audioPath, String project, String unit, int blade, boolean b,String unitName,String temperature,String humidity,String weather,String imageSource) {
public ImageEntity(String path, long time, double latitude, double longitude, double altitude, String user, String audioPath, String project, String partId, String partName, String turbineId, String turbineName, boolean b, String temperature, String humidity, String weather, String imageSource) {
this.path = path;
this.time = time;
@ -35,13 +35,14 @@ import androidx.room.PrimaryKey;
this.user = user;
this.audioPath = audioPath;
this.project=project;
this.unit=unit;
this.blade=blade;
this.partId = partId;
this.partName = partName;
this.turbineId=turbineId;
this.turbineName=turbineName;
this.b=b;
this.weather=weather;
this.temperature=temperature;
this.humidity=humidity;
this.unitName=unitName;
this.imageSource=imageSource;
}

View File

@ -1,169 +0,0 @@
package com.example.myapplication.model;
import android.graphics.Bitmap;
public class ImageInfo {
private String path; // 图片路径
private long time; // 时间戳单位
private Bitmap bitmap; // 图片缩略图
private double latitude;
private double longitude;
private double altitude;
private String user;
public boolean isB() {
return b;
}
public void setB(boolean b) {
this.b = b;
}
public ImageInfo(String path, long time, Bitmap bitmap, double latitude, double longitude, double altitude, String user, String audioPath, boolean b, Bitmap thumbnail, String unit, String project, int blade) {
this.path = path;
this.time = time;
this.bitmap = bitmap;
this.latitude = latitude;
this.longitude = longitude;
this.altitude = altitude;
this.user = user;
this.audioPath = audioPath;
this.b = b;
this.thumbnail = thumbnail;
this.unit = unit;
this.project = project;
this.blade = blade;
}
boolean b;
public Bitmap getThumbnail() {
return thumbnail;
}
public void setThumbnail(Bitmap thumbnail) {
this.thumbnail = thumbnail;
}
public String getProject() {
return project;
}
public void setProject(String project) {
this.project = project;
}
public String getUnit() {
return unit;
}
public void setUnit(String unit) {
this.unit = unit;
}
public int getBlade() {
return blade;
}
public void setBlade(int blade) {
this.blade = blade;
}
private String audioPath;
public Bitmap thumbnail;
private String project;
private String unit;
private int blade;
public void setPath(String path) {
this.path = path;
}
public void setTime(long time) {
this.time = time;
}
public void setBitmap(Bitmap bitmap) {
this.bitmap = bitmap;
}
public void setLatitude(double latitude) {
this.latitude = latitude;
}
public void setLongitude(double longitude) {
this.longitude = longitude;
}
public void setAltitude(double altitude) {
this.altitude = altitude;
}
public void setUser(String user) {
this.user = user;
}
public String getAudioPath() {
return audioPath;
}
public void setAudioPath(String audioPath) {
this.audioPath = audioPath;
}
// 修改构造函数
public ImageInfo(String path, long time, Bitmap thumbnail, double latitude, double longitude, double altitude, String user, String audioPath) {
this.path = path;
this.time = time;
this.bitmap = bitmap;
this.latitude = latitude;
this.longitude = longitude;
this.altitude=altitude;
this.user=user;
this.audioPath=audioPath;
}
public ImageInfo(String path, long time, Bitmap bitmap, double latitude, double longitude, double altitude, String user, String audioPath, Bitmap thumbnail, String project, String unit, int blade) {
this.path = path;
this.time = time;
this.bitmap = bitmap;
this.latitude = latitude;
this.longitude = longitude;
this.altitude = altitude;
this.user = user;
this.audioPath = audioPath;
this.thumbnail = thumbnail;
this.project = project;
this.unit = unit;
this.blade = blade;
}
public double getLongitude() {
return longitude;
}
public double getLatitude() {
return latitude;
}
public String getUser() {
return user;
}
public double getAltitude() {
return altitude;
}
public ImageInfo(String path, long time, Bitmap bitmap) {
this.path = path;
this.time = time;
this.bitmap = bitmap;
}
// Getter 方法
public String getPath() { return path; }
public long getTime() { return time; }
public Bitmap getBitmap() { return bitmap; }
}

View File

@ -1,5 +1,6 @@
package com.example.myapplication.model;
public class PartResponse {
private String partCode;
private String partId;

View File

@ -0,0 +1,231 @@
package com.example.myapplication.model;
import android.location.Location;
import java.util.ArrayList;
import java.util.List;
// SharedDataManager.java
public class SharedDataManager {
private static SharedDataManager instance;
private String imageId ="-1";
private String token="-1";
private String user="-1";
private String audioPath="-1";
private String projectId="-1";
private String turbineId ="-1";
private String turbineName ="-1";
private String partId ="-1";
private String partName ="-1";
private String chooseImageSource="-1";
private boolean isRecording=false;
private boolean isUploading=false; // 新增上传状态字段
private Location location = null; // 新增位置字段
private final List<DataChangeListener> listeners = new ArrayList<>();
private String othermsg;
public interface DataChangeListener {
void onImageIdChanged(String newId);
void onTokenChanged(String newToken);
void onUserChanged(String newUser);
void onAudioPathChanged(String newAudioPath);
void onProjectIdChanged(String newProjectId);
void onPartIdChanged(String newPartId);
void onPartNameChanged(String newPartName);
void onTurbineIdChanged(String newTurbineId);
void onTurbineNameChanged(String newTurbineName);
void onChooseImageSourceChanged(String newChooseImageSource);
void onIsRecordingChanged(boolean newisRecording);
void onUploadingStatusChanged(boolean isUploading); // 新增上传状态变化回调
void onLocationChanged(Location newLocation); // 新增位置变化回调
void onOtherMsgChanged(String msg); // 新增位置变化回调
}
private SharedDataManager() {}
public static synchronized SharedDataManager getInstance() {
if (instance == null) {
instance = new SharedDataManager();
}
return instance;
}
public boolean getIsRecording(){return isRecording;}
public String getImageId() {return imageId;}
public String getUser() { return user; }
public String getAudioPath() { return audioPath; }
public String getProjectId() { return projectId; }
public String getPartId() { return partId; }
public String getChooseImageSource() { return chooseImageSource; }
public String getToken() {return token;}
public boolean getisUploading() {
return isUploading;
}
public Location getLocation() {
return location;
}
public String getOthermsg()
{
return othermsg;
}
public String getTurbineId() {
return turbineId;
}
public void setOthermsg(String othermsg)
{
this.othermsg=othermsg;
notifyOtherMsgChanged(othermsg);
}
public void setLocation(Location location) {
this.location = location;
notifyLocationChanged(location);
}
public void setUploading(boolean uploading) {
if (this.isUploading != uploading) {
this.isUploading = uploading;
notifyUploadingStatusChanged(uploading);
}
}
public void setImageId(String imageId) {
this.imageId = imageId;
notifyImageIdChanged(imageId);
}
public void setUser(String user) {
this.user = user;
notifyUserChanged(user);
}
public void setAudioPath(String audioPath) {
this.audioPath = audioPath;
notifyAudioPathChanged(audioPath);
}
public void setProjectId(String projectId) {
this.projectId = projectId;
notifyProjectIdChanged(projectId);
}
public void setPartId(String partId) {
this.partId = partId;
notifyPartIdChanged(partId);
}
public void setPartName(String partName) {
this.partName = partName;
notifyPartNameChanged(partName);
}
public void setChooseImageSource(String chooseImageSource) {
this.chooseImageSource = chooseImageSource;
notifyChooseImageSourceChanged(chooseImageSource);
}
public void setToken(String token) {
this.token = token;
notifyTokenChanged(token);
}
public void setIsRecording(boolean isRecording) {
this.isRecording =isRecording ;
notifyIsRecordingChanged(isRecording);
}
public void setTurbineId(String turbineId) {
this.turbineId = turbineId;
notifyTurbineIdChanged(turbineId);
}
public void setTurbineName(String turbineName) {
this.turbineName = turbineName;
notifyTurbineNameChanged(turbineName);
}
public void addListener(DataChangeListener listener) {
listeners.add(listener);
}
public void removeListener(DataChangeListener listener) {
listeners.remove(listener);
}
private void notifyImageIdChanged(String newPath) {
for (DataChangeListener listener : listeners) {
listener.onImageIdChanged(newPath);
}
}
private void notifyTokenChanged(String newToken) {
for (DataChangeListener listener : listeners) {
listener.onTokenChanged(newToken);
}
}
private void notifyUserChanged(String newUser) {
for (DataChangeListener listener : listeners) {
listener.onUserChanged(newUser);
}
}
private void notifyAudioPathChanged(String newAudioPath) {
for (DataChangeListener listener : listeners) {
listener.onAudioPathChanged(newAudioPath);
}
}
private void notifyProjectIdChanged(String newProjectId) {
for (DataChangeListener listener : listeners) {
listener.onProjectIdChanged(newProjectId);
}
}
private void notifyPartIdChanged(String newPartId) {
for (DataChangeListener listener : listeners) {
listener.onPartIdChanged(newPartId);
}
}
private void notifyPartNameChanged(String newPartName) {
for (DataChangeListener listener : listeners) {
listener.onPartNameChanged(newPartName);
}
}
private void notifyTurbineIdChanged(String newTurbineId) {
for (DataChangeListener listener : listeners) {
listener.onTurbineIdChanged(newTurbineId);
}
}
private void notifyTurbineNameChanged(String newTurbineName) {
for (DataChangeListener listener : listeners) {
listener.onTurbineNameChanged(newTurbineName);
}
}
private void notifyChooseImageSourceChanged(String newChooseImageSource) {
for (DataChangeListener listener : listeners) {
listener.onChooseImageSourceChanged(newChooseImageSource);
}
}
private void notifyIsRecordingChanged(boolean newIsRecording) {
for (DataChangeListener listener : listeners) {
listener.onIsRecordingChanged(newIsRecording);
}
}
private void notifyUploadingStatusChanged(boolean isUploading) {
for (DataChangeListener listener : listeners) {
listener.onUploadingStatusChanged(isUploading);
}
}
private void notifyLocationChanged(Location newLocation) {
for (DataChangeListener listener : listeners) {
listener.onLocationChanged(newLocation);
}
}
private void notifyOtherMsgChanged(String newothermsg) {
for (DataChangeListener listener : listeners) {
listener.onOtherMsgChanged(newothermsg);
}
}
}

View File

@ -12,8 +12,6 @@ public class Turbine {
public String turbineName;
public String projectId;
@Override
public String toString() {
return
@ -22,4 +20,29 @@ public class Turbine {
}
// 构造方法getter/setter省略
@NonNull
public String getTurbineId() {
return turbineId;
}
public void setTurbineId(@NonNull String turbineId) {
this.turbineId = turbineId;
}
public String getTurbineName() {
return turbineName;
}
public void setTurbineName(String turbineName) {
this.turbineName = turbineName;
}
public String getProjectId() {
return projectId;
}
public void setProjectId(String projectId) {
this.projectId = projectId;
}
}

View File

@ -0,0 +1,373 @@
# 文档处理工具
import json
from .tools.document_tools import (
create_document, add_documents,add_table_and_replace,
add_table_to_document,process_images_table,
)
# 内容处理工具
from .tools.content_tools import (
add_picture,split_table_by_row_content,
search_and_replace,add_picture_to_table
)
from .tools.get_pictures import (
get_dict,
)
from .tools.dataproccess import (
caculate_work_days,
get_year_month,merge_info,
)
import asyncio
from .tools.defines import *
import os, re, datetime
async def generate_report(base_info, baogao_info):
#获取模板编号、模板名称
num_to_chinese = {1 : '', 2 : '', 3 : '', 4 : '', 5 : '', 6 : '', 7 : '', 8 : '', 9 : '', 10 : '', 11 : '十一', 12 : '十二'}
cover_encode = "encode"
cover_project = "project"
baogao_name1 = "baogaoname1"
baogao_name2 = "baogaoname2"
company_name_yi = "company_name_yi"
cover_date = "time"
TITLE_OF_REPORT = "companyencode"
jiegou_xuhao = 'num'
try:
base_info = merge_info(base_info, DEFAULT_BASE_INFO)
jizu_data = base_info["jizu_data"]
project_data = base_info["project_data"]
shigong_data = base_info["shigong_data"]
fengchang_name = project_data['farmName']
Yi_company = project_data['inspectionUnit']
yi_fuzeren = project_data['inspectionContact']
yi_phone = project_data['inspectionPhone']
fengchang_location = project_data['farmAddress']
Jia_company = project_data['client']
jia_fuzeren = project_data['clientContact']
jia_phone = project_data['clientPhone']
jizu_num = project_data['scale']
jizu_xinghao = project_data['turbineModel']
project_name = project_data['projectName']
jizu_bianhao = jizu_data["turbineName"]
start_date = project_data['startDate']
end_date = project_data['endDate']
fengmian_dir = project_data["fengmian_dir"]
try:
gongqi = caculate_work_days(start_date, end_date)
except:
gongqi = "日期获取失败"
except Exception as e:
print(f"数据库的项目-机组基本信息获取失败:{e}")
return
try:
baogao_date = datetime.datetime.now().strftime("%Y年%m月%d%H:%M") #现在的时间
date_year_month = get_year_month(baogao_date)
#前端信息
baogao_info = merge_info(baogao_info, DEFAULT_BAOGAO_INFO)
key_words= re.compile('|'.join(map(re.escape, baogao_info['key_words'].split(','))))
shengcheng_dir = baogao_info['shengcheng_dir']
muban_dir = baogao_info['muban_dir']
if muban_dir == "" or shengcheng_dir == "":
print("未配置模板/生成路径,请检查配置")
return
if_waibu = baogao_info["if_waibu"]
if_neibu = baogao_info["if_neibu"]
if_fanglei = baogao_info["if_fanglei"]
total_pic_num = baogao_info["total_pic_num"]
jiancha_renyuan = baogao_info["jiancha_renyuan"]
baogao_zongjie = baogao_info["baogao_zongjie"]
try:
Jiancha_date = shigong_data["startTime"].replace("T", " ") #检查日期
except:
Jiancha_date = "日期获取失败"
image_count = shigong_data['imageCount'] #从施工方案获取的图片数量,待定!!!
temperature = shigong_data['temperature'] #温度
wind_speed = shigong_data['windSpeed'] #风速
weather = shigong_data["weatherCode"] #天气 不从此接口获取,待定!!!
Y1 = base_info["Y1"]
Y2 = base_info["Y2"]
Y3 = base_info["Y3"]
Y_Code = [Y1["Code"], Y2["Code"], Y3["Code"]] #Y1,Y2,Y3的编号
partManufacturer = base_info["partManufacturer"] #厂商
print(f"找到叶片号{Y_Code},厂商{partManufacturer}")
baogao_label = []
renyuan_peizhi = []
gongzuo_neirong = []
shigong_fangan = []
shebei_peizhi = []
beizhu = []
jiancha = []
neirong = []
#获取对应枚举字段
if if_waibu:
baogao_label.append("外观")
if baogao_info["shigong_fangan"] == "None":
print("未传入施工方案,使用已有枚举")
renyuan_peizhi.append(SHIGONG_FANGAN_ENUM.WAIBU.RENYUAN_PEIZHI)
gongzuo_neirong.append(SHIGONG_FANGAN_ENUM.WAIBU.GONGZUO_NEIRONG)
shebei_peizhi.append(SHIGONG_FANGAN_ENUM.WAIBU.SHEBEI_PEIZHI)
shigong_fangan.append(SHIGONG_FANGAN_ENUM.WAIBU.SHIGONG_FANGAN)
else:
pass #待添加如果从平台传入枚举,但目前没必要,如果这种枚举标准信息库有更好的优化则可以实现
jiancha.append("无人机近距离外观检查")
neirong.append(f"".join(Y_Code) + f"{len(Y_Code)}支叶片的前缘、后缘、迎风面、背风面。")
if if_neibu:
baogao_label.append("内部")
if baogao_info["shigong_fangan"] == "None":
renyuan_peizhi.append(SHIGONG_FANGAN_ENUM.NEIBU.RENYUAN_PEIZHI)
gongzuo_neirong.append(SHIGONG_FANGAN_ENUM.NEIBU.GONGZUO_NEIRONG)
shebei_peizhi.append(SHIGONG_FANGAN_ENUM.NEIBU.SHEBEI_PEIZHI)
shigong_fangan.append(SHIGONG_FANGAN_ENUM.NEIBU.SHIGONG_FANGAN)
else:
pass
jiancha.append("人工内部拍摄")
neirong.append(f"".join(Y_Code) + f"{len(Y_Code)}支叶片的内部导雷卡、腹板、透光、人孔盖版、叶根盖板...")
if if_fanglei:
baogao_label.append("防雷")
if baogao_info["shigong_fangan"] == "None":
renyuan_peizhi.append(SHIGONG_FANGAN_ENUM.FANGLEI.YEPIAN.RENYUAN_PEIZHI)
gongzuo_neirong.append(SHIGONG_FANGAN_ENUM.FANGLEI.YEPIAN.GONGZUO_NEIRONG)
shebei_peizhi.append(SHIGONG_FANGAN_ENUM.FANGLEI.YEPIAN.SHEBEI_PEIZHI)
shigong_fangan.append(SHIGONG_FANGAN_ENUM.FANGLEI.YEPIAN.SHIGONG_FANGAN)
else:
pass
jiancha.append("人工防雷")
neirong.append(f"轮毂至塔基导通、内部导线线阻、外部导线线阻...")
baogao_bianzhi = baogao_info["userName"]
baogao_shenghe = baogao_info["baogaoCheck"]
tatong_dir = baogao_info["tatong_dir"]
total_pic_dir = baogao_info["total_pic_dir"]
except Exception as e:
print(f"报告基本信息获取失败:{e}")
return
output_doc = None
head_num = 1
###封面创建###
cover_dirs = [os.path.join(muban_dir,"fengmian1.docx"),fengmian_dir,os.path.join(muban_dir,"fengmian2.docx")]
#输出目录
baogao_name = "叶片" + "".join(baogao_label) + "检查报告"
output_dir = os.path.normpath(f"{shengcheng_dir}/{project_name}项目{baogao_name}{jizu_bianhao}{baogao_date.split(' ')[0]}版.docx")
version = 1
while os.path.exists(output_dir):
if version != 1:
output_dir = output_dir.replace(f"{version - 1}",f"{version}")
else:
output_dir = output_dir.replace("",f"{version}")
version += 1
mianzhe_shengming = f"本报告仅涵盖{''.join(baogao_label)}检测内容"
#创建文档、添加封面
print(await create_document(output_dir))
print(add_documents(output_dir, cover_dirs[0]))
print(await add_picture(output_dir, cover_dirs[1], 6.41, 4))
print(add_documents(output_dir, cover_dirs[2]))
print("封面创建成功")
#更改文档信息
print(await search_and_replace(output_dir, TITLE_OF_REPORT, jizu_bianhao))
print(await search_and_replace(output_dir, baogao_name1, baogao_name))
print(await search_and_replace(output_dir, baogao_name2, baogao_name))
print(await search_and_replace(output_dir, company_name_yi, Yi_company))
print(await search_and_replace(output_dir, cover_project, fengchang_name))
print(await search_and_replace(output_dir, cover_encode, jizu_bianhao))
print(await search_and_replace(output_dir, cover_date, date_year_month))
print(await search_and_replace(output_dir, 'bianzhi', baogao_bianzhi))
print(await search_and_replace(output_dir, 'shenghe', baogao_shenghe))
print(await search_and_replace(output_dir, 'mianzhe_shengming', mianzhe_shengming))
total_table_num = 0
#项目概况表
print("开始添加项目概况表")
XIANG_MU_GAI_KUANG = os.path.join(muban_dir,"xiangmugaikuo.docx")
print(f"查找模板,找到模板:{XIANG_MU_GAI_KUANG}")
project_location = fengchang_location
company_name_jia = Jia_company
fuzeren = yi_fuzeren
phone_fuzeren = yi_phone
xiangmuguige = jizu_num
Yi_company = Yi_company
XIANGMU_GAIKUO = list(list("" for i in range(6)) for j in range(5))
XIANGMU_GAIKUO[0][1] = fengchang_name
#XIANGMU_GAIKUO[0][3]=XIANGMU_GAIKUO[0][4] = "盐城市滨海县"
XIANGMU_GAIKUO[0][4] = project_location
#XIANGMU_GAIKUO[1][1]=XIANGMU_GAIKUO[2,1]=XIANGMU_GAIKUO[3,1] = "国家电投集团滨海风力发电有限公司"
XIANGMU_GAIKUO[1][1] = company_name_jia
XIANGMU_GAIKUO[1][4] = Yi_company
XIANGMU_GAIKUO[2][1] = jia_fuzeren
XIANGMU_GAIKUO[2][4] = fuzeren
XIANGMU_GAIKUO[3][2] = jia_phone
XIANGMU_GAIKUO[3][5] = phone_fuzeren
XIANGMU_GAIKUO[4][1] = jizu_xinghao
XIANGMU_GAIKUO[4][3] = xiangmuguige
XIANGMU_GAIKUO[4][5] = gongqi
print("建立表结构完毕,开始插入模板")
#添加项目概况表
print(f"输出路径:{output_dir},模板路径:{XIANG_MU_GAI_KUANG},插入数据:{XIANGMU_GAIKUO}")
output_doc, message = await add_table_to_document(output_dir, XIANG_MU_GAI_KUANG,5,5,total_table_num,XIANGMU_GAIKUO)
print(message)
print("模板插入完毕,开始替换内容")
total_table_num += 1
print(await search_and_replace(output_dir, jiegou_xuhao, num_to_chinese[head_num]))
head_num += 1
#检查方案描述
FANGAN_JIANCHA_DIR = os.path.join(muban_dir,"checkmethod.docx")
list_to_replace = {
'renyuan_peizhi' : "\n".join(renyuan_peizhi),
'shebei_peizhi' : "\n".join(shebei_peizhi),
'shigong_fangan' : "\n".join(shigong_fangan),
'gongzuo_neirong' : "\n".join(gongzuo_neirong),
'beizhu' : beizhu,
'num' : num_to_chinese[head_num],
}
print(await add_table_and_replace(output_dir, FANGAN_JIANCHA_DIR, 0, list_to_replace))
print(split_table_by_row_content(output_dir, output_dir, total_table_num))
total_table_num += 1
head_num += 1
JIANCHA_XINGXI_DIR = os.path.join(muban_dir,"checkinfo.docx")
JIANCHA_XINGXI = list(list("" for i in range(4)) for j in range(9))
JIANCHA_XINGXI[0][1] = jiancha_renyuan
JIANCHA_XINGXI[1][1] = Jiancha_date.split(' ')[0]
JIANCHA_XINGXI[1][3] = jizu_bianhao
JIANCHA_XINGXI[2][1] = "风力发电机组" + baogao_name
JIANCHA_XINGXI[2][3] = "".join(jiancha)
JIANCHA_XINGXI[3][2] = partManufacturer
JIANCHA_XINGXI[4][1] = '叶片型号:' + jizu_xinghao
JIANCHA_XINGXI[5][1] = Y_Code[0] if len(Y_Code) > 0 else ""
JIANCHA_XINGXI[6][1] = Y_Code[1] if len(Y_Code) > 1 else ""
JIANCHA_XINGXI[7][1] = Y_Code[2] if len(Y_Code) > 2 else ""
JIANCHA_XINGXI[8][0] = "本次" + "".join(_ for _ in jiancha) + f"检查,采集叶片图片{total_pic_num}张,内容覆盖" + ";".join(_ for _ in neirong)
#新建检查信息表
output_doc, message = await add_table_to_document(output_dir, JIANCHA_XINGXI_DIR,9,4,total_table_num ,JIANCHA_XINGXI,False)
#添加塔筒图片
print(await add_picture_to_table(output_doc, output_dir, 4, 2, tatong_dir, total_table_num, width = 1.18))
#添加略缩图
print(await add_picture_to_table(output_doc, output_dir, 8, 0, total_pic_dir, total_table_num))
print(message)
print(await search_and_replace(output_dir, jiegou_xuhao, num_to_chinese[head_num]))
head_num += 1
total_table_num += 1
# 添加成果递交表
CHENGGUO_DIJIAO_DIR = os.path.join(muban_dir,"chengguo_sub.docx")
CHENGGUO_DIJIAO = list(list("" for i in range(4)) for j in range(6))
CHENGGUO_DIJIAO[0][1] = jiancha_renyuan
CHENGGUO_DIJIAO[1][1] = jia_fuzeren
try:
CHENGGUO_DIJIAO[2][1] = Jiancha_date.split(' ')[0]
except:
CHENGGUO_DIJIAO[2][1] = "日期获取失败"
CHENGGUO_DIJIAO[3][1] = jiancha_renyuan
CHENGGUO_DIJIAO[4][1] = baogao_bianzhi
CHENGGUO_DIJIAO[5][1] = baogao_shenghe
try:
CHENGGUO_DIJIAO[2][3] = Jiancha_date.split(' ')[1]
CHENGGUO_DIJIAO[3][3] = baogao_date.split(' ')[0]
CHENGGUO_DIJIAO[4][3] = baogao_date.split(' ')[0]
except:
CHENGGUO_DIJIAO[2][3] = "日期获取失败"
CHENGGUO_DIJIAO[3][3] = "日期获取失败"
CHENGGUO_DIJIAO[4][3] = "日期获取失败"
CHENGGUO_DIJIAO[5][3] = "未审核"
output_doc, message = await add_table_to_document(output_dir, CHENGGUO_DIJIAO_DIR,5,5,total_table_num,CHENGGUO_DIJIAO,True,0.04)
print(message)
print(await search_and_replace(output_dir, jiegou_xuhao, num_to_chinese[head_num]))
head_num += 1
total_table_num += 1
#主要部位图片展示表/检查内容表
#获取典型图信息
"""
需要获取
Y1Y2Y3叶片的图片数量字典{描述图片路径}
"""
try:
Y1_dict = get_dict(Y1)
Y2_dict = get_dict(Y2)
Y3_dict = get_dict(Y3)
picture_Y1_num = len(Y1_dict)
picture_Y2_num = len(Y2_dict)
picture_Y3_num = len(Y3_dict)
except Exception as e:
print(f"获取图失败:{e}")
print(f"图片、文字数量:{picture_Y1_num} {picture_Y2_num} {picture_Y3_num}")
JIANCHA_NEIRONG_TOTAL_NUM = picture_Y1_num+ picture_Y2_num + picture_Y3_num
col ,row = 3, 0
JIANCHA_NEIRONG_PICTURES_TABLE = os.path.join(muban_dir,"check2.docx")
JIANCHA_NEIRONG_Y1_DIR = os.path.join(muban_dir,"check_content.docx")
JIANCHA_NEIRONG_Y1 = list(list("" for _ in range(3)) for j in range(1))
JIANCHA_NEIRONG_Y1[0][0] = f"叶片1{Y_Code[0]}检查内容"
print(f"Y1标题内容{JIANCHA_NEIRONG_Y1}")
JIANCHA_NEIRONG_Y2_DIR = os.path.join(muban_dir,"check3.docx")
JIANCHA_NEIRONG_Y2 = list(list("" for _ in range(3)) for j in range(1))
JIANCHA_NEIRONG_Y2[0][0] = f"叶片2{Y_Code[1]}检查内容"
print(f"Y2标题内容{JIANCHA_NEIRONG_Y2}")
JIANCHA_NEIRONG_Y3_DIR = os.path.join(muban_dir,"check3.docx")
JIANCHA_NEIRONG_Y3 = list(list("" for _ in range(3)) for j in range(1))
JIANCHA_NEIRONG_Y3[0][0] = f"叶片3{Y_Code[2]}检查内容"
print(f"Y3标题内容{JIANCHA_NEIRONG_Y3}")
print(f"当前表格序号为 {total_table_num}")
print(key_words)
output_doc, message = await add_table_to_document(output_dir, JIANCHA_NEIRONG_Y1_DIR,1,3,total_table_num,JIANCHA_NEIRONG_Y1,True, 1)
print(message)
total_table_num += 1
total_table_num = await process_images_table(Y1_dict, output_dir, total_table_num, JIANCHA_NEIRONG_PICTURES_TABLE, key_words)
output_doc, message = await add_table_to_document(output_dir, JIANCHA_NEIRONG_Y2_DIR,1,3,total_table_num,JIANCHA_NEIRONG_Y2,True, 1)
print(message)
total_table_num += 1
total_table_num = await process_images_table(Y2_dict, output_dir, total_table_num, JIANCHA_NEIRONG_PICTURES_TABLE, key_words)
output_doc, message = await add_table_to_document(output_dir, JIANCHA_NEIRONG_Y3_DIR,1,3,total_table_num,JIANCHA_NEIRONG_Y3,True, 1)
print(message)
total_table_num += 1
total_table_num = await process_images_table(Y3_dict, output_dir, total_table_num, JIANCHA_NEIRONG_PICTURES_TABLE, key_words)
print(await search_and_replace(output_dir, jiegou_xuhao, num_to_chinese[head_num]))
head_num += 1
#总结
ZONG_JIE_DIR = os.path.join(muban_dir,"result.docx")
ZONG_JIE_BEFORE = "result"
ZONG_JIE = baogao_zongjie
print(add_documents(output_dir, ZONG_JIE_DIR))
print(await search_and_replace(output_dir, ZONG_JIE_BEFORE, ZONG_JIE))
print(await search_and_replace(output_dir, 'company_yi', Yi_company))
print(await search_and_replace(output_dir, 'baogao_date', baogao_date.split(' ')[0]))
print(await search_and_replace(output_dir, jiegou_xuhao, num_to_chinese[head_num]))
return output_dir
def main(json_data1,json_data2):
json_data1=json.loads(json_data1)
json_data2=json.loads(json_data2)
asyncio.run(generate_report(json_data1,json_data2))
print('文档生成完毕')

View File

@ -0,0 +1 @@
报告生成接入服务器的测试开发调用服务器的api测试

View File

@ -0,0 +1,138 @@
"""
Style-related functions for Word Document Server.
"""
from docx.shared import Pt
from docx.enum.style import WD_STYLE_TYPE
def ensure_heading_style(doc):
"""
Ensure Heading styles exist in the document.
Args:
doc: Document object
"""
for i in range(1, 10): # Create Heading 1 through Heading 9
style_name = f'Heading {i}'
try:
# Try to access the style to see if it exists
style = doc.styles[style_name]
except KeyError:
# Create the style if it doesn't exist
try:
from docx.oxml.ns import qn
style = doc.styles.add_style(style_name, WD_STYLE_TYPE.PARAGRAPH)
style.font.name = '宋体(中文正文)'
style.font.size = Pt(22) # 根据需要设置字体大小
style._element.rPr.rFonts.set(qn('w:eastAsia'), '宋体(中文正文)')
if i == 1:
style.font.size = Pt(16)
style.font.bold = True
elif i == 2:
style.font.size = Pt(14)
style.font.bold = True
else:
style.font.size = Pt(12)
style.font.bold = True
except Exception:
# If style creation fails, we'll just use default formatting
pass
def ensure_table_style(doc):
"""
Ensure Table Grid style exists in the document.
Args:
doc: Document object
"""
try:
# Try to access the style to see if it exists
style = doc.styles['Table Grid']
except KeyError:
# If style doesn't exist, we'll handle it at usage time
pass
def create_style(doc, style_name, style_type, base_style=None, font_properties=None, paragraph_properties=None):
"""
Create a new style in the document.
Args:
doc: Document object
style_name: Name for the new style
style_type: Type of style (WD_STYLE_TYPE)
base_style: Optional base style to inherit from
font_properties: Dictionary of font properties (bold, italic, size, name, color)
paragraph_properties: Dictionary of paragraph properties (alignment, spacing)
Returns:
The created style
"""
from docx.shared import Pt
try:
# Check if style already exists
style = doc.styles.get_by_id(style_name, WD_STYLE_TYPE.PARAGRAPH)
return style
except:
# Create new style
new_style = doc.styles.add_style(style_name, style_type)
# Set base style if specified
if base_style:
new_style.base_style = doc.styles[base_style]
# Set font properties
if font_properties:
font = new_style.font
if 'bold' in font_properties:
font.bold = font_properties['bold']
if 'italic' in font_properties:
font.italic = font_properties['italic']
if 'size' in font_properties:
font.size = Pt(font_properties['size'])
if 'name' in font_properties:
font.name = font_properties['name']
if 'color' in font_properties:
from docx.shared import RGBColor
# Define common RGB colors
color_map = {
'red': RGBColor(255, 0, 0),
'blue': RGBColor(0, 0, 255),
'green': RGBColor(0, 128, 0),
'yellow': RGBColor(255, 255, 0),
'black': RGBColor(0, 0, 0),
'gray': RGBColor(128, 128, 128),
'white': RGBColor(255, 255, 255),
'purple': RGBColor(128, 0, 128),
'orange': RGBColor(255, 165, 0)
}
color_value = font_properties['color']
try:
# Handle string color names
if isinstance(color_value, str) and color_value.lower() in color_map:
font.color.rgb = color_map[color_value.lower()]
# Handle RGBColor objects
elif hasattr(color_value, 'rgb'):
font.color.rgb = color_value
# Try to parse as RGB string
elif isinstance(color_value, str):
font.color.rgb = RGBColor.from_string(color_value)
# Use directly if it's already an RGB value
else:
font.color.rgb = color_value
except Exception as e:
# Fallback to black if all else fails
font.color.rgb = RGBColor(0, 0, 0)
# Set paragraph properties
if paragraph_properties:
if 'alignment' in paragraph_properties:
new_style.paragraph_format.alignment = paragraph_properties['alignment']
if 'spacing' in paragraph_properties:
new_style.paragraph_format.line_spacing = paragraph_properties['spacing']
return new_style

View File

@ -0,0 +1,283 @@
"""
Table-related operations for Word Document Server.
"""
from docx.oxml.shared import OxmlElement, qn
from docx.oxml.ns import nsdecls
from docx.oxml import parse_xml
def set_cell_border(cell, **kwargs):
"""
Set cell border properties.
Args:
cell: The cell to modify
**kwargs: Border properties (top, bottom, left, right, val, color)
"""
tc = cell._tc
tcPr = tc.get_or_add_tcPr()
# Create border elements
for key, value in kwargs.items():
if key in ['top', 'left', 'bottom', 'right']:
tag = 'w:{}'.format(key)
element = OxmlElement(tag)
element.set(qn('w:val'), kwargs.get('val', 'single'))
element.set(qn('w:sz'), kwargs.get('sz', '4'))
element.set(qn('w:space'), kwargs.get('space', '0'))
element.set(qn('w:color'), kwargs.get('color', 'auto'))
tcBorders = tcPr.first_child_found_in("w:tcBorders")
if tcBorders is None:
tcBorders = OxmlElement('w:tcBorders')
tcPr.append(tcBorders)
tcBorders.append(element)
def apply_table_style(table, has_header_row=False, border_style=None, shading=None):
"""
Apply formatting to a table.
Args:
table: The table to format
has_header_row: If True, formats the first row as a header
border_style: Style for borders ('none', 'single', 'double', 'thick')
shading: 2D list of cell background colors (by row and column)
Returns:
True if successful, False otherwise
"""
try:
# Format header row if requested
if has_header_row and table.rows:
header_row = table.rows[0]
for cell in header_row.cells:
for paragraph in cell.paragraphs:
if paragraph.runs:
for run in paragraph.runs:
run.bold = True
# Apply border style if specified
if border_style:
val_map = {
'none': 'nil',
'single': 'single',
'double': 'double',
'thick': 'thick'
}
val = val_map.get(border_style.lower(), 'single')
# Apply to all cells
for row in table.rows:
for cell in row.cells:
set_cell_border(
cell,
top=True,
bottom=True,
left=True,
right=True,
val=val,
color="000000"
)
# Apply cell shading if specified
if shading:
for i, row_colors in enumerate(shading):
if i >= len(table.rows):
break
for j, color in enumerate(row_colors):
if j >= len(table.rows[i].cells):
break
try:
# Apply shading to cell
cell = table.rows[i].cells[j]
shading_elm = parse_xml(f'<w:shd {nsdecls("w")} w:fill="{color}"/>')
cell._tc.get_or_add_tcPr().append(shading_elm)
except:
# Skip if color format is invalid
pass
return True
except Exception:
return False
def copy_table(source_table, target_doc, ifadjustheight=True, height = 1):
"""
Copy a table from one document to another.
Args:
source_table: The table to copy
target_doc: The document to copy the table to
Returns:
The new table in the target document
"""
# Create a new table with the same dimensions
new_table = target_doc.add_table(rows=len(source_table.rows), cols=len(source_table.columns))
# Try to apply the same style
try:
if source_table.style:
new_table.style = 'Table Grid'
except:
# Fall back to default grid style
try:
new_table.style = 'Table Grid'
except:
pass
from docx.enum.table import WD_TABLE_ALIGNMENT
from docx.enum.table import WD_ALIGN_VERTICAL
from docx.shared import Pt, Inches, Cm, RGBColor
# Copy cell contents
for i, row in enumerate(source_table.rows):
for j, cell in enumerate(row.cells):
for paragraph in cell.paragraphs:
average_char_width_in_points = 6
if paragraph.text:
new_table.cell(i,j).text = paragraph.text
new_table.cell(i,j).paragraphs[0].runs[0].font.name = "Times New Roman" #设置英文字体
new_table.cell(i,j).paragraphs[0].runs[0].font.size = Pt(10.5) # 字体大小
new_table.cell(i,j).paragraphs[0].runs[0]._element.rPr.rFonts.set(qn('w:eastAsia'), '仿宋') #设置中文字体
new_table.cell(i,j).paragraphs[0].paragraph_format.alignment = WD_TABLE_ALIGNMENT.CENTER
new_table.cell(i,j).vertical_alignment = WD_ALIGN_VERTICAL.CENTER
"""
待添加如何让表格自适应大小autofit目前不知为何没有作用
"""
if ifadjustheight:
new_table.rows[i].height = Cm(height)
try:
new_table = merge_tables(new_table)
except Exception as e:
print(f"合并表格失败:{e}")
from docx.shared import Inches
return new_table
from collections import deque
def merge_tables(table):
"""BFS遍历将相邻且相同单元格合并
Args:
table: 表格docx库的Table类型
Returns:
合并后的表格
"""
if not table or len(table.rows) == 0:
return table
rows = len(table.rows)
cols = len(table.columns)
# 创建访问标记矩阵
visited = [[False for _ in range(cols)] for _ in range(rows)]
# 定义四个方向的移动:上、右、下、左
directions = [(0, 0), (0, 1), (1, 0), (0, 0)]
for i in range(rows):
for j in range(cols):
if not visited[i][j]:
current_cell = table.cell(i, j)
current_text = current_cell.text.strip()
if not current_text: # 跳过空单元格
visited[i][j] = True
continue
# BFS队列
queue = deque()
queue.append((i, j))
visited[i][j] = True
# 记录需要合并的单元格
cells_to_merge = []
while queue:
x, y = queue.popleft()
cells_to_merge.append((x, y))
for dx, dy in directions:
nx, ny = x + dx, y + dy
# 检查边界和访问状态
if 0 <= nx < rows and 0 <= ny < cols and not visited[nx][ny]:
neighbor_cell = table.cell(nx, ny)
neighbor_text = neighbor_cell.text.strip()
if neighbor_text == current_text:
visited[nx][ny] = True
queue.append((nx, ny))
# 如果有需要合并的单元格
if len(cells_to_merge) > 1:
# 按行和列排序,确保左上角是第一个单元格
cells_to_merge.sort()
min_row, min_col = cells_to_merge[0]
max_row, max_col = cells_to_merge[-1]
# 清空所有待合并单元格(包括换行符)
for x, y in cells_to_merge[1:]:
cell = table.cell(x, y)
# 删除所有段落(彻底清空)
for paragraph in list(cell.paragraphs):
p = paragraph._element
p.getparent().remove(p)
# 可选:添加一个空段落防止格式问题
cell.add_paragraph()
# 执行合并
if max_row > min_row or max_col > min_col:
table.cell(min_row, min_col).merge(table.cell(max_row, max_col))
return table
def fill_tables(Y_table_list, row, col, Y_Table_num, Y):
"""根据前端返回json块填写表格list并实时跟进已填写表格数量
目前只支持固定的缺陷图的填写
Args:
Y_table_list (list): 前端返回的json块
row (int): 表格行数
col (int): 表格列数
Y_Table_num: json块中有几个表格
Xu_Hao: 是第几个json块
Y: 其他参数
Return:
Y1_TABLES: 三维表和对应元素
table_index_to_images: 字典表索引到图片路径列表的映射
Xu_Hao到达第几个表了
"""
table_index_to_images = {}
Y_TABLES = [[["" for _ in range(row)] for _ in range(col)] for _ in range(Y_Table_num)]
# 处理前端返回数据
for l, table_dict in enumerate(Y_table_list):
if table_dict:
Y_TABLES[l][1][0] = Y
Y_TABLES[l][1][1] = table_dict["QueXianLeiXing"]
Y_TABLES[l][1][2] = table_dict["QueXianWeiZhi"]
Y_TABLES[l][1][3] = table_dict["QueXianChiCun"]
Y_TABLES[l][3][0] = table_dict["WeiZongDengJi"]
Y_TABLES[l][3][1] = table_dict["visibility"]
Y_TABLES[l][3][2] = table_dict["urgency"]
Y_TABLES[l][3][3] = table_dict["repair_suggestion"]
# 获取图片路径
image_path = table_dict['Tupian_Dir']
if image_path:
# 确保路径是字符串形式
if isinstance(image_path, list):
table_index_to_images[l] = image_path.copy()
else:
table_index_to_images[l] = [str(image_path)]
return Y_TABLES, table_index_to_images

Binary file not shown.

After

Width:  |  Height:  |  Size: 38 KiB

View File

@ -0,0 +1,637 @@
"""
Content tools for Word Document Server.
These tools add various types of content to Word documents,
including headings, paragraphs, tables, images, and page breaks.
"""
import os
from typing import List, Optional, Dict, Any
from docx import Document
from docx.shared import Inches, Pt
from docx.oxml.shared import qn
from ..utils.file_utils import check_file_writeable, ensure_docx_extension
from ..utils.document_utils import find_and_replace_text
from ..core.styles import ensure_heading_style, ensure_table_style
def split_table_by_row_content(
doc_path: str,
output_path: str,
table_num: int = 0
) -> str:
"""
根据表格第二行第一列内容的行数对指定表格进行分行处理
并将每列内容按相同行数分割不足则重复
参数:
doc_path: 输入Word文档路径
output_path: 输出Word文档路径
table_num: 要处理的表格序号(从0开始)
"""
try:
from docx import Document
from docx.shared import Pt
from docx.oxml.shared import qn
from docx.enum.table import WD_TABLE_ALIGNMENT, WD_ALIGN_VERTICAL
# 打开文档
doc = Document(doc_path)
# 检查表格是否存在
if len(doc.tables) <= table_num:
return f"文档中不存在第{table_num+1}个表格"
# 获取指定表格
table = doc.tables[table_num]
# 获取表格行数和列数
row_count = len(table.rows)
col_count = len(table.columns)
# 如果表格行数小于2无法处理
if row_count < 2:
doc.save(output_path)
return "表格行数少于2行无法按照要求分行"
# 获取第二行第一列的文本内容
second_row_first_cell = table.cell(1, 0)
second_row_text = second_row_first_cell.text
# 计算第二行第一列文本的行数(按换行符分割)
lines_in_second_row = len(second_row_text.split('\n'))
# 如果行数为0设置为1至少分为1部分
split_count = max(1, lines_in_second_row)
print(f'原表格行数:{row_count},第二行第一列内容行数:{split_count},需要分割为:{split_count}部分')
# 创建新表格来替代原表格(分割后的表格)
# 新表格的行数 = 标题行(1) + 原数据行数 × 分割部分数
new_table = doc.add_table(rows=1 + (row_count-1)*split_count, cols=col_count)
# 设置表格样式
new_table.style = table.style
new_table.autofit = True
# 1. 处理标题行(第一行)保持不变
for col_idx in range(col_count):
orig_cell = table.cell(0, col_idx)
new_cell = new_table.cell(0, col_idx)
# 复制内容并设置格式
new_cell.text = orig_cell.text
if orig_cell.paragraphs:
# 设置格式
new_cell.paragraphs[0].runs[0].font.name = "Times New Roman"
new_cell.paragraphs[0].runs[0].font.size = Pt(10.5)
new_cell.paragraphs[0].runs[0]._element.rPr.rFonts.set(qn('w:eastAsia'), '仿宋')
new_cell.paragraphs[0].paragraph_format.alignment = WD_TABLE_ALIGNMENT.CENTER
new_cell.vertical_alignment = WD_ALIGN_VERTICAL.CENTER
new_cell.width = orig_cell.width
# 2. 处理数据行(从第二行开始)
for orig_row_idx in range(1, row_count): # 遍历原表格的每一行数据
for col_idx in range(col_count): # 遍历每一列
orig_cell = table.cell(orig_row_idx, col_idx)
cell_text = orig_cell.text
# 分割当前单元格内容
cell_lines = cell_text.split('\n')
cell_line_count = len(cell_lines)
# 如果内容行数不足分割数,则重复最后一行
if cell_line_count < split_count:
cell_lines += [cell_lines[-1]] * (split_count - cell_line_count)
# 在新表格中对应的位置写入分割后的内容
for part_idx in range(split_count):
# 计算新表格中的行位置
new_row_idx = 1 + (orig_row_idx-1)*split_count + part_idx
# 获取新单元格
new_cell = new_table.cell(new_row_idx, col_idx)
# 写入分割后的内容
line_text = cell_lines[part_idx] if part_idx < len(cell_lines) else cell_lines[-1]
new_cell.text = line_text
# 设置格式
if new_cell.paragraphs:
new_cell.paragraphs[0].runs[0].font.name = "Times New Roman"
new_cell.paragraphs[0].runs[0].font.size = Pt(10.5)
new_cell.paragraphs[0].runs[0]._element.rPr.rFonts.set(qn('w:eastAsia'), '仿宋')
new_cell.paragraphs[0].paragraph_format.alignment = WD_TABLE_ALIGNMENT.CENTER
new_cell.vertical_alignment = WD_ALIGN_VERTICAL.CENTER
# 复制单元格宽度
new_cell.width = orig_cell.width
# 删除原表格
table._element.getparent().remove(table._element)
# 保存文档
doc.save(output_path)
return f"{table_num+1}个表格已成功分行处理"
except Exception as e:
return f"处理表格时出错: {str(e)}"
async def add_heading(filename: str, text: str, level: int = 1) -> str:
"""对文档增加标题
Args:
filename: 目标文档路径
text: 标题文本
level: 标题级别1为最高级
"""
filename = ensure_docx_extension(filename)
# Ensure level is converted to integer
try:
level = int(level)
except (ValueError, TypeError):
return "Invalid parameter: level must be an integer between 1 and 9"
# Validate level range
if level < 1 or level > 9:
return f"Invalid heading level: {level}. Level must be between 1 and 9."
if not os.path.exists(filename):
return f"Document {filename} does not exist"
# Check if file is writeable
is_writeable, error_message = check_file_writeable(filename)
if not is_writeable:
# Suggest creating a copy
return f"Cannot modify document: {error_message}. Consider creating a copy first or creating a new document."
try:
doc = Document(filename)
# Ensure heading styles exist
ensure_heading_style(doc)
# Try to add heading with style
try:
heading = doc.add_heading(text, level=level)
doc.save(filename)
return f"Heading '{text}' (level {level}) added to {filename}"
except Exception as style_error:
# If style-based approach fails, use direct formatting
paragraph = doc.add_paragraph(text)
paragraph.style = doc.styles['Normal']
run = paragraph.runs[0]
run.bold = True
rPr = run.element.get_or_add_rPr()
rFonts = rPr.get_or_add_rFonts()
from docx.oxml.shared import qn
rFonts.set(qn('w:eastAsia'), '宋体(中文正文)')
# Adjust size based on heading level
if level == 1:
run.font.size = Pt(12)
elif level == 2:
run.font.size = Pt(14)
else:
run.font.size = Pt(12)
doc.save(filename)
return f"Heading '{text}' added to {filename} with direct formatting (style not available)"
except Exception as e:
return f"Failed to add heading: {str(e)}"
async def add_paragraph(filename: str, text: str, style: Optional[str] = None) -> str:
"""对文档添加一个段落(一行)
Args:
filename: 目标文档路径
text: 段落内容
style: 段落样式可选
"""
filename = ensure_docx_extension(filename)
if not os.path.exists(filename):
return f"Document {filename} does not exist"
# Check if file is writeable
is_writeable, error_message = check_file_writeable(filename)
if not is_writeable:
# Suggest creating a copy
return f"Cannot modify document: {error_message}. Consider creating a copy first or creating a new document."
try:
doc = Document(filename)
paragraph = doc.add_paragraph(text)
if style:
try:
paragraph.style = style
except KeyError:
# Style doesn't exist, use normal and report it
paragraph.style = doc.styles['Normal']
# Copy run formatting
# for i, run in enumerate(paragraph.runs):
# if i < len(paragraph.runs):
# new_run = paragraph.runs[i]
# # Copy basic formatting
# new_run.bold = run.bold
# new_run.italic = run.italic
# new_run.underline = run.underline
# #添加同时合并字体2025427
# new_run.font.name = run.font.name
# rPr = new_run.element.get_or_add_rPr()
# rFonts = rPr.get_or_add_rFonts()
# # 检查 run.font.name 是否为 None
# if run.font.name is None:
# # 设置默认的中文字体名称
# run.font.name = '宋体 (中文正文)' # 或者使用其他你喜欢的中文字体
# rFonts.set(qn('w:eastAsia'), run.font.name)
# new_run.font.color.rgb = run.font.color.rgb
# # Font size if specified
# if run.font.size:
# new_run.font.size = run.font.size
doc.save(filename)
return f"Style '{style}' not found, paragraph added with default style to {filename}"
doc.save(filename)
return f"Paragraph added to {filename}"
except Exception as e:
return f"Failed to add paragraph: {str(e)}"
async def add_table(filename: str, rows: int, cols: int, data: Optional[List[List[str]]] = None) -> str:
"""对文档添加一个表格
Args:
filename: 目标文档路径
rows: 表格行数
cols: 表格列数
data: 二维数组列表每一项为单元格内容默认为空
"""
filename = ensure_docx_extension(filename)
if not os.path.exists(filename):
return f"Document {filename} does not exist"
# Check if file is writeable
is_writeable, error_message = check_file_writeable(filename)
if not is_writeable:
# Suggest creating a copy
return f"Cannot modify document: {error_message}. Consider creating a copy first or creating a new document."
try:
doc = Document(filename)
table = doc.add_table(rows=rows, cols=cols)
# Try to set the table style
try:
table.style = 'Table Grid'
except KeyError:
# If style doesn't exist, add basic borders
pass
# Fill table with data if provided
if data:
for i, row_data in enumerate(data):
if i >= rows:
break
for j, cell_text in enumerate(row_data):
if j >= cols:
break
table.cell(i, j).text = str(cell_text)
doc.save(filename)
return f"Table ({rows}x{cols}) added to {filename}"
except Exception as e:
return f"Failed to add table: {str(e)}"
async def add_picture_to_table(target_doc: Document, target_filename: str, row: int, col: int, image_path: str,table_num: int = -1, width: Optional[float] = None) -> str:
"""向文档中对应表格添加图片
Args:
target_doc: 目标文档
target_filename: 目标文档保存路径
row: 表格行数
col: 表格列数
image_path: 图片路径
table_num: 表格序号默认为-1即最后一个表格
width: 图片宽度默认为None表示使用原始图片大小
"""
from PIL import Image
if not os.path.exists(image_path):
return f"Image file not found: {image_path}"
# Check image file size
try:
image_size = os.path.getsize(image_path) / 1024 # Size in KB
if image_size <= 0:
return f"Image file appears to be empty: {image_path} (0 KB)"
elif image_size > 9126:
# Create the output directory if it doesn't exist
output_dir = os.path.join(os.path.dirname(image_path), "压缩图片")
if not os.path.exists(output_dir):
os.makedirs(output_dir)
# Define the output path for the compressed image
image_name = os.path.basename(image_path)
output_path = os.path.join(output_dir, image_name)
# Compress the image
while image_size > 9126:
print(f"压缩图片:{image_path} ({image_size:.2f} KB) -> {output_path} (9126 KB)")
with Image.open(image_path) as img:
img.save(output_path, optimize=True, quality=85)
image_size = os.path.getsize(output_path) / 1024 # Size in KB
# Update the image path to the compressed image path
image_path = output_path
except Exception as size_error:
return f"Error checking image file: {str(size_error)}"
try:
table = target_doc.tables[table_num]
# Add the picture to the cell
cell = table.cell(row, col)
if len(cell.text) == 1: cell.text = ""
paragraph = cell.paragraphs[-1]
run = paragraph.add_run()
try:
if width:
run.add_picture(image_path, width=Inches(width))
else:
run.add_picture(image_path)
except Exception as e:
# 如果添加图片时出现问题尝试将图片转换为PNG格式
try:
print(f"正常添加失败,尝试转换图片后添加:{image_path}")
# 打开图片
img = Image.open(image_path)
# 转换为PNG格式
temp_image_path = os.path.splitext(image_path)[0] + '.png'
img.save(temp_image_path, 'PNG')
# 尝试添加转换后的图片
if width:
run.add_picture(temp_image_path, width=Inches(width))
else:
run.add_picture(temp_image_path)
# 添加完成后删除转换后的图片
os.remove(temp_image_path)
except Exception as e:
# 如果转换或添加转换后的图片时出现问题,返回错误信息
return f"调用add_picture函数出现问题: {str(e)}"
from docx.enum.table import WD_TABLE_ALIGNMENT
from docx.enum.table import WD_ALIGN_VERTICAL
cell.paragraphs[0].paragraph_format.alignment = WD_TABLE_ALIGNMENT.CENTER
cell.vertical_alignment = WD_ALIGN_VERTICAL.CENTER
target_doc.save(target_filename)
return f"Picture {image_path} added to table {table_num} cell ({row},{col})"
except Exception as e:
return f"Failed to add picture to table: {str(e)}"
async def add_picture(filename: str, image_path: str, width: Optional[float] = None, height: Optional[float] = None) -> str:
"""添加一个图片到文档中
Args:
filename: 文档路径
image_path: 图片路径
width: 图片大小
"""
filename = ensure_docx_extension(filename)
# Validate document existence
if not os.path.exists(filename):
return f"Document {filename} does not exist"
# Get absolute paths for better diagnostics
abs_filename = os.path.abspath(filename)
abs_image_path = os.path.abspath(image_path)
# Validate image existence with improved error message
if not os.path.exists(abs_image_path):
return f"Image file not found: {abs_image_path}"
# Check image file size
try:
image_size = os.path.getsize(abs_image_path) / 1024 # Size in KB
if image_size <= 0:
return f"Image file appears to be empty: {abs_image_path} (0 KB)"
except Exception as size_error:
return f"Error checking image file: {str(size_error)}"
# Check if file is writeable
is_writeable, error_message = check_file_writeable(abs_filename)
if not is_writeable:
return f"Cannot modify document: {error_message}. Consider creating a copy first or creating a new document."
try:
doc = Document(abs_filename)
# Additional diagnostic info
diagnostic = f"Attempting to add image ({abs_image_path}, {image_size:.2f} KB) to document ({abs_filename})"
try:
if width:
doc.add_picture(abs_image_path, width=Inches(width), height=Inches(height))
else:
doc.add_picture(abs_image_path)
doc.save(abs_filename)
return f"Picture {image_path} added to {filename}"
except Exception as inner_error:
# More detailed error for the specific operation
error_type = type(inner_error).__name__
error_msg = str(inner_error)
return f"Failed to add picture: {error_type} - {error_msg or 'No error details available'}\nDiagnostic info: {diagnostic}"
except Exception as outer_error:
# Fallback error handling
error_type = type(outer_error).__name__
error_msg = str(outer_error)
return f"Document processing error: {error_type} - {error_msg or 'No error details available'}"
async def add_page_break(filename: str) -> str:
"""增加分页符
Args:
filename: 目标文档
"""
filename = ensure_docx_extension(filename)
if not os.path.exists(filename):
return f"Document {filename} does not exist"
# Check if file is writeable
is_writeable, error_message = check_file_writeable(filename)
if not is_writeable:
return f"Cannot modify document: {error_message}. Consider creating a copy first."
try:
doc = Document(filename)
doc.add_page_break()
doc.save(filename)
return f"Page break added to {filename}."
except Exception as e:
return f"Failed to add page break: {str(e)}"
async def add_table_of_contents(filename: str, title: str = "Table of Contents", max_level: int = 3) -> str:
"""根据标题样式向Word文档添加目录。
参数:
filename: Word文档的路径
title: 可自行选择的一个标题
max_level: 要包含的最大标题级别1-9
"""
filename = ensure_docx_extension(filename)
if not os.path.exists(filename):
return f"Document {filename} does not exist"
# Check if file is writeable
is_writeable, error_message = check_file_writeable(filename)
if not is_writeable:
return f"Cannot modify document: {error_message}. Consider creating a copy first."
try:
# Ensure max_level is within valid range
max_level = max(1, min(max_level, 9))
doc = Document(filename)
# Collect headings and their positions
headings = []
for i, paragraph in enumerate(doc.paragraphs):
# Check if paragraph style is a heading
if paragraph.style and paragraph.style.name.startswith('Heading '):
try:
# Extract heading level from style name
level = int(paragraph.style.name.split(' ')[1])
if level <= max_level:
headings.append({
'level': level,
'text': paragraph.text,
'position': i
})
except (ValueError, IndexError):
# Skip if heading level can't be determined
pass
if not headings:
return f"No headings found in document {filename}. Table of contents not created."
# Create a new document with the TOC
toc_doc = Document()
# Add title
if title:
toc_doc.add_heading(title, level=1)
# Add TOC entries
for heading in headings:
# Indent based on level (using tab characters)
indent = ' ' * (heading['level'] - 1)
toc_doc.add_paragraph(f"{indent}{heading['text']}")
# Add page break
toc_doc.add_page_break()
# Get content from original document
for paragraph in doc.paragraphs:
p = toc_doc.add_paragraph(paragraph.text)
# Copy style if possible
try:
if paragraph.style:
p.style = paragraph.style.name
except:
pass
# Copy tables
for table in doc.tables:
# Create a new table with the same dimensions
new_table = toc_doc.add_table(rows=len(table.rows), cols=len(table.columns))
# Copy cell contents
for i, row in enumerate(table.rows):
for j, cell in enumerate(row.cells):
for paragraph in cell.paragraphs:
new_table.cell(i, j).text = paragraph.text
# Save the new document with TOC
toc_doc.save(filename)
return f"Table of contents with {len(headings)} entries added to {filename}"
except Exception as e:
return f"Failed to add table of contents: {str(e)}"
async def delete_paragraph(filename: str, paragraph_index: int) -> str:
"""通过行索引从文档中删除一段
Args:
filename: Path to the Word document
paragraph_index: 段落位置第几行
"""
filename = ensure_docx_extension(filename)
if not os.path.exists(filename):
return f"Document {filename} does not exist"
# Check if file is writeable
is_writeable, error_message = check_file_writeable(filename)
if not is_writeable:
return f"Cannot modify document: {error_message}. Consider creating a copy first."
try:
doc = Document(filename)
# Validate paragraph index
if paragraph_index < 0 or paragraph_index >= len(doc.paragraphs):
return f"Invalid paragraph index. Document has {len(doc.paragraphs)} paragraphs (0-{len(doc.paragraphs)-1})."
# Delete the paragraph (by removing its content and setting it empty)
# Note: python-docx doesn't support true paragraph deletion, this is a workaround
paragraph = doc.paragraphs[paragraph_index]
p = paragraph._p
p.getparent().remove(p)
doc.save(filename)
return f"Paragraph at index {paragraph_index} deleted successfully."
except Exception as e:
return f"Failed to delete paragraph: {str(e)}"
async def search_and_replace(filename: str, find_text: str, replace_text: str) -> str:
"""替换所有find_text为replace_text
Args:
filename: Path to the Word document
find_text: Text to search for
replace_text: Text to replace with
"""
filename = ensure_docx_extension(filename)
if not os.path.exists(filename):
return f"Document {filename} does not exist"
# Check if file is writeable
is_writeable, error_message = check_file_writeable(filename)
if not is_writeable:
return f"Cannot modify document: {error_message}. Consider creating a copy first."
try:
doc = Document(filename)
# Perform find and replace
count = find_and_replace_text(doc, find_text, replace_text)
if count > 0:
doc.save(filename)
return f"Replaced {count} occurrence(s) of '{find_text}' with '{replace_text}'."
else:
return f"No occurrences of '{find_text}' found."
except Exception as e:
return f"Failed to search and replace: {str(e)}"

View File

@ -0,0 +1,107 @@
from .content_tools import add_picture_to_table
from .document_tools import add_table_to_document,search_and_replace
from .get_pictures import resize_and_reduce_quality
def caculate_work_days(start_date : str, end_date : str) -> str:
"""根据起止日期计算工期
Args:
start_date (str): 格式: yyyy-mm-ddThh:mm:ss
end_date (str): 格式: yyyy-mm-ddThh:mm:ss
Returns:
str: 计算出来的总工期单位为天
"""
import datetime
if start_date == None or end_date == None:
return f"开始时间{start_date} ---- 结束时间{end_date} 日期格式错误"
start_date = datetime.datetime.strptime(start_date, '%Y-%m-%dT%H:%M:%S')
end_date = datetime.datetime.strptime(end_date, '%Y-%m-%dT%H:%M:%S')
return (end_date - start_date).days
async def add_dynamic_table(output_doc, output_dir, table_num, TABLES, JIANCHA_XIANGQING_DIR, PICTURES, row, col, i, FLAG, xuhao):
"""创建动态表
Args:
output_doc (Document): 文档对象
output_dir (str): 输出目录
table_num (int): 表格序号
TABLES (list): 表格数据
JIANCHA_XIANGQING_DIR (str): 检查详情表目录
PICTURES (dict): 图片数据字典键为表索引值为图片路径列表
row (int): 行数
col (int): 列数
i (int): 表格序号
FLAG: 其他标志
Returns:
tuple: (i, table_num) 更新后的表格序号和表格数量
"""
for table_idx, Table in enumerate(TABLES):
print(Table)
output_doc, message = await add_table_to_document(output_dir, JIANCHA_XIANGQING_DIR, row, col, i, Table, FLAG)
print(message)
# 获取当前表格对应的图片
current_table_pictures = PICTURES.get(table_idx, [])
print(f"开始处理图片列表: {current_table_pictures}")
for picturedir in current_table_pictures:
try:
print(f"添加 {picturedir} {type(picturedir)}到表格{table_idx}")
resize_and_reduce_quality(picturedir, picturedir)
await add_picture_to_table(output_doc, output_dir, 4, 0, picturedir, i, 4.7232)
except Exception as e:
print(f"添加图片失败:{e}")
print(await search_and_replace(output_dir, 'tupian_xuhao', f'{xuhao}'))
table_num += 1
i += 1
xuhao += 1
return i, table_num, xuhao
def get_year_month(date):
"""根据格式化date字符串获取年月 'date': '二〇二一年十二月十日 9:00'
Args: date (str): 日期字符串
Returns: 年月字符串 '二〇二一年十二月'
"""
unit_map = {'1' : '', '2' : '', '3' : '', '4' : '', '5' : '', '6' : '', '7' : '', '8' : '', '9' : '', '0' : ''}
unit_map_month = {1 : '', 2 : '', 3 : '', 4 : '', 5 : '', 6 : '', 7 : '', 8 : '', 9 : '', 10 : '', 11 : '十一', 12 : '十二'}
year = date.split('')[0]
month = date.split('')[1].split('')[0]
year = ''.join([unit_map[i] for i in year])
month = unit_map_month[int(month)]
return f"{year}{month}"
def merge_info(frontend_info, default_info):
"""
合并前端传入的 info 和默认 info
规则如果前端传入的值为空None 或空字符串则使用默认值
Args:
frontend_info: 前端传入的字典
default_info: 默认的完整字典
Returns:
合并后的完整字典
"""
if not isinstance(frontend_info, dict) or frontend_info is None:
return default_info.copy()
merged_info = {}
for key, default_value in default_info.items():
# 获取前端传入的值
frontend_value = frontend_info.get(key)
# 判断前端值是否为空None 或空字符串)
if frontend_value is None or frontend_value == "":
merged_info[key] = default_value
else:
merged_info[key] = frontend_value
return merged_info

View File

@ -0,0 +1,205 @@
"""
缺陷图目录格式
缺陷图期望格式 _隔开
外部内部命名格式都如下
图片名xuhao_缺陷类型_缺陷位置_缺陷尺寸_可见程度_紧急程度_危重等级_维修建议
涂层损伤_叶片ps面距叶根3m处_缺陷尺寸弦向100mm轴向800mm_轻微_紧急_重要_建议打磨维修
每个的选项见我发的图
防雷
轮毂至塔基导通阻值_169mΩ
缺陷例轮毂至塔基未导通 #即标明未导通即可
"""
DEFAULT_BASE_INFO = { #项目基本信息
"jizu_data" : {
'turbineName' : "",
},
"project_data" : {
"fengmian_dir" : "",
'farmName' : "",
'inspectionUnit' : "",
'inspectionContact' : "",
'inspectionPhone' : "",
'farmAddress' : "",
'client' : "",
'clientContact' : "",
'clientPhone' : "",
'scale' : "",
'turbineModel' : "",
'projectName' : "",
'startDate' : "",
'endDate' : "",
},
"shigong_data" : {
"startTime" : "",
"endTime" : "",
"weatherCode" : "",
"windSpeed" : "",
"imageCount" : "",
"temperature" : "",
},
"partManufacturer" : "",
"Y1" : {
"Code" : "1",
"list_dict" : {
"描述" : "路径",
#.....
},
},
"Y2" : {
"Code" : "2",
"list_dict" : {
"描述" : "路径",
#.....
},
},
"Y3" : {
"Code" : "3",
"list_dict" : {
"描述" : "路径",
#.....
},
}
}
DEFAULT_BAOGAO_INFO = {
#目录
'shengcheng_dir': "", #报告生成的路径
'muban_dir': "", #文档模板存放路径
"if_waibu" : False,
"if_neibu" : True,
"if_fanglei" : True,
"userName" : "admin",
"baogaoCheck" : "未审核",
'key_words': '缺,损,裂,脱,污', #关键字,用于汇总图的名字包含缺陷时标红,匹配逻辑为正则匹配单个字则为红 后续可优化
"shigong_fangan" : "None",
'jiancha_renyuan': '张三',
"baogao_zongjie" : "总结",
"total_pic_num" : "总图片数量",
"tatong_dir" : "塔筒图片",
"total_pic_dir" : "略缩图",
}
class JIANCHA_ENUM :
class WAIBU:
PART = "无人机外部高精度飞行"
#NEIRONG =
class SHIGONG_FANGAN_ENUM :
class WAIBU:
GONGZUO_NEIRONG = "无人机叶片外观巡检"
RENYUAN_PEIZHI = "1人主检飞手1人"
SHEBEI_PEIZHI = "1、大疆无人机1台M350rtkM300rtkM30TM30精灵4PRO2、大疆精灵4PRO+索尼A7R2机身+索尼200-600mm镜头/适马150-600mm镜头"
SHIGONG_FANGAN = ""
class NEIBU:
GONGZUO_NEIRONG = "叶片内部检查"
RENYUAN_PEIZHI = "2人轮毂作业检查2人"
SHEBEI_PEIZHI = "1、人工检查照明设备2套视频记录手机2台含氧量监测仪1台电动扳手2套卷尺1个。2、爬壁机器人检查无人作业校车+视频图传1套照明设备2套含氧量监测仪1台电动扳手2套卷尺1个。"
SHIGONG_FANGAN = ""
class FANGLEI:
class YEPIAN:
GONGZUO_NEIRONG = "无人机叶片防雷导通测试"
RENYUAN_PEIZHI = "2人主检飞手1人副检抄表1人"
SHEBEI_PEIZHI = "1四轴电阻无人机1套电子微欧计1台视频记录手机1台"
SHIGONG_FANGAN = ""
class DIAOLAN:
GONGZUO_NEIRONG = "无人吊篮叶片导通测试(含机舱设备、)"
RENYUAN_PEIZHI = "3人轮毂机舱作业1人揽风绳作业1人无人设备操作员及抄表1人"
SHEBEI_PEIZHI = "无人吊篮系统1套爬绳器+接触平台、电子微欧计1套视频记录手机1台对讲机2台"
SHIGONG_FANGAN = ""
class SHESHI:
GONGZUO_NEIRONG = "风机基础、办公楼、变电站防雷接地检测及浪涌保护器测试"
RENYUAN_PEIZHI = "1人抄表人员1人检测人员1人监护1人。"
SHEBEI_PEIZHI = "接地电阻测试仪1套、SPD测试仪1套、对讲机2个、"
SHIGONG_FANGAN = ""
FEISHOURENYUAN_PEIZHI = "1人主检飞手1人"
LUNGUZUOYERENYUAN_PEIZHI = "2人轮毂作业检查2人"
oneproject = {
"status": 200,
"data": {
"projectId": "96e0debf78187300f144d7f3450a2477",
"projectName": "三峡能源阿城万兴风电场防雷通道检测项目",
"coverUrl": "",
"farmName": "三峡能源阿城万兴风电场",
"farmAddress": "哈尔滨市阿城区",
"client": "辽宁信达检测有限公司",
"clientContact": "李经理",
"clientPhone": "13504783720",
"inspectionUnit": "武汉市迪特影像科技有限公司",
"inspectionContact": "吴名州",
"inspectionPhone": "18807109269",
"scale": "",
"turbineModel": "",
"constructorIds": "5709ccfece2685090ff700a3469f2539,a76d78f1325deda1790a12bdad4aad4e",
"auditorId": "ca37c4337df8673a5c045b6c25acf74a",
"qualityOfficerId": "862e027910c2562d2b67d88ec33d77ba",
"projectManagerId": "fbaa9e0aecf2ce287138c38a4b654085",
"constructionTeamLeaderId": "",
"status": 0,
"startDate": "",
"endDate": "",
"constructorName": "",
"auditorName": "李四",
"qualityOfficerName": "辛奇",
"projectManagerName": "张三",
"constructionTeamLeaderName": "",
"statusLabel": "待施工"
},
"msg": "",
"code": 200,
"success": True
}
onejizu = {
"status": 200,
"data": [
{
"turbineId": "183463dbf40d9278549a76b82b175dd9",
"projectId": "96e0debf78187300f144d7f3450a2477",
"projectName": "三峡能源阿城万兴风电场防雷通道检测项目",
"turbineName": "一期012号",
"turbineCode": "00000",
"turbineDesc": "一期012号全新设备",
"turbineManufacturer": "",
"turbineModel": "",
"turbineCoverUrl": ""
}
],
"msg": "",
"code": 200,
"success": True
}
yepian = {
"status": 200,
"data": [
{
"partId": "12bc30fb209f3af3bf530541c5b062bc",
"projectId": "96e0debf78187300f144d7f3450a2477",
"projectName": "三峡能源阿城万兴风电场防雷通道检测项目",
"turbineId": "183463dbf40d9278549a76b82b175dd9",
"turbineName": "一期012号",
"partName": "叶片2",
"partCode": "0001",
"partType": "VANE-2",
"partTypeLabel": "叶片2"
},
{
"partId": "12bc30fb209f3af3bf530541c5b062bd",
"projectId": "96e0debf78187300f144d7f3450a2477",
"projectName": "三峡能源阿城万兴风电场防雷通道检测项目",
"turbineId": "183463dbf40d9278549a76b82b175dd9",
"turbineName": "一期012号",
"partName": "叶片1",
"partCode": "0000",
"partType": "VANE-1",
"partTypeLabel": "叶片1"
}
],
"msg": "",
"code": 200,
"success": True
}

View File

@ -0,0 +1,610 @@
"""
Document creation and manipulation tools for Word Document Server.
"""
import os
import json, re
from typing import Dict, List, Optional, Any
from docx import Document
from ..utils.file_utils import check_file_writeable, ensure_docx_extension, create_document_copy
from ..utils.document_utils import get_document_properties, extract_document_text, get_document_structure
from ..core.styles import ensure_heading_style, ensure_table_style
from docx.oxml.shared import qn
from docx.oxml import OxmlElement
from .content_tools import search_and_replace,add_picture_to_table
async def create_document(filename: str, title: Optional[str] = None, author: Optional[str] = None) -> str:
"""创建一个包含可选元数据的新Word文档。
参数:
filename: 要创建的文档名称带或不带.docx扩展名
title: 可选标题
author: 可选作者
"""
filename = ensure_docx_extension(filename)
# Check if file is writeable
is_writeable, error_message = check_file_writeable(filename)
if not is_writeable:
return f"Cannot create document: {error_message}"
try:
doc = Document()
# Set properties if provided
if title:
doc.core_properties.title = title
if author:
doc.core_properties.author = author
# Ensure necessary styles exist
ensure_heading_style(doc)
ensure_table_style(doc)
# 更改纸张大小为A4
from docx.shared import Mm, Inches
sections = doc.sections
for section in sections:
section.page_height = Mm(297)
section.page_width = Mm(210)
section.left_margin = Inches(0.94)
section.right_margin = Inches(0.94)
# Save the document
doc.save(filename)
return f"Document {filename} created successfully"
except Exception as e:
return f"Failed to create document: {str(e)}"
async def get_document_info(filename: str) -> str:
"""获得文档信息
Args:
filename: 目标文档
"""
filename = ensure_docx_extension(filename)
if not os.path.exists(filename):
return f"Document {filename} does not exist"
try:
properties = get_document_properties(filename)
return json.dumps(properties, indent=2)
except Exception as e:
return f"Failed to get document info: {str(e)}"
async def get_document_text(filename: str) -> str:
"""获得文档的所有文本
Args:
filename: 目标文档
"""
filename = ensure_docx_extension(filename)
return extract_document_text(filename)
async def get_document_outline(filename: str) -> str:
"""获得文档的所有结构信息
Args:
filename: 目标文档
"""
filename = ensure_docx_extension(filename)
structure = get_document_structure(filename)
return json.dumps(structure, indent=2)
async def list_available_documents(directory: str = ".") -> str:
"""列出目录下所有Word文档
Args:
directory: 目录
"""
try:
if not os.path.exists(directory):
return f"Directory {directory} does not exist"
docx_files = [f for f in os.listdir(directory) if f.endswith('.docx')]
if not docx_files:
return f"No Word documents found in {directory}"
result = f"Found {len(docx_files)} Word documents in {directory}:\n"
for file in docx_files:
file_path = os.path.join(directory, file)
size = os.path.getsize(file_path) / 1024 # KB
result += f"- {file} ({size:.2f} KB)\n"
return result
except Exception as e:
return f"Failed to list documents: {str(e)}"
async def copy_document(source_filename: str, destination_filename: Optional[str] = None) -> str:
"""创建文档的副本
Args:
source_filename: 源文档路径
destination_filename: 目标文档路径为空则为当前目录
"""
source_filename = ensure_docx_extension(source_filename)
if destination_filename:
destination_filename = ensure_docx_extension(destination_filename)
success, message, new_path = create_document_copy(source_filename, destination_filename)
if success:
return message
else:
return f"Failed to copy document: {message}"
def add_documents(target_filename: str, source_filename: str) -> str:
"""将源文档(文本)添加到目标文档尾部
Args:
target_doc: 目标文档
source_filename: 源文档路径
"""
target_doc = Document(target_filename)
source_filename = ensure_docx_extension(source_filename)
source_doc = Document(source_filename)
for source_paragraph in source_doc.paragraphs:
new_paragraph = target_doc.add_paragraph(source_paragraph.text)
new_paragraph.style = target_doc.styles['Normal'] # Default style
#获取合并等样式2025427
new_paragraph.alignment = source_paragraph.alignment
print(f"Source paragraph alignment: {source_paragraph.alignment}")
# Try to match the style if possible
try:
if source_paragraph.style and source_paragraph.style.name in target_doc.styles:
new_paragraph.style = target_doc.styles[source_paragraph.style.name]
except Exception as e:
print(f"Failed to apply style: {e}")
# Copy run formatting
for i, run in enumerate(source_paragraph.runs):
if i < len(new_paragraph.runs):
new_run = new_paragraph.runs[i]
# Copy basic formatting
new_run.bold = run.bold
new_run.italic = run.italic
new_run.underline = run.underline
#添加同时合并字体2025427
new_run.font.name = run.font.name
rPr = new_run.element.get_or_add_rPr()
rFonts = rPr.get_or_add_rFonts()
# 检查 run.font.name 是否为 None
if run.font.name is None:
# 设置默认的中文字体名称
run.font.name = '宋体 (中文正文)' # 或者使用其他你喜欢的中文字体
rFonts.set(qn('w:eastAsia'), run.font.name)
new_run.font.color.rgb = run.font.color.rgb
# Font size if specified
if run.font.size:
new_run.font.size = run.font.size
target_doc.save(target_filename)
return f"{target_filename}添加{source_filename}成功"
def write_table(target_filename: str, rows: int, cols: int, table_num: int, data: Optional[List[List[str]]] = None, ifadjustheight: Optional[bool] = True, height: Optional[float] = 1, key_words: re.Pattern[str] = None, ALIGMENT: Optional[str] = 'CENTER') -> Document:
"""填写word文档里的表格返回填写后的文档
Args:
target_filename: 目标文档路径
rows: 表格行数
cols: 表格列数
table_num: 表格序号
data: 表格数据二维列表每个单元格为字符串
ifadjustheight: bool为真则表格行高自动调整
"""
target_filename = ensure_docx_extension(target_filename)
# Check if target file is writeable
is_writeable, error_message = check_file_writeable(target_filename)
if not is_writeable:
return f"Cannot create target document: {error_message}"
try:
target_filename = ensure_docx_extension(target_filename)
target_doc = Document(target_filename)
except Exception as e:
print(f"获取{target_filename}失败:{str(e)}")
# Try to set the table style
try:
target_doc.tables[table_num].style = 'Table Grid'
except KeyError as k:
pass
except Exception as e:
print(f"{target_doc}最后一个表格更改样式失败: {str(e)}")
print("开始写入表格")
from docx.enum.table import WD_TABLE_ALIGNMENT
from docx.enum.table import WD_ALIGN_VERTICAL
from docx.shared import Pt, Inches, Cm, RGBColor
try:
if data:
for i, row_data in enumerate(data):
if i >= rows + 1:
break
for j, cell_text in enumerate(row_data):
if j >= cols + 1:
break
if str(cell_text) == "": continue
print(f"在[{i},{j}]处写入{str(cell_text)}")
target_doc.tables[table_num].cell(i,j).text = str(cell_text)
print(key_words, cell_text)
if key_words and key_words.search(str(cell_text)):
print(f'{cell_text}包含关键之,已置红')
target_doc.tables[table_num].cell(i,j).paragraphs[0].runs[0].font.color.rgb = RGBColor(255, 0, 0)
target_doc.tables[table_num].cell(i,j).paragraphs[0].runs[0].font.name = "Times New Roman" #设置英文字体
target_doc.tables[table_num].cell(i,j).paragraphs[0].runs[0].font.size = Pt(10.5) # 字体大小
target_doc.tables[table_num].cell(i,j).paragraphs[0].runs[0]._element.rPr.rFonts.set(qn('w:eastAsia'), '仿宋') #设置中文字体
if ALIGMENT == 'CENTER':
target_doc.tables[table_num].cell(i,j).paragraphs[0].paragraph_format.alignment = WD_TABLE_ALIGNMENT.CENTER
elif ALIGMENT == 'LEFT':
target_doc.tables[table_num].cell(i,j).paragraphs[0].paragraph_format.alignment = WD_TABLE_ALIGNMENT.LEFT
target_doc.tables[table_num].cell(i,j).vertical_alignment = WD_ALIGN_VERTICAL.CENTER
if ifadjustheight:
target_doc.tables[table_num].rows[i].height = Cm(height)
except Exception as e:
print(f"写入{target_filename}tables.cell({i},{j})失败:{str(e)}")
print("表格写入完成")
return target_doc
def set_document_para(target_doc: Document) -> Document:
"""设置文档的段落格式
"""
paragraphs_to_remove = []
for i, paragraph in enumerate(target_doc.paragraphs):
if i <= 11:
continue
if not paragraph.text.strip():
paragraphs_to_remove.append(paragraph)
for paragraph in paragraphs_to_remove:
p = paragraph._element
p.getparent().remove(p)
return target_doc
async def add_table_to_document(target_filename: str, source_filename: str, rows: int, cols: int, table_num: int, data: Optional[List[List[str]]] = None, ifadjustheight: Optional[bool] = True, height: Optional[float] = 1, key_words: re.Pattern[str] = None, ALIGMENT: Optional[str] = 'CENTER') -> str:
"""复制源文件中的文字与表格(先文字后表格格式)到目标文档
Args:
target_filename: 目标文档路径
source_doc: 源文档路径
rows: 表格行数
cols: 表格列数
table_num: 表格序号
data: 表格数据二维列表每个单元格为字符串
ifadjustheight: bool为真则表格行高自动调整
key_words: list, 关键字
"""
target_filename = ensure_docx_extension(target_filename)
source_filename = ensure_docx_extension(source_filename)
source_doc = Document(source_filename)
target_doc = Document(target_filename)
try:
# Copy all paragraphs
for paragraph in source_doc.paragraphs:
# Create a new paragraph with the same text and style
new_paragraph = target_doc.add_paragraph(paragraph.text)
new_paragraph.style = target_doc.styles['Normal'] # Default style
#获取合并等样式2025427
new_paragraph.alignment = paragraph.alignment
# 复制段落分页属性
new_paragraph.paragraph_format.page_break_before = paragraph.paragraph_format.page_break_before
# Try to match the style if possible
try:
if paragraph.style and paragraph.style.name in target_doc.styles:
new_paragraph.style = target_doc.styles[paragraph.style.name]
except:
pass
# Copy run formatting
for i, run in enumerate(paragraph.runs):
if i < len(new_paragraph.runs):
new_run = new_paragraph.runs[i]
# Copy basic formatting
new_run.bold = run.bold
new_run.italic = run.italic
new_run.underline = run.underline
#添加同时合并字体2025427
new_run.font.name = run.font.name
rPr = new_run.element.get_or_add_rPr()
rFonts = rPr.get_or_add_rFonts()
# 检查 run.font.name 是否为 None
if run.font.name is None:
# 设置默认的中文字体名称
run.font.name = '宋体(中文正文)' # 或者使用其他你喜欢的中文字体
rFonts.set(qn('w:eastAsia'), run.font.name)
new_run.font.color.rgb = run.font.color.rgb
# Font size if specified
if run.font.size:
new_run.font.size = run.font.size
# 复制分页符处理w:br标签
for element in run._element:
if element.tag.endswith('br'):
br_type = element.get(qn('type'), '')
if br_type == 'page':
new_br = OxmlElement('w:br')
new_br.set(qn('type'), 'page')
new_run._element.append(new_br)
except Exception as e:
print(f"添加表格前文章失败:{str(e)}")
try:# Copy all tables
from ..core.tables import copy_table
copy_table(source_doc.tables[0], target_doc, ifadjustheight, height)
except Exception as e:
print(f"添加表格失败:{str(e)}")
print(f"{target_doc}写入表格{source_doc.tables[0]}成功")
target_doc = set_document_para(target_doc)
target_doc.save(target_filename)
target_doc = Document(target_filename)
try:
target_doc = write_table(target_filename, rows, cols, table_num, data, ifadjustheight, height, key_words, ALIGMENT)
except Exception as e:
print(f"{target_filename}写入{data}失败:{str(e)}")
target_doc.save(target_filename)
return target_doc,f"{target_filename}添加表格{source_doc}成功"
async def add_table_and_replace(target_filename: str, source_filename: str, ifadjustheight: Optional[bool] = True, list_to_replace: dict = {}, height: Optional[float] = 1):
"""复制源文件中的文字与表格(先文字后表格格式)到目标文档
Args:
target_filename: 目标文档路径
source_doc: 源文档路径
ifadjustheight: bool为真则表格行高自动调整
list_to_replace: dict, 待替换内容和替换内容
"""
target_filename = ensure_docx_extension(target_filename)
source_filename = ensure_docx_extension(source_filename)
source_doc = Document(source_filename)
target_doc = Document(target_filename)
try:
# Copy all paragraphs
for paragraph in source_doc.paragraphs:
# Create a new paragraph with the same text and style
new_paragraph = target_doc.add_paragraph(paragraph.text)
new_paragraph.style = target_doc.styles['Normal'] # Default style
#获取合并等样式2025427
new_paragraph.alignment = paragraph.alignment
# 复制段落分页属性
new_paragraph.paragraph_format.page_break_before = paragraph.paragraph_format.page_break_before
# Try to match the style if possible
try:
if paragraph.style and paragraph.style.name in target_doc.styles:
new_paragraph.style = target_doc.styles[paragraph.style.name]
except:
pass
# Copy run formatting
for i, run in enumerate(paragraph.runs):
if i < len(new_paragraph.runs):
new_run = new_paragraph.runs[i]
# Copy basic formatting
new_run.bold = run.bold
new_run.italic = run.italic
new_run.underline = run.underline
#添加同时合并字体2025427
new_run.font.name = run.font.name
rPr = new_run.element.get_or_add_rPr()
rFonts = rPr.get_or_add_rFonts()
# 检查 run.font.name 是否为 None
if run.font.name is None:
# 设置默认的中文字体名称
run.font.name = '宋体(中文正文)' # 或者使用其他你喜欢的中文字体
rFonts.set(qn('w:eastAsia'), run.font.name)
new_run.font.color.rgb = run.font.color.rgb
# Font size if specified
if run.font.size:
new_run.font.size = run.font.size
# 复制分页符处理w:br标签
for element in run._element:
if element.tag.endswith('br'):
br_type = element.get(qn('type'), '')
if br_type == 'page':
new_br = OxmlElement('w:br')
new_br.set(qn('type'), 'page')
new_run._element.append(new_br)
except Exception as e:
print(f"添加表格前文章失败:{str(e)}")
try:# Copy all tables
from ..core.tables import copy_table
copy_table(source_doc.tables[0], target_doc, ifadjustheight, height)
target_doc.save(target_filename)
except Exception as e:
print(f"添加表格失败:{str(e)}")
for find_text, replace_text in list_to_replace.items():
print(await search_and_replace(target_filename, find_text, replace_text))
async def merge_documents(target_filename: str, source_filenames: List[str], add_page_breaks: bool = True) -> str:
"""合并文档(文本) 表格会添加到最后
Args:
target_filename: 合并后文档路径
source_filenames: 源文档路径列表
add_page_breaks: bool为真则每个源文档中间加入分页符
"""
from ..core.tables import copy_table
target_filename = ensure_docx_extension(target_filename)
# Check if target file is writeable
is_writeable, error_message = check_file_writeable(target_filename)
if not is_writeable:
return f"Cannot create target document: {error_message}"
# Validate all source documents exist
missing_files = []
for filename in source_filenames:
doc_filename = ensure_docx_extension(filename)
if not os.path.exists(doc_filename):
missing_files.append(doc_filename)
if missing_files:
return f"Cannot merge documents. The following source files do not exist: {', '.join(missing_files)}"
try:
# Create a new document for the merged result
target_doc = Document()
# Process each source document
for i, filename in enumerate(source_filenames):
doc_filename = ensure_docx_extension(filename)
source_doc = Document(doc_filename)
# Add page break between documents (except before the first one)
if add_page_breaks and i > 0:
target_doc.add_page_break()
# Copy all paragraphs
for paragraph in source_doc.paragraphs:
# Create a new paragraph with the same text and style
new_paragraph = target_doc.add_paragraph(paragraph.text)
new_paragraph.style = target_doc.styles['Normal'] # Default style
#获取合并等样式2025427
new_paragraph.alignment = paragraph.alignment
# Try to match the style if possible
try:
if paragraph.style and paragraph.style.name in target_doc.styles:
new_paragraph.style = target_doc.styles[paragraph.style.name]
except:
pass
# Copy run formatting
for i, run in enumerate(paragraph.runs):
if i < len(new_paragraph.runs):
new_run = new_paragraph.runs[i]
# Copy basic formatting
new_run.bold = run.bold
new_run.italic = run.italic
new_run.underline = run.underline
#添加同时合并字体2025427
new_run.font.name = run.font.name
rPr = new_run.element.get_or_add_rPr()
rFonts = rPr.get_or_add_rFonts()
# 检查 run.font.name 是否为 None
if run.font.name is None:
# 设置默认的中文字体名称
run.font.name = '宋体(中文正文)' # 或者使用其他你喜欢的中文字体
rFonts.set(qn('w:eastAsia'), run.font.name)
new_run.font.color.rgb = run.font.color.rgb
# Font size if specified
if run.font.size:
new_run.font.size = run.font.size
# Copy all tables
for table in source_doc.tables:
copy_table(table, target_doc)
# Save the merged document
target_doc.save(target_filename)
return f"Successfully merged {len(source_filenames)} documents into {target_filename}"
except Exception as e:
return f"Failed to merge documents: {str(e)}"
async def right_align_last_three_para(target_filename: str) -> str:
"""右对齐最后三个段落
Args:
target_filename: 目标文档路径
"""
target_filename = ensure_docx_extension(target_filename)
# Check if target file is writeable
is_writeable, error_message = check_file_writeable(target_filename)
if not is_writeable:
return f"Cannot right align paragraphs: {error_message}"
try:
# Open the target document
target_doc = Document(target_filename)
# Get the last three paragraphs
paragraphs = target_doc.paragraphs[-3:]
# Set the alignment of each paragraph to right
from docx.enum.text import WD_ALIGN_PARAGRAPH
for paragraph in paragraphs:
paragraph.alignment = WD_ALIGN_PARAGRAPH.RIGHT
# Save the modified document
target_doc.save(target_filename)
return f"Successfully right aligned the last three paragraphs in {target_filename}"
except Exception as e:
return f"Failed to right align paragraphs: {str(e)}"
async def process_images_table(data_dict, output_dir, start_i, JIANCHA_NEIRONG_PICTURES_TABLE, key_words = None):
"""添加对应表格且填写图片名与插入图片
Args:
data_dict (dict): dict内容图片:图片路径
output_dir (str): 输出路径
start_i (int): 总表格数量
JIANCHA_NEIRONG_PICTURES_TABLE (str): 二维表模板路径
Returns:
int: 最后使用的表格序号
"""
items = list(data_dict.items())
picture_num = len(items)
line_index = 0
picture_index = 0
i = start_i
for content_row in range(((picture_num + 2) // 3) * 2):
if content_row % 2 == 1:
# 文字行(从 items 取图片名)
JIANCHA_NEIRONG_TEXT = [["" for _ in range(3)] for _ in range(1)] # 1行3列
for k in range(1): # 只有1行
for l in range(3):
if line_index >= picture_num:
break
JIANCHA_NEIRONG_TEXT[k][l] = items[line_index][0] # 图片名
print(f'当前为文字表格,在({k},{l})位置插入文字: {items[line_index][0]}')
line_index += 1
print(f"当前待插入表格: {JIANCHA_NEIRONG_TEXT}")
print(f"当前表格序号为 {i}")
output_doc, message = await add_table_to_document(
output_dir, JIANCHA_NEIRONG_PICTURES_TABLE, 1, 3, i, JIANCHA_NEIRONG_TEXT, False, None, key_words
)
i += 1
else:
# 图片行(从 items 取图片路径)
print(f"当前表格序号为 {i}")
output_doc, message = await add_table_to_document(
output_dir, JIANCHA_NEIRONG_PICTURES_TABLE, 1, 3, i, None, False
)
for k in range(3):
if picture_index < picture_num:
pic_path = items[picture_index][1] # 图片路径
print(f"当前为图片表格,在(0,{k})位置插入图片: {pic_path}")
print(await add_picture_to_table(output_doc, output_dir, 0, k, pic_path, i, 1.8898))
picture_index += 1
i += 1
print(message)
return i # 返回最后使用的表格序号

View File

@ -0,0 +1,182 @@
from .content_tools import add_picture_to_table, search_and_replace
from .get_pictures import resize_and_reduce_quality
from .document_tools import add_table_to_document
def fill_tables(Y_table_list, row, col, Y_Table_num, Y):
"""根据前端返回json块填写表格list并实时跟进已填写表格数量
目前只支持固定的缺陷图的填写
Args:
Y_table_list (list): 前端返回的json块
row (int): 表格行数
col (int): 表格列数
Y_Table_num: json块中有几个表格
Xu_Hao: 是第几个json块
Y: 其他参数
Return:
Y1_TABLES: 三维表和对应元素
table_index_to_images: 字典表索引到图片路径列表的映射
Xu_Hao到达第几个表了
"""
table_index_to_images = {}
Y_TABLES = [[["" for _ in range(row)] for _ in range(col)] for _ in range(Y_Table_num)]
# 处理前端返回数据
for l, table_dict in enumerate(Y_table_list):
if table_dict:
Y_TABLES[l][1][0] = Y
Y_TABLES[l][1][1] = table_dict["QueXianLeiXing"]
Y_TABLES[l][1][2] = table_dict["QueXianWeiZhi"]
Y_TABLES[l][1][3] = table_dict["QueXianChiCun"]
Y_TABLES[l][3][0] = table_dict["WeiZongDengJi"]
Y_TABLES[l][3][1] = table_dict["visibility"]
Y_TABLES[l][3][2] = table_dict["urgency"]
Y_TABLES[l][3][3] = table_dict["repair_suggestion"]
# 获取图片路径
image_path = table_dict['Tupian_Dir']
if image_path:
# 确保路径是字符串形式
if isinstance(image_path, list):
table_index_to_images[l] = image_path.copy()
else:
table_index_to_images[l] = [str(image_path)]
return Y_TABLES, table_index_to_images
async def process_images_table(data_dict, output_dir, start_i, JIANCHA_NEIRONG_PICTURES_TABLE, key_words = None):
"""添加对应表格且填写图片名与插入图片
Args:
data_dict (dict): dict内容图片:图片路径
output_dir (str): 输出路径
start_i (int): 总表格数量
JIANCHA_NEIRONG_PICTURES_TABLE (str): 二维表模板路径
Returns:
int: 最后使用的表格序号
"""
items = list(data_dict.items())
picture_num = len(items)
line_index = 0
picture_index = 0
i = start_i
for content_row in range(((picture_num + 2) // 3) * 2):
if content_row % 2 == 1:
# 文字行(从 items 取图片名)
JIANCHA_NEIRONG_TEXT = [["" for _ in range(3)] for _ in range(1)] # 1行3列
for k in range(1): # 只有1行
for l in range(3):
if line_index >= picture_num:
break
JIANCHA_NEIRONG_TEXT[k][l] = items[line_index][0] # 图片名
print(f'当前为文字表格,在({k},{l})位置插入文字: {items[line_index][0]}')
line_index += 1
print(f"当前待插入表格: {JIANCHA_NEIRONG_TEXT}")
print(f"当前表格序号为 {i}")
output_doc, message = await add_table_to_document(
output_dir, JIANCHA_NEIRONG_PICTURES_TABLE, 1, 3, i, JIANCHA_NEIRONG_TEXT, False, None, key_words
)
i += 1
else:
# 图片行(从 items 取图片路径)
print(f"当前表格序号为 {i}")
output_doc, message = await add_table_to_document(
output_dir, JIANCHA_NEIRONG_PICTURES_TABLE, 1, 3, i, None, False
)
for k in range(3):
if picture_index < picture_num:
pic_path = items[picture_index][1] # 图片路径
print(f"当前为图片表格,在(0,{k})位置插入图片: {pic_path}")
print(await add_picture_to_table(output_doc, output_dir, 0, k, pic_path, i, 1.8898))
picture_index += 1
i += 1
print(message)
return i # 返回最后使用的表格序号
async def add_dynamic_table(output_doc, output_dir, table_num, TABLES, JIANCHA_XIANGQING_DIR, PICTURES, row, col, i, FLAG, xuhao):
"""创建动态表
Args:
output_doc (Document): 文档对象
output_dir (str): 输出目录
table_num (int): 表格序号
TABLES (list): 表格数据
JIANCHA_XIANGQING_DIR (str): 检查详情表目录
PICTURES (dict): 图片数据字典键为表索引值为图片路径列表
row (int): 行数
col (int): 列数
i (int): 表格序号
FLAG: 其他标志
Returns:
tuple: (i, table_num) 更新后的表格序号和表格数量
"""
for table_idx, Table in enumerate(TABLES):
print(Table)
output_doc, message = await add_table_to_document(output_dir, JIANCHA_XIANGQING_DIR, row, col, i, Table, FLAG)
print(message)
# 获取当前表格对应的图片
current_table_pictures = PICTURES.get(table_idx, [])
print(f"开始处理图片列表: {current_table_pictures}")
for picturedir in current_table_pictures:
try:
print(f"添加 {picturedir} {type(picturedir)}到表格{table_idx}")
resize_and_reduce_quality(picturedir, picturedir)
await add_picture_to_table(output_doc, output_dir, 4, 0, picturedir, i, 4.7232)
except Exception as e:
print(f"添加图片失败:{e}")
print(await search_and_replace(output_dir, 'tupian_xuhao', f'{xuhao}'))
table_num += 1
i += 1
xuhao += 1
return i, table_num, xuhao
def get_year_month(date):
"""根据格式化date字符串获取年月 'date': '二〇二一年十二月十日 9:00'
Args: date (str): 日期字符串
Returns: 年月字符串 '二〇二一年十二月'
"""
unit_map = {'1' : '', '2' : '', '3' : '', '4' : '', '5' : '', '6' : '', '7' : '', '8' : '', '9' : '', '0' : ''}
unit_map_month = {1 : '', 2 : '', 3 : '', 4 : '', 5 : '', 6 : '', 7 : '', 8 : '', 9 : '', 10 : '', 11 : '十一', 12 : '十二'}
year = date.split('')[0]
month = date.split('')[1].split('')[0]
year = ''.join([unit_map[i] for i in year])
month = unit_map_month[int(month)]
return f"{year}{month}"
def merge_info(frontend_info, default_info):
"""
合并前端传入的 info 和默认 info
规则如果前端传入的值为空None 或空字符串则使用默认值
Args:
frontend_info: 前端传入的字典
default_info: 默认的完整字典
Returns:
合并后的完整字典
"""
if not isinstance(frontend_info, dict) or frontend_info is None:
return default_info.copy()
merged_info = {}
for key, default_value in default_info.items():
# 获取前端传入的值
frontend_value = frontend_info.get(key)
# 判断前端值是否为空None 或空字符串)
if frontend_value is None or frontend_value == "":
merged_info[key] = default_value
else:
merged_info[key] = frontend_value
return merged_info

View File

@ -0,0 +1,283 @@
import os
import math
from PIL import Image
from concurrent.futures import ThreadPoolExecutor
def get_dict(Y : dict) -> dict:
"""给指定格式的Y字典返回字典的除第一行外的内容
Args:
Y (str): Y字典形如
"Y" : {
"Code" : "1",
"list_dict" : {
"描述" : "路径",
#.....
},
},
Returns:
dict: list_dict
"""
return Y['list_dict']
def resize_and_reduce_quality(image_path, output_path, target_width = None):
try:
# 检查图片文件大小
if os.path.getsize(image_path) < 10 * 1024 * 1024: # 10MB
print("图片文件大小小于10MB不进行调整")
return image_path
# 打开图片
with Image.open(image_path) as img:
# 计算新的高度以保持宽高比
if target_width is None:
target_width = img.width
aspect_ratio = img.height / img.width
new_height = int(target_width * aspect_ratio)
# 调整图片大小
img_resized = img.resize((target_width, new_height), Image.LANCZOS)
# 降低图片质量
quality = 70 # 质量从1最差到95最好可以根据需要调整
img_resized.save(output_path, quality=quality)
return output_path
except Exception as e:
return f"调整图片大小和质量时出现问题: {str(e)}"
def get_picture_nums(source_path: str) -> int:
picture_count = 0
for root, dirs, files in os.walk(source_path):
for file in files:
if file.lower().endswith(('.jpg', '.jpeg', '.png')) and not file.startswith('merged_thumbnail'):
picture_count += 1
return picture_count
def collect_defect_data(
Y: str,
picture_dir: str,
search_file_list: list = [],
) -> tuple[int, dict]:
"""
收集指定年份的缺陷图片数据并根据布尔值决定是否扫描特定类型的缺陷图
Args:
Y: 叶片号 "Y1""Y2""Y3"
picture_dir: 图片根目录
search_file_list (list, optional): 要搜索的文件列表.规定为3个元素
Returns:
(缺陷图片总数, 缺陷图片文件名字典)
"""
total_num = 0
result_dict = {}
try:
for defect_type in search_file_list:
dir_path = os.path.join(picture_dir, Y, defect_type)
num, img_dict = get_picture_nums_and_image_with_name(dir_path)
total_num += num
result_dict.update(img_dict)
except Exception as e:
print(f"获取图片数据时出现问题: {str(e)},搜寻的目录:{dir_path}")
return total_num, result_dict
def get_picture_nums_and_image_with_name(source_path: str) -> tuple[int, dict]:
"""
获取指定目录下图片的数量并返回每个图片的路径和名称字典
Args:
source_path (str): 要搜索的目录路径
Returns:
tuple: 包含两个元素的元组
picture_count (int): 图片数量
image_with_name (dict): 图片路径和名称的字典格式为 {图片名称: 图片完整路径}
"""
picture_count = 0
image_with_name = {}
name_list = []
for root, dirs, files in os.walk(source_path):
for file in files:
if file.lower().endswith(('.jpg', '.jpeg', '.png')) and not file.startswith('merged_thumbnail'):
picture_count += 1
image_with_name[os.path.splitext(file)[0]] = os.path.join(root, file)
return picture_count, image_with_name
def find_image(directory, image_name):
"""
在指定目录中查找指定名称的图片文件
参数:
directory (str): 要搜索的目录路径
image_name (str): 要查找的图片文件名(可带扩展名或不带)
返回:
str: 找到的图片完整路径如果未找到则返回None
"""
# 支持的图片扩展名列表
image_extensions = ['.jpg', '.jpeg', '.png', '.gif', '.bmp', '.tiff', '.webp']
# 遍历目录中的所有文件
for root, dirs, files in os.walk(directory):
for file in files:
# 获取文件名和扩展名
filename, ext = os.path.splitext(file)
# 检查是否匹配图片名称(带或不带扩展名)
if (file.lower() == image_name.lower() or
filename.lower() == image_name.lower() and ext.lower() in image_extensions):
return os.path.join(root, file)
return None
async def make_Thumbnail(source_path: str, output_path: str, size: tuple = (436, 233)) -> str:
"""获取目录下所有图片,将所有图片合并制作略缩图并保存
Args:
source_path: 源目录
output_path: 输出目录
size: 合并后的略缩图总大小 (宽度, 高度)
"""
print("略缩图处理中")
try:
if not os.path.exists(output_path):
print(f"无输出目录,创建中,输出目录为:{output_path}")
os.makedirs(output_path)
except Exception as e:
print(f"输出目录有问题:{e}")
return ""
#如果存在merged_thumbnail.jpg文件则直接返回该文件路径
if os.path.exists(os.path.join(output_path,'merged_thumbnail.jpg')):
print(f"已有略缩图,不用处理, 目前如需重新生成,请去往{output_path}目录 删除 merged_thumbnail.jpg 图片")
"""
此处可预留接口询问用户是否重新生成一份略缩图
"""
return os.path.join(output_path,'merged_thumbnail.jpg')
print("目录中无略缩图,合并略缩图中")
# 获取源目录下所有的图片文件
try:
image_files = []
for root, dirs, files in os.walk(source_path):
for file in files:
if file.lower().endswith(('.jpg', '.jpeg', '.png')):
image_files.append(os.path.join(root, file))
except Exception as e:
print(f"递归获取图片失败,原因:{e}")
if not image_files:
print("源目录中没有找到图片文件")
return ""
# 计算每个缩略图的大小
num_images = len(image_files)
target_width, target_height = size
# 计算最佳的缩略图排列方式
# 先尝试计算每行可以放多少个缩略图
aspect_ratio = target_width / target_height
cols = math.ceil(math.sqrt(num_images * aspect_ratio))
rows = math.ceil(num_images / cols)
# 计算单个缩略图的大小
thumb_width = target_width // cols
thumb_height = target_height // rows
# 创建线程池处理图片
with ThreadPoolExecutor() as executor:
thumbnails = list(executor.map(
lambda file: create_thumbnail(file, (thumb_width, thumb_height)),
image_files
))
# 过滤掉 None 值
thumbnails = [thumb for thumb in thumbnails if thumb is not None]
if not thumbnails:
print("没有成功创建任何略缩图")
return ""
# 计算实际需要的行数和列数
actual_cols = min(len(thumbnails), cols)
actual_rows = math.ceil(len(thumbnails) / actual_cols)
# 创建合并后的图像
merged_image = Image.new('RGB', (actual_cols * thumb_width, actual_rows * thumb_height))
# 粘贴缩略图
for index, thumb in enumerate(thumbnails):
row = index // actual_cols
col = index % actual_cols
merged_image.paste(thumb, (col * thumb_width, row * thumb_height))
# 如果最终尺寸不完全匹配,调整大小
if merged_image.size != size:
merged_image = merged_image.resize(size, Image.LANCZOS)
# 保存合并后的略缩图
merged_thumbnail_path = os.path.join(output_path, 'merged_thumbnail.jpg')
merged_image.save(merged_thumbnail_path)
print(f"合并后的略缩图已保存到:{merged_thumbnail_path}")
return merged_thumbnail_path
def create_thumbnail(file_path: str, size: tuple) -> Image:
"""创建单个图片的略缩图
Args:
file_path: 图片文件路径
size: 缩略图大小
"""
try:
with Image.open(file_path) as img:
# 保持原始宽高比
img.thumbnail(size, Image.LANCZOS)
# 创建新图像确保尺寸一致
new_img = Image.new('RGB', size)
new_img.paste(img, ((size[0] - img.width) // 2, (size[1] - img.height) // 2))
return new_img
except Exception as e:
print(f"图片处理有问题:{e}")
return None
def process_picture_data(picture_data : list[dict],
image_source_to_find : list[str],
quexian_type : str,
dianxing_type : str,
other_type : str
) -> tuple[list[str], list[str]]:
"""处理从数据库获取的图片数据
Args:
picture_data (list[dict]): 图片数据
image_source_to_find (list[str]): 要查找的图片来源枚举(外内防雷)
quexian_type (str): 缺陷类型枚举值
dianxing_type (str): 典型类型枚举值
other_type (str): 其他类型枚举值
Returns:
tuple(
defct_pictures, 缺陷图列表
dianxing_pictures, 典型图列表
other_pictures, 其他图列表
total_num 总图片数量
)
"""
#过滤目标来源的图片数据
picture_data = [pic for pic in picture_data if pic['imageSource'] in image_source_to_find]
#分别择出缺陷图和典型图
return ([pic for pic in picture_data if pic['imageType'] == quexian_type],
[pic for pic in picture_data if pic['imageType'] == dianxing_type],
[pic for pic in picture_data if pic['imageType'] == other_type],
len(picture_data))

View File

@ -0,0 +1,8 @@
"""
Utility functions for the Word Document Server.
This package contains utility modules for file operations and document handling.
"""
from .file_utils import check_file_writeable, create_document_copy, ensure_docx_extension
from .document_utils import get_document_properties, extract_document_text, get_document_structure, find_paragraph_by_text, find_and_replace_text

View File

@ -0,0 +1,167 @@
"""
Document utility functions for Word Document Server.
"""
import json
from typing import Dict, List, Any
from docx import Document
def get_document_properties(doc_path: str) -> Dict[str, Any]:
"""Get properties of a Word document."""
import os
if not os.path.exists(doc_path):
return {"error": f"Document {doc_path} does not exist"}
try:
doc = Document(doc_path)
core_props = doc.core_properties
return {
"title": core_props.title or "",
"author": core_props.author or "",
"subject": core_props.subject or "",
"keywords": core_props.keywords or "",
"created": str(core_props.created) if core_props.created else "",
"modified": str(core_props.modified) if core_props.modified else "",
"last_modified_by": core_props.last_modified_by or "",
"revision": core_props.revision or 0,
"page_count": len(doc.sections),
"word_count": sum(len(paragraph.text.split()) for paragraph in doc.paragraphs),
"paragraph_count": len(doc.paragraphs),
"table_count": len(doc.tables)
}
except Exception as e:
return {"error": f"Failed to get document properties: {str(e)}"}
def extract_document_text(doc_path: str) -> str:
"""Extract all text from a Word document."""
import os
if not os.path.exists(doc_path):
return f"Document {doc_path} does not exist"
try:
doc = Document(doc_path)
text = []
for paragraph in doc.paragraphs:
text.append(paragraph.text)
for table in doc.tables:
for row in table.rows:
for cell in row.cells:
for paragraph in cell.paragraphs:
text.append(paragraph.text)
return "\n".join(text)
except Exception as e:
return f"Failed to extract text: {str(e)}"
def get_document_structure(doc_path: str) -> Dict[str, Any]:
"""Get the structure of a Word document."""
import os
if not os.path.exists(doc_path):
return {"error": f"Document {doc_path} does not exist"}
try:
doc = Document(doc_path)
structure = {
"paragraphs": [],
"tables": []
}
# Get paragraphs
for i, para in enumerate(doc.paragraphs):
structure["paragraphs"].append({
"index": i,
"text": para.text[:100] + ("..." if len(para.text) > 100 else ""),
"style": para.style.name if para.style else "Normal"
})
# Get tables
for i, table in enumerate(doc.tables):
table_data = {
"index": i,
"rows": len(table.rows),
"columns": len(table.columns),
"preview": []
}
# Get sample of table data
max_rows = min(3, len(table.rows))
for row_idx in range(max_rows):
row_data = []
max_cols = min(3, len(table.columns))
for col_idx in range(max_cols):
try:
cell_text = table.cell(row_idx, col_idx).text
row_data.append(cell_text[:20] + ("..." if len(cell_text) > 20 else ""))
except IndexError:
row_data.append("N/A")
table_data["preview"].append(row_data)
structure["tables"].append(table_data)
return structure
except Exception as e:
return {"error": f"Failed to get document structure: {str(e)}"}
def find_paragraph_by_text(doc, text, partial_match=False):
"""
Find paragraphs containing specific text.
Args:
doc: Document object
text: Text to search for
partial_match: If True, matches paragraphs containing the text; if False, matches exact text
Returns:
List of paragraph indices that match the criteria
"""
matching_paragraphs = []
for i, para in enumerate(doc.paragraphs):
if partial_match and text in para.text:
matching_paragraphs.append(i)
elif not partial_match and para.text == text:
matching_paragraphs.append(i)
return matching_paragraphs
def find_and_replace_text(doc, old_text, new_text):
"""
Find and replace text throughout the document.
Args:
doc: Document object
old_text: Text to find
new_text: Text to replace with
Returns:
Number of replacements made
"""
count = 0
# Search in paragraphs
for para in doc.paragraphs:
if old_text in para.text:
for run in para.runs:
if old_text in run.text:
run.text = run.text.replace(old_text, new_text)
count += 1
# Search in tables
for table in doc.tables:
for row in table.rows:
for cell in row.cells:
for para in cell.paragraphs:
if old_text in para.text:
for run in para.runs:
if old_text in run.text:
run.text = run.text.replace(old_text, new_text)
count += 1
return count

View File

@ -0,0 +1,165 @@
"""
Extended document utilities for Word Document Server.
"""
from typing import Dict, List, Any, Tuple
from docx import Document
def get_paragraph_text(doc_path: str, paragraph_index: int) -> Dict[str, Any]:
"""
Get text from a specific paragraph in a Word document.
Args:
doc_path: Path to the Word document
paragraph_index: Index of the paragraph to extract (0-based)
Returns:
Dictionary with paragraph text and metadata
"""
import os
if not os.path.exists(doc_path):
return {"error": f"Document {doc_path} does not exist"}
try:
doc = Document(doc_path)
# Check if paragraph index is valid
if paragraph_index < 0 or paragraph_index >= len(doc.paragraphs):
return {"error": f"Invalid paragraph index: {paragraph_index}. Document has {len(doc.paragraphs)} paragraphs."}
paragraph = doc.paragraphs[paragraph_index]
return {
"index": paragraph_index,
"text": paragraph.text,
"style": paragraph.style.name if paragraph.style else "Normal",
"is_heading": paragraph.style.name.startswith("Heading") if paragraph.style else False
}
except Exception as e:
return {"error": f"Failed to get paragraph text: {str(e)}"}
def find_text(doc_path: str, text_to_find: str, match_case: bool = True, whole_word: bool = False) -> Dict[str, Any]:
"""
Find all occurrences of specific text in a Word document.
Args:
doc_path: Path to the Word document
text_to_find: Text to search for
match_case: Whether to perform case-sensitive search
whole_word: Whether to match whole words only
Returns:
Dictionary with search results
"""
import os
if not os.path.exists(doc_path):
return {"error": f"Document {doc_path} does not exist"}
if not text_to_find:
return {"error": "Search text cannot be empty"}
try:
doc = Document(doc_path)
results = {
"query": text_to_find,
"match_case": match_case,
"whole_word": whole_word,
"occurrences": [],
"total_count": 0
}
# Search in paragraphs
for i, para in enumerate(doc.paragraphs):
# Prepare text for comparison
para_text = para.text
search_text = text_to_find
if not match_case:
para_text = para_text.lower()
search_text = search_text.lower()
# Find all occurrences (simple implementation)
start_pos = 0
while True:
if whole_word:
# For whole word search, we need to check word boundaries
words = para_text.split()
found = False
for word_idx, word in enumerate(words):
if (word == search_text or
(not match_case and word.lower() == search_text.lower())):
results["occurrences"].append({
"paragraph_index": i,
"position": word_idx,
"context": para.text[:100] + ("..." if len(para.text) > 100 else "")
})
results["total_count"] += 1
found = True
# Break after checking all words
break
else:
# For substring search
pos = para_text.find(search_text, start_pos)
if pos == -1:
break
results["occurrences"].append({
"paragraph_index": i,
"position": pos,
"context": para.text[:100] + ("..." if len(para.text) > 100 else "")
})
results["total_count"] += 1
start_pos = pos + len(search_text)
# Search in tables
for table_idx, table in enumerate(doc.tables):
for row_idx, row in enumerate(table.rows):
for col_idx, cell in enumerate(row.cells):
for para_idx, para in enumerate(cell.paragraphs):
# Prepare text for comparison
para_text = para.text
search_text = text_to_find
if not match_case:
para_text = para_text.lower()
search_text = search_text.lower()
# Find all occurrences (simple implementation)
start_pos = 0
while True:
if whole_word:
# For whole word search, check word boundaries
words = para_text.split()
found = False
for word_idx, word in enumerate(words):
if (word == search_text or
(not match_case and word.lower() == search_text.lower())):
results["occurrences"].append({
"location": f"Table {table_idx}, Row {row_idx}, Column {col_idx}",
"position": word_idx,
"context": para.text[:100] + ("..." if len(para.text) > 100 else "")
})
results["total_count"] += 1
found = True
# Break after checking all words
break
else:
# For substring search
pos = para_text.find(search_text, start_pos)
if pos == -1:
break
results["occurrences"].append({
"location": f"Table {table_idx}, Row {row_idx}, Column {col_idx}",
"position": pos,
"context": para.text[:100] + ("..." if len(para.text) > 100 else "")
})
results["total_count"] += 1
start_pos = pos + len(search_text)
return results
except Exception as e:
return {"error": f"Failed to search for text: {str(e)}"}

View File

@ -0,0 +1,85 @@
"""
File utility functions for Word Document Server.
"""
import os
from typing import Tuple, Optional
import shutil
def check_file_writeable(filepath: str) -> Tuple[bool, str]:
"""
Check if a file can be written to.
Args:
filepath: Path to the file
Returns:
Tuple of (is_writeable, error_message)
"""
# If file doesn't exist, check if directory is writeable
if not os.path.exists(filepath):
directory = os.path.dirname(filepath)
# If no directory is specified (empty string), use current directory
if directory == '':
directory = '.'
if not os.path.exists(directory):
return False, f"Directory {directory} does not exist"
if not os.access(directory, os.W_OK):
return False, f"Directory {directory} is not writeable"
return True, ""
# If file exists, check if it's writeable
if not os.access(filepath, os.W_OK):
return False, f"File {filepath} is not writeable (permission denied)"
# Try to open the file for writing to see if it's locked
try:
with open(filepath, 'a'):
pass
return True, ""
except IOError as e:
return False, f"File {filepath} is not writeable: {str(e)}"
except Exception as e:
return False, f"Unknown error checking file permissions: {str(e)}"
def create_document_copy(source_path: str, dest_path: Optional[str] = None) -> Tuple[bool, str, Optional[str]]:
"""
Create a copy of a document.
Args:
source_path: Path to the source document
dest_path: Optional path for the new document. If not provided, will use source_path + '_copy.docx'
Returns:
Tuple of (success, message, new_filepath)
"""
if not os.path.exists(source_path):
return False, f"Source document {source_path} does not exist", None
if not dest_path:
# Generate a new filename if not provided
base, ext = os.path.splitext(source_path)
dest_path = f"{base}_copy{ext}"
try:
# Simple file copy
shutil.copy2(source_path, dest_path)
return True, f"Document copied to {dest_path}", dest_path
except Exception as e:
return False, f"Failed to copy document: {str(e)}", None
def ensure_docx_extension(filename: str) -> str:
"""
Ensure filename has .docx extension.
Args:
filename: The filename to check
Returns:
Filename with .docx extension
"""
if not filename.endswith('.docx'):
return filename + '.docx'
return filename

View File

@ -0,0 +1,5 @@
<vector xmlns:android="http://schemas.android.com/apk/res/android" android:height="24dp" android:tint="#EC0F0F" android:viewportHeight="24" android:viewportWidth="24" android:width="24dp">
<path android:fillColor="@android:color/white" android:pathData="M12,2C6.48,2 2,6.48 2,12c0,5.52 4.48,10 10,10s10,-4.48 10,-10C22,6.48 17.52,2 12,2zM16,16H8V8h8V16z"/>
</vector>

View File

@ -0,0 +1,5 @@
<vector xmlns:android="http://schemas.android.com/apk/res/android" android:height="24dp" android:tint="#537461" android:viewportHeight="24" android:viewportWidth="24" android:width="24dp">
<path android:fillColor="@android:color/white" android:pathData="M5,20h14v-2H5V20zM5,10h4v6h6v-6h4l-7,-7L5,10z"/>
</vector>

View File

@ -30,95 +30,12 @@
android:hint="密码"
android:inputType="textPassword"
android:layout_marginBottom="16dp"/>
<LinearLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:orientation="horizontal"
android:layout_marginBottom="16dp"
android:gravity="center_vertical">
<com.google.android.material.textfield.TextInputLayout
android:layout_width="0dp"
android:layout_height="wrap_content"
android:layout_weight="1"
style="@style/Widget.MaterialComponents.TextInputLayout.OutlinedBox.ExposedDropdownMenu"
android:hint="项目">
<AutoCompleteTextView
android:id="@+id/etProject"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:inputType="textMultiLine"
android:completionThreshold="1"
android:maxLines="3"
android:ellipsize="none"
android:singleLine="false"
android:focusable="true"
android:focusableInTouchMode="true"
android:imeOptions="actionDone"
android:dropDownWidth="match_parent"
android:dropDownHeight="wrap_content"
android:dropDownVerticalOffset="4dp"
android:dropDownAnchor="@id/etProject"
android:dropDownSelector="@android:color/darker_gray"/>
</com.google.android.material.textfield.TextInputLayout>
<Button
android:id="@+id/btnClearProject"
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:text="清除"
android:textSize="14sp"
android:textColor="#FF0000"
android:paddingStart="12dp"
android:paddingEnd="12dp"
android:minWidth="0dp"
android:minHeight="0dp"
android:background="?attr/selectableItemBackgroundBorderless"
android:layout_marginStart="8dp"/>
</LinearLayout>
<CheckBox
android:id="@+id/cbRemember"
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:text="记住登录信息"
android:layout_marginBottom="24dp"/>
<LinearLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:orientation="vertical"
android:layout_marginTop="8dp"
android:layout_marginBottom="16dp"
android:background="@android:color/darker_gray"
android:padding="8dp">
<TextView
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:text="已选项目信息:"
android:textSize="14sp"
android:textColor="@android:color/white"
android:layout_marginBottom="4dp"/>
<TextView
android:id="@+id/tvSelectedProjectName"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:textSize="16sp"
android:padding="4dp"
android:text="项目名称:未选择"
android:textColor="@android:color/white"
android:background="@android:color/transparent"/>
<TextView
android:id="@+id/tvSelectedProjectId"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:textSize="16sp"
android:padding="4dp"
android:text="项目ID未选择"
android:textColor="@android:color/white"
android:background="@android:color/transparent"/>
</LinearLayout>
<Button
android:id="@+id/btnLogin"
android:layout_width="match_parent"

View File

@ -52,56 +52,54 @@
android:orientation="vertical"
android:layout_marginBottom="4dp">
<!-- 项目名称(带图标和固定标签宽度) -->
<LinearLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:orientation="horizontal"
android:layout_marginBottom="4dp">
android:layout_marginBottom="16dp"
android:gravity="center_vertical">
<TextView
<com.google.android.material.textfield.TextInputLayout
android:layout_width="0dp"
android:layout_height="wrap_content"
android:layout_weight="1"
style="@style/Widget.MaterialComponents.TextInputLayout.OutlinedBox.ExposedDropdownMenu"
android:hint="项目">
<AutoCompleteTextView
android:id="@+id/etProject"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:inputType="textMultiLine"
android:completionThreshold="1"
android:maxLines="1"
android:ellipsize="none"
android:singleLine="true"
android:focusable="true"
android:focusableInTouchMode="true"
android:imeOptions="actionDone"
android:dropDownWidth="match_parent"
android:dropDownHeight="wrap_content"
android:dropDownVerticalOffset="4dp"
android:dropDownAnchor="@id/etProject"
android:dropDownSelector="@android:color/darker_gray"
android:text="项目: 未设置"
/>
</com.google.android.material.textfield.TextInputLayout>
<Button
android:id="@+id/btnClearProject"
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:text="📁 项目名称:"
android:text="清除"
android:textSize="14sp"
android:textColor="@color/text_secondary"
android:layout_marginRight="8dp"/>
<TextView
android:id="@+id/tv_project"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:text="未设置"
android:textSize="14sp"
android:textColor="@color/text_primary"
android:ellipsize="end"
android:maxLines="2"/>
</LinearLayout>
<!-- 项目ID带图标和固定标签宽度 -->
<LinearLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:orientation="horizontal"
android:layout_marginBottom="8dp">
<TextView
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:text="🆔 项目ID:"
android:textSize="14sp"
android:textColor="@color/text_secondary"
android:layout_marginRight="8dp"/>
<TextView
android:id="@+id/tv_projectId"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:text="未设置"
android:textSize="14sp"
android:textColor="@color/text_primary"
android:ellipsize="end"
android:maxLines="2"/>
android:textColor="#FF0000"
android:paddingStart="12dp"
android:paddingEnd="12dp"
android:minWidth="0dp"
android:minHeight="0dp"
android:background="?attr/selectableItemBackgroundBorderless"
android:layout_marginStart="8dp"/>
</LinearLayout>
</LinearLayout>
@ -109,84 +107,102 @@
<LinearLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:orientation="vertical">
android:orientation="horizontal"
android:layout_marginBottom="16dp"
android:gravity="center_vertical">
<!-- 机组ID -->
<LinearLayout
<com.google.android.material.textfield.TextInputLayout
android:layout_width="0dp"
android:layout_height="wrap_content"
android:layout_weight="1"
style="@style/Widget.MaterialComponents.TextInputLayout.OutlinedBox.ExposedDropdownMenu"
android:hint="机组">
<AutoCompleteTextView
android:id="@+id/etTurbine"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:orientation="horizontal"
android:layout_marginBottom="4dp">
android:inputType="textMultiLine"
android:completionThreshold="1"
android:maxLines="1"
android:ellipsize="none"
android:singleLine="true"
android:focusable="true"
android:focusableInTouchMode="true"
android:imeOptions="actionDone"
android:dropDownWidth="match_parent"
android:dropDownHeight="wrap_content"
android:dropDownVerticalOffset="4dp"
android:dropDownAnchor="@id/etTurbine"
android:dropDownSelector="@android:color/darker_gray"
android:text="机组: 未设置"
/>
</com.google.android.material.textfield.TextInputLayout>
<TextView
<Button
android:id="@+id/btnClearTurbine"
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:text="⚙️ 机组ID:"
android:text="清除"
android:textSize="14sp"
android:textColor="@color/text_secondary"
android:layout_marginRight="8dp"/>
<TextView
android:id="@+id/tvUnit"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:text="未设置"
android:textSize="14sp"
android:textColor="@color/text_primary"
android:ellipsize="end"
android:maxLines="1"/>
</LinearLayout>
<!-- 机组名称 -->
<LinearLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:orientation="horizontal"
android:layout_marginBottom="8dp">
<TextView
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:text="🏷️ 机组名字:"
android:textSize="14sp"
android:textColor="@color/text_secondary"
android:layout_marginRight="8dp"/>
<TextView
android:id="@+id/tvUnitName"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:text="未设置"
android:textSize="14sp"
android:textColor="@color/text_primary"
android:ellipsize="end"
android:maxLines="1"/>
</LinearLayout>
android:textColor="#FF0000"
android:paddingStart="12dp"
android:paddingEnd="12dp"
android:minWidth="0dp"
android:minHeight="0dp"
android:background="?attr/selectableItemBackgroundBorderless"
android:layout_marginStart="8dp"/>
</LinearLayout>
<!-- 叶片信息 -->
<LinearLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:orientation="horizontal">
android:orientation="horizontal"
android:layout_marginBottom="16dp"
android:gravity="center_vertical">
<TextView
android:layout_width="wrap_content"
<com.google.android.material.textfield.TextInputLayout
android:layout_width="0dp"
android:layout_height="wrap_content"
android:text="🍃 叶片号:"
android:textSize="14sp"
android:textColor="@color/text_secondary"
android:layout_marginRight="8dp"/>
android:layout_weight="1"
style="@style/Widget.MaterialComponents.TextInputLayout.OutlinedBox.ExposedDropdownMenu"
android:hint="部件">
<TextView
android:id="@+id/tvBlade"
<AutoCompleteTextView
android:id="@+id/etPart"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:text="未设置"
android:inputType="textMultiLine"
android:completionThreshold="1"
android:maxLines="1"
android:ellipsize="none"
android:singleLine="true"
android:focusable="true"
android:focusableInTouchMode="true"
android:imeOptions="actionDone"
android:dropDownWidth="match_parent"
android:dropDownHeight="wrap_content"
android:dropDownVerticalOffset="4dp"
android:dropDownAnchor="@id/etPart"
android:dropDownSelector="@android:color/darker_gray"
android:text="部件: 未设置"
/>
</com.google.android.material.textfield.TextInputLayout>
<Button
android:id="@+id/btnClearPart"
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:text="清除"
android:textSize="14sp"
android:textColor="@color/text_primary"
android:ellipsize="end"
android:maxLines="1"/>
android:textColor="#FF0000"
android:paddingStart="12dp"
android:paddingEnd="12dp"
android:minWidth="0dp"
android:minHeight="0dp"
android:background="?attr/selectableItemBackgroundBorderless"
android:layout_marginStart="8dp"/>
</LinearLayout>
</LinearLayout>
</com.google.android.material.card.MaterialCardView>
@ -204,6 +220,143 @@
android:orientation="vertical"
android:padding="8dp">
<!-- 天气信息编辑区 -->
<com.google.android.material.card.MaterialCardView
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:layout_marginBottom="12dp"
app:cardCornerRadius="8dp"
app:cardElevation="2dp">
<LinearLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:orientation="vertical"
android:padding="12dp">
<TextView
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:text="天气信息"
android:textSize="16sp"
android:textStyle="bold"
android:layout_marginBottom="8dp"/>
<!-- 天气信息编辑行 -->
<LinearLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:orientation="horizontal"
android:layout_marginBottom="8dp">
<TextView
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:text="天气:"
android:textSize="14sp"
android:layout_marginRight="8dp"/>
<com.google.android.material.textfield.TextInputLayout
android:layout_width="0dp"
android:layout_height="wrap_content"
android:layout_weight="1"
style="@style/Widget.MaterialComponents.TextInputLayout.OutlinedBox"
app:boxStrokeColor="#C0FFEB3B">
<com.google.android.material.textfield.TextInputEditText
android:id="@+id/et_weather"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:hint="请输入天气"
android:textSize="14sp"
android:inputType="text"/>
</com.google.android.material.textfield.TextInputLayout>
</LinearLayout>
<!-- 温度湿度双列布局 -->
<LinearLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:orientation="horizontal"
android:layout_marginBottom="8dp">
<!-- 温度输入 -->
<LinearLayout
android:layout_width="0dp"
android:layout_height="wrap_content"
android:layout_weight="1"
android:orientation="horizontal"
android:layout_marginEnd="4dp">
<TextView
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:text="温度:"
android:textSize="14sp"
android:layout_marginRight="8dp"/>
<com.google.android.material.textfield.TextInputLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
style="@style/Widget.MaterialComponents.TextInputLayout.OutlinedBox"
app:boxStrokeColor="#FFEB3B">
<com.google.android.material.textfield.TextInputEditText
android:id="@+id/et_temperature"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:hint="℃"
android:textSize="14sp"
android:inputType="number|numberDecimal"/>
</com.google.android.material.textfield.TextInputLayout>
</LinearLayout>
<!-- 湿度输入 -->
<LinearLayout
android:layout_width="0dp"
android:layout_height="wrap_content"
android:layout_weight="1"
android:orientation="horizontal"
android:layout_marginStart="4dp">
<TextView
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:text="湿度:"
android:textSize="14sp"
android:layout_marginRight="8dp"/>
<com.google.android.material.textfield.TextInputLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
style="@style/Widget.MaterialComponents.TextInputLayout.OutlinedBox"
app:boxStrokeColor="#FFEB3B">
<com.google.android.material.textfield.TextInputEditText
android:id="@+id/et_humidity"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:hint="%"
android:textSize="14sp"
android:inputType="number|numberDecimal"/>
</com.google.android.material.textfield.TextInputLayout>
</LinearLayout>
</LinearLayout>
<!-- 确认按钮 -->
<com.google.android.material.button.MaterialButton
android:id="@+id/btn_save_weather"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:text="保存天气信息"
android:textColor="#FFFFFF"
android:backgroundTint="#0288D1"
app:cornerRadius="8dp"/>
</LinearLayout>
</com.google.android.material.card.MaterialCardView>
<!-- 状态卡置顶 -->
<com.google.android.material.card.MaterialCardView
android:layout_width="match_parent"
@ -236,27 +389,41 @@
android:text="当前状态:未监听"
android:textColor="#212121"
android:textSize="14sp" />
<TextView
android:id="@+id/tem_text"
<!-- 显示天气信息 -->
<LinearLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:text="当前温度:为获取"
android:textColor="#212121"
android:textSize="14sp" />
android:orientation="horizontal"
android:layout_marginTop="8dp">
<TextView
android:id="@+id/hum_text"
android:layout_width="match_parent"
android:id="@+id/tv_weather_display"
android:layout_width="0dp"
android:layout_height="wrap_content"
android:text="当前湿度:未获取"
android:layout_weight="1"
android:text="天气: 未设置"
android:textColor="#212121"
android:textSize="14sp" />
android:textSize="14sp"/>
<TextView
android:id="@+id/weather_text"
android:layout_width="match_parent"
android:id="@+id/tv_temperature_display"
android:layout_width="0dp"
android:layout_height="wrap_content"
android:text="当前天气:为获取"
android:layout_weight="1"
android:text="温度: 未设置"
android:textColor="#212121"
android:textSize="14sp" />
android:textSize="14sp"/>
<TextView
android:id="@+id/tv_humidity_display"
android:layout_width="0dp"
android:layout_height="wrap_content"
android:layout_weight="1"
android:text="湿度: 未设置"
android:textColor="#212121"
android:textSize="14sp"/>
</LinearLayout>
</LinearLayout>
</com.google.android.material.card.MaterialCardView>
@ -482,28 +649,191 @@
android:textSize="14sp"/>
</LinearLayout>
</com.google.android.material.card.MaterialCardView>
<com.google.android.material.card.MaterialCardView
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:layout_marginBottom="12dp"
app:cardBackgroundColor="#FFFFFF"
app:cardCornerRadius="8dp"
app:cardElevation="2dp">
<LinearLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:orientation="vertical"
android:padding="12dp">
<TextView
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:text="录音管理区"
android:textColor="#9C27B0"
android:textSize="16sp"
android:textStyle="bold"
android:layout_marginBottom="8dp"/>
<LinearLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:orientation="horizontal"
android:layout_marginBottom="8dp">
<Button
android:id="@+id/btn_failed_audios"
android:layout_width="wrap_content"
style="@style/Widget.MaterialComponents.Button.OutlinedButton"
android:layout_width="0dp"
android:layout_height="wrap_content"
android:layout_weight="1"
android:layout_marginEnd="4dp"
android:text="查看失败录音"
android:layout_margin="8dp"/>
app:strokeColor="#F44336"
app:strokeWidth="1dp"/>
<Button
android:id="@+id/btn_success_audios"
android:layout_width="wrap_content"
style="@style/Widget.MaterialComponents.Button.OutlinedButton"
android:layout_width="0dp"
android:layout_height="wrap_content"
android:text="查看成功上传录音"
android:layout_margin="8dp"/>
android:layout_weight="1"
android:layout_marginStart="4dp"
android:text="查看成功录音"
app:strokeColor="#4CAF50"
app:strokeWidth="1dp"/>
</LinearLayout>
</LinearLayout>
</com.google.android.material.card.MaterialCardView>
<com.google.android.material.card.MaterialCardView
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:layout_marginBottom="12dp"
app:cardBackgroundColor="#FFFFFF"
app:cardCornerRadius="8dp"
app:cardElevation="2dp">
<LinearLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:orientation="vertical"
android:padding="12dp">
<TextView
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:text="日报管理区"
android:textColor="#3F51B5"
android:textSize="16sp"
android:textStyle="bold"
android:layout_marginBottom="8dp"/>
<LinearLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:orientation="horizontal"
android:layout_marginBottom="8dp">
<Button
android:id="@+id/btn_convert"
android:layout_width="wrap_content"
style="@style/Widget.MaterialComponents.Button.OutlinedButton"
android:layout_width="0dp"
android:layout_height="wrap_content"
android:text="根据日期生成日报" />
android:layout_weight="1"
android:layout_marginEnd="4dp"
android:text="生成日报"
app:strokeColor="#3F51B5"
app:strokeWidth="1dp"/>
<Button
android:id="@+id/btn_readdocx"
android:layout_width="wrap_content"
style="@style/Widget.MaterialComponents.Button.OutlinedButton"
android:layout_width="0dp"
android:layout_height="wrap_content"
android:text="查看已经生成日报" />
android:layout_weight="1"
android:layout_marginStart="4dp"
android:text="查看日报"
app:strokeColor="#3F51B5"
app:strokeWidth="1dp"/>
</LinearLayout>
</LinearLayout>
</com.google.android.material.card.MaterialCardView>
<!-- ==================== 正式报告管理区 ==================== -->
<com.google.android.material.card.MaterialCardView
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:layout_marginBottom="12dp"
app:cardBackgroundColor="#FFFFFF"
app:cardCornerRadius="8dp"
app:cardElevation="2dp">
<LinearLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:orientation="vertical"
android:padding="12dp">
<TextView
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:text="正式报告管理区(已经接入,未设置输入待完善)"
android:textColor="#009688"
android:textSize="16sp"
android:textStyle="bold"
android:layout_marginBottom="8dp"/>
<Button
android:id="@+id/btn_generate_report"
style="@style/Widget.MaterialComponents.Button.OutlinedButton"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:text="生成正式报告"
app:strokeColor="#009688"
app:strokeWidth="1dp"/>
<Button
android:id="@+id/btn_read_report"
style="@style/Widget.MaterialComponents.Button.OutlinedButton"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:text="查看正式报告"
app:strokeColor="#009688"
app:strokeWidth="1dp"/>
</LinearLayout>
</com.google.android.material.card.MaterialCardView>
<!-- ==================== 响应管理区 ==================== -->
<com.google.android.material.card.MaterialCardView
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:layout_marginBottom="12dp"
app:cardBackgroundColor="#FFFFFF"
app:cardCornerRadius="8dp"
app:cardElevation="2dp">
<LinearLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:orientation="vertical"
android:padding="12dp">
<TextView
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:text="响应管理区"
android:textColor="#FF5722"
android:textSize="16sp"
android:textStyle="bold"
android:layout_marginBottom="8dp"/>
<Button
android:id="@+id/readothermsg"
style="@style/Widget.MaterialComponents.Button.OutlinedButton"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:text="查看响应信息"
app:strokeColor="#FF5722"
app:strokeWidth="1dp"/>
</LinearLayout>
</com.google.android.material.card.MaterialCardView>
<Button
android:id="@+id/changePassword"
android:layout_width="match_parent"
@ -515,6 +845,7 @@
android:layout_height="wrap_content"
android:layout_marginTop="16dp"
android:text="准备就绪" />
<!-- 定时上传图片设置(移动到最下方) -->
<com.google.android.material.card.MaterialCardView
android:layout_width="match_parent"
@ -627,15 +958,6 @@
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:prompt="@string/image_source_prompt"/>
<Button
android:id="@+id/btnShowFloating"
style="@style/Widget.MaterialComponents.Button.OutlinedButton"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:layout_marginBottom="8dp"
android:text="显示悬浮窗"
app:strokeColor="#9C27B0"
app:strokeWidth="1dp"/>
<com.google.android.material.switchmaterial.SwitchMaterial
android:id="@+id/switch_monitor"

View File

@ -0,0 +1,20 @@
<?xml version="1.0" encoding="utf-8"?>
<LinearLayout xmlns:android="http://schemas.android.com/apk/res/android"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:orientation="vertical"
android:padding="8dp">
<TextView
android:id="@+id/audio_item_text"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:lineSpacingMultiplier="1.2"/>
<View
android:layout_width="match_parent"
android:layout_height="1dp"
android:background="@color/dividerColor"
android:layout_marginTop="8dp"/>
</LinearLayout>

View File

@ -33,21 +33,22 @@
android:layout_height="wrap_content"
android:orientation="horizontal"
android:layout_marginTop="16dp"
android:weightSum="4">
android:weightSum="4"
android:gravity="center">
<Button
android:id="@+id/btn_delete_selected"
android:layout_width="5dp"
android:layout_width="0dp"
android:layout_height="wrap_content"
android:layout_marginEnd="4dp"
android:layout_marginEnd="8dp"
android:layout_weight="1"
android:text="删除选中" />
<Button
android:id="@+id/btn_delete_all"
android:layout_width="52dp"
android:layout_height="match_parent"
android:layout_marginEnd="4dp"
android:layout_width="0dp"
android:layout_height="wrap_content"
android:layout_marginEnd="8dp"
android:layout_weight="1"
android:text="删除全部" />
@ -55,9 +56,9 @@
android:id="@+id/btn_share"
android:layout_width="0dp"
android:layout_height="wrap_content"
android:layout_marginEnd="8dp"
android:layout_weight="1"
android:text="分享"
android:layout_marginEnd="4dp"/>
android:text="分享"/>
<Button
android:id="@+id/btn_cancel"

View File

@ -7,12 +7,42 @@
android:background="#88F5F5F5"
android:padding="8dp">
<!-- 标题栏(新增) -->
<LinearLayout
android:layout_width="match_parent"
android:layout_height="30dp"
android:orientation="horizontal"
android:gravity="center_vertical"
android:background="#88000000">
<TextView
android:layout_width="0dp"
android:layout_height="wrap_content"
android:layout_weight="1"
android:text="悬浮窗"
android:textColor="#FFFFFF"
android:textSize="12sp"/>
<!-- 关闭按钮 -->
<TextView
android:id="@+id/btnClose"
android:layout_width="24dp"
android:layout_height="24dp"
android:text="×"
android:textSize="18sp"
android:textColor="#FFFFFF"
android:gravity="center"
android:clickable="true"
android:focusable="true"/>
</LinearLayout>
<!-- 机组号行 -->
<LinearLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:orientation="horizontal"
android:gravity="center_vertical">
android:gravity="center_vertical"
android:layout_marginTop="4dp">
<TextView
android:layout_width="wrap_content"

View File

@ -7,4 +7,9 @@
<color name="text_primary">#212121</color>
<color name="text_secondary">#424242</color>
<color name="divider">#E0E0E0</color>
<color name="colorPrimary">#3F51B5</color>
<color name="colorPrimaryDark">#303F9F</color>
<color name="colorAccent">#FF4081</color>
<color name="textColor">#333333</color>
<color name="dividerColor">#EEEEEE</color>
</resources>

View File

@ -1,7 +1,7 @@
<?xml version="1.0" encoding="utf-8"?>
<resources>
<!-- 应用名称 -->
<string name="app_name">My Application</string>
<string name="app_name">迪特聚能</string>
<string name="image_source_prompt">请选择图像来源</string>
<!-- 解决错误所需的字符串 -->

View File

@ -0,0 +1,17 @@
<?xml version="1.0" encoding="utf-8"?>
<resources>
<!-- 基础应用主题 -->
<style name="AppTheme" parent="Theme.AppCompat.Light.DarkActionBar">
<item name="colorPrimary">@color/colorPrimary</item>
<item name="colorPrimaryDark">@color/colorPrimaryDark</item>
<item name="colorAccent">@color/colorAccent</item>
</style>
<!-- 自定义对话框样式 -->
<style name="CustomAlertDialog" parent="Theme.AppCompat.Light.Dialog.Alert">
<item name="colorPrimary">@color/colorPrimary</item>
<item name="colorAccent">@color/colorAccent</item>
<item name="android:windowBackground">@android:color/white</item>
<item name="android:textColorPrimary">@color/textColor</item>
</style>
</resources>

View File

@ -4,6 +4,7 @@
<external-path name="external_files" path="." />
<!-- 应用内部存储目录 -->
<files-path name="internal_files" path="." />
<!-- 缓存目录 -->
<cache-path name="cache_files" path="." />
<!-- 外部存储中应用专属目录 -->

View File

@ -11,6 +11,7 @@ navigationUi = "2.6.0"
okhttp = "4.12.0"
room = "2.6.1"
firebaseCrashlyticsBuildtools = "3.0.3"
activity = "1.8.0"
[libraries]
junit = { group = "junit", name = "junit", version.ref = "junit" }
ext-junit = { group = "androidx.test.ext", name = "junit", version.ref = "junitVersion" }
@ -24,6 +25,7 @@ okhttp = { group = "com.squareup.okhttp3", name = "okhttp", version.ref = "okhtt
room-runtime = { group = "androidx.room", name = "room-runtime", version.ref = "room" }
room-compiler = { group = "androidx.room", name = "room-compiler", version.ref = "room" }
firebase-crashlytics-buildtools = { group = "com.google.firebase", name = "firebase-crashlytics-buildtools", version.ref = "firebaseCrashlyticsBuildtools" }
activity = { group = "androidx.activity", name = "activity", version.ref = "activity" }
[plugins]
android-application = { id = "com.android.application", version.ref = "agp" }