beenull/ente
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[mob][photos] Use EnteWakeLock in ML settings page

Browse source
This commit is contained in:
laurenspriem 2024-05-24 16:39:42 +05:30
parent 7b0f5909b5
commit f301ab57f2
1 changed files with 4 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
mobile/lib/ui/settings/machine_learning_settings_page.dart
View file

@ -27,7 +27,7 @@ import "package:photos/ui/components/toggle_switch_widget.dart";
import "package:photos/utils/data_util.dart";
import "package:photos/utils/local_settings.dart";
import "package:photos/utils/ml_util.dart";
import "package:wakelock_plus/wakelock_plus.dart";
import "package:photos/utils/wakelock_util.dart";
final _logger = Logger("MachineLearningSettingsPage");
@ -42,6 +42,7 @@ class MachineLearningSettingsPage extends StatefulWidget {
class _MachineLearningSettingsPageState
extends State<MachineLearningSettingsPage> {
late InitializationState _state;
final EnteWakeLock _wakeLock = EnteWakeLock();
late StreamSubscription<MLFrameworkInitializationUpdateEvent>
_eventSubscription;
@ -55,9 +56,7 @@ class _MachineLearningSettingsPageState
setState(() {});
});
_fetchState();
if (flagService.internalUser) {
unawaited(WakelockPlus.enable());
}
_wakeLock.enable();
}
void _fetchState() {
@ -68,9 +67,7 @@ class _MachineLearningSettingsPageState
void dispose() {
super.dispose();
_eventSubscription.cancel();
if (flagService.internalUser) {
unawaited(WakelockPlus.disable());
}
_wakeLock.disable();
}
@override