前言:
由于公司是做的医疗服务这块的,所以提出了工作室预约,设置日期为,工作日,半休日,休息日三种,还要支持多选功能,下面提供下具体的做法,需要的小伙伴可以拿去哈
具体的效果如下:
Untitled01.gif
具体用法:
由于封装的方法比较多,我就先把使用方法写出来,改变下套路.
// 为了防止有重影,最好是先移除
[self.calendarView removeFromSuperview];
self.calendarView = nil;
NSDateFormatter *Mformatter = [[NSDateFormatter alloc] init];
Mformatter.dateFormat = @"yyyy-MM";
NSString *monthStr = [Mformatter stringFromDate:self.date];
// 网络请求,设置工作室已经设置的日期
NSMutableDictionary *listPrams = [NSMutableDictionary dictionary];
listPrams[@"month"] = monthStr;
listPrams[@"studioid"] = _studio_id;
__weak typeof(self) weakself = self;
[WSService requestBusiness:URL_BOOKING_WORKDAY_LISTBYMONTH params:listPrams sucCallBack:^(id result) {
BOOL isValid = [APIRequestHandler isValidDataWithResult:result];
if (isValid) {
NSArray *data = result[@"data"];
// 三种不同的日期类型
NSMutableArray *restArray = [NSMutableArray array];
NSMutableArray *mendArray = [NSMutableArray array];
NSMutableArray *workArray = [NSMutableArray array];
for (int i = 0; i < data.count; i ++) {
NSDictionary *dict = data[i];
NSString *caltype = dict[@"caltype"];
NSString *calander = dict[@"calander"];
calander = [calander substringWithRange:NSMakeRange(calander.length - 2,2)];
if ([caltype isEqualToString:CALTYPE_RESTDAY]) { // 休息日
[restArray addObject:calander];
}else if ([caltype isEqualToString:CALTYPE_MENDDAY]) { // 半休日
[mendArray addObject:calander];
}else if ([caltype isEqualToString:CALTYPE_WORKDAY]) { // 工作日
[workArray addObject:calander];
}
}
weakself.calendarView = [[FyCalendarView alloc] initWithFrame:CGRectMake(0, 64, PDScreeenW, PDScreeenW)];
weakself.calendarView.allDaysArr = workArray;
weakself.calendarView.allDaysColor = workColor;
weakself.calendarView.partDaysArr = mendArray;
weakself.calendarView.partDaysColor = mendColor;
weakself.calendarView.otherDaysArr = restArray;
weakself.calendarView.otherDaysColor = restColor;
weakself.calendarView.date = [NSDate date];
weakself.calendarView.dateColor = selcetColor;
[weakself.view addSubview:weakself.calendarView];
// 默认选中当前日期
UIButton *defaultButton = weakself.calendarView.defaultBtn;
NSDateFormatter *formatter = [[NSDateFormatter alloc] init];
formatter.dateFormat = @"yyyy-MM-dd";
NSString *dateStr = [formatter stringFromDate:weakself.date];
if (defaultButton.selected) {
[weakself.buttonsSelet addObject:defaultButton];
[weakself.datesArray addObject:dateStr];
}
__weak typeof(weakself) weak_self = weakself;
weakself.calendarView.calendarBlock = ^(NSString *selectDate, UIButton *selectBtn){
if (selectBtn.selected) {
[weak_self.buttonsSelet addObject:selectBtn];
[weak_self.datesArray addObject:selectDate];
}else{
[weak_self.buttonsSelet removeObject:selectBtn];
[weak_self.datesArray removeObject:selectDate];
}
};
// 设置下个月
weakself.calendarView.nextMonthBlock = ^(){
[weak_self setupNextMonth];
};
// last month
weakself.calendarView.lastMonthBlock = ^(){
[weak_self setupLastMonth];
};
}
} failCallBack:^(id error) {
} hasHud:NO];
接下来的这两个方法自己实现下,一样的道理
[weak_self setupNextMonth];
[weak_self setupLastMonth];
具体实现代码:
- h 文件
#import <UIKit/UIKit.h>
@protocol CalendarView <NSObject>
/**
* 点击返回的日期
*/
- (void)setupToday:(NSInteger)day Month:(NSInteger)month Year:(NSInteger)year;
@end
@interface CalendarView : UIView
//set 选择日期
@property (nonatomic, strong) NSDate *date;
@property (nonatomic, strong) UIColor *dateColor;
//年月label
@property (nonatomic, strong) UILabel *headlabel;
// 选中文字颜色
@property (nonatomic, strong) UIColor *headColor;
//weekView
@property (nonatomic, strong) UIView *weekBg;
@property (nonatomic, strong) UIColor *weekDaysColor;
// 全天可用
@property (nonatomic, strong) NSMutableArray *allDaysArr;
@property (nonatomic, strong) UIColor *allDaysColor;
// 部分时段可用
@property (nonatomic, strong) NSMutableArray *partDaysArr;
@property (nonatomic, strong) UIColor *partDaysColor;
// 外加部分选择
@property (nonatomic, strong) NSMutableArray *otherDaysArr;
@property (nonatomic, strong) UIColor *otherDaysColor;
// 是否只显示本月日期,默认->NO
@property (nonatomic, assign) BOOL isShowOnlyMonthDays;
// 默认当前日期的按钮
@property (weak, nonatomic) UIButton *defaultBtn;
//创建日历
- (void)createCalendarViewWith:(NSDate *)date;
/**
* nextMonth
*
* @param date nil = 当前日期的下一个月
*/
//- (void)setNextMonth:(NSDate *)date;
- (NSDate *)nextMonth:(NSDate *)date;
/**
* lastMonth
*
* @param date nil -> 当前日期的上一个月
*/
//- (void)setLastMonth:(NSDate *)date;
- (NSDate *)lastMonth:(NSDate *)date;
/**
* nextMonth and lastMonth
*/
@property (nonatomic, copy) void(^nextMonthBlock)();
@property (nonatomic, copy) void(^lastMonthBlock)();
/**
* 点击返回日期
*/
@property (nonatomic, copy) void(^calendarBlock)(NSString *seletDate, UIButton *selectBtn);
- m 文件
#import "CalendarView.h"
#import "UIColor+Hex.h"
@interface CalendarView ()
@property (nonatomic, strong) UIButton *selectBtn;
@property (nonatomic, strong) NSMutableArray *daysArray;
@end
@implementation CalendarView
- (id)initWithFrame:(CGRect)frame {
self = [super initWithFrame:frame];
if (self) {
[self setupDate];
[self setupNextAndLastMonthView];
}
return self;
}
- (void)setupDate {
self.daysArray = [NSMutableArray arrayWithCapacity:42];
for (int i = 0; i < 42; i++) {
UIButton *button = [[UIButton alloc] init];
[self addSubview:button];
[_daysArray addObject:button];
[button addTarget:self action:@selector(logDate:) forControlEvents:UIControlEventTouchUpInside];
}
}
- (void)setupNextAndLastMonthView {
UIButton *leftBtn = [UIButton buttonWithType:UIButtonTypeCustom];
[leftBtn setImage:[UIImage imageNamed:@"to_left"] forState:UIControlStateNormal];
[leftBtn addTarget:self action:@selector(nextAndLastMonth:) forControlEvents:UIControlEventTouchUpInside];
[self addSubview:leftBtn];
leftBtn.tag = 1;
leftBtn.frame = CGRectMake(0, 0, 40, 40);
UIButton *rightBtn = [UIButton buttonWithType:UIButtonTypeCustom];
[rightBtn setImage:[UIImage imageNamed:@"to_right"] forState:UIControlStateNormal];
rightBtn.tag = 2;
[rightBtn addTarget:self action:@selector(nextAndLastMonth:) forControlEvents:UIControlEventTouchUpInside];
[self addSubview:rightBtn];
rightBtn.frame = CGRectMake(self.frame.size.width - 40, 0, 40, 40);
}
- (void)nextAndLastMonth:(UIButton *)button {
if (button.tag == 1) {
if (self.lastMonthBlock) {
self.lastMonthBlock();
}
} else {
if (self.nextMonthBlock) {
self.nextMonthBlock();
}
}
}
#pragma mark - create View
- (void)setDate:(NSDate *)date{
_date = date;
[self createCalendarViewWith:date];
}
- (void)createCalendarViewWith:(NSDate *)date{
CGFloat itemW = self.frame.size.width / 7;
CGFloat itemH = self.frame.size.width / 7;
// 1.year month
if (self.headlabel == nil) {
self.headlabel = [[UILabel alloc] init];
}
self.headlabel.text = [NSString stringWithFormat:@"%li年%li月",(long)[self year:date],(long)[self month:date]];
NSLog(@"%@", self.headlabel.text);
self.headlabel.font = [UIFont systemFontOfSize:14];
self.headlabel.frame = CGRectMake(0, 0, self.frame.size.width, itemH * 0.8);
self.headlabel.textAlignment = NSTextAlignmentCenter;
self.headlabel.textColor = self.headColor;
[self addSubview: self.headlabel];
// 2.weekday
NSArray *array = @[@"日", @"一", @"二", @"三", @"四", @"五", @"六"];
self.weekBg = [[UIView alloc] init];
// weekBg.backgroundColor = [UIColor orangeColor];
self.weekBg.frame = CGRectMake(0, CGRectGetMaxY(self.headlabel.frame), self.frame.size.width, itemH * 0.6);
[self addSubview:self.weekBg];
UIView *line1 = [[UIView alloc] initWithFrame:CGRectMake(0, 0, PDScreeenW, 1.0)];
line1.backgroundColor = LineCommanColorA;
[self.weekBg addSubview:line1];
UIView *line2 = [[UIView alloc] initWithFrame:CGRectMake(0, self.weekBg.frame.size.height - 1.0, PDScreeenW, 1.0)];
line2.backgroundColor = LineCommanColorA;
[self.weekBg addSubview:line2];
for (int i = 0; i < 7; i++) {
UILabel *week = [[UILabel alloc] init];
week.text = array[i];
week.font = [UIFont systemFontOfSize:14];
week.frame = CGRectMake(itemW * i, (itemH * 0.6 - 20) * 0.5, itemW, 20);
week.textAlignment = NSTextAlignmentCenter;
week.backgroundColor = [UIColor clearColor];
week.textColor = self.weekDaysColor;
[self.weekBg addSubview:week];
}
// 3.days (1-31)
for (int i = 0; i < 42; i++) {
int x = (i % 7) * itemW ;
int y = (i / 7) * itemH + CGRectGetMaxY(self.weekBg.frame) + 5;
UIButton *dayButton = _daysArray[i];
dayButton.selected = NO;
dayButton.frame = CGRectMake(x, y, itemW-6, itemH-6);
dayButton.layer.cornerRadius = dayButton.frame.size.width / 2;
dayButton.layer.masksToBounds = YES;
dayButton.layer.borderWidth = 3.0 ;
dayButton.layer.borderColor = [[UIColor clearColor] CGColor];
dayButton.titleLabel.font = [UIFont systemFontOfSize:14.0];
dayButton.titleLabel.textAlignment = NSTextAlignmentCenter;
[dayButton addTarget:self action:@selector(logDate:) forControlEvents:UIControlEventTouchUpInside];
NSInteger daysInLastMonth = [self totaldaysInMonth:[self lastMonth:date]];
NSInteger daysInThisMonth = [self totaldaysInMonth:date];
NSInteger firstWeekday = [self firstWeekdayInThisMonth:date];
NSInteger day = 0;
if (i < firstWeekday) {
day = daysInLastMonth - firstWeekday + i + 1;
[self setStyle_BeyondThisMonth:dayButton];
}else if (i > firstWeekday + daysInThisMonth - 1){
day = i + 1 - firstWeekday - daysInThisMonth;
[self setStyle_BeyondThisMonth:dayButton];
}else{
day = i - firstWeekday + 1;
[dayButton setTitle:[NSString stringWithFormat:@"%li", (long)day] forState:UIControlStateNormal];
[self setStyle_AfterToday:dayButton];
}
[dayButton setTitle:[NSString stringWithFormat:@"%li", (long)day] forState:UIControlStateNormal];
dayButton.titleLabel.font = [UIFont systemFontOfSize:21];
// this month
if ([self month:date] == [self month:[NSDate date]]) {
NSInteger todayIndex = [self day:date] + firstWeekday - 1;
if (i < todayIndex && i >= firstWeekday) {
// [self setStyle_BeforeToday:dayButton];
}else if(i == todayIndex){
[self setStyle_Today:dayButton];
}
}
}
}
#pragma mark - output date
-(void)logDate:(UIButton *)dayBtn
{
dayBtn.selected = !dayBtn.selected;
if (dayBtn.selected) {
dayBtn.layer.borderColor = [self.dateColor CGColor];
}else{
dayBtn.layer.borderColor = [[UIColor clearColor] CGColor];
}
NSInteger day = [[dayBtn titleForState:UIControlStateNormal] integerValue];
NSString *dayStr;
if (day < 10) {
dayStr = [NSString stringWithFormat:@"0%zd",day];
}else{
dayStr = [NSString stringWithFormat:@"%zd",day];
}
NSDateComponents *comp = [[NSCalendar currentCalendar] components:(NSCalendarUnitYear | NSCalendarUnitMonth | NSCalendarUnitDay) fromDate:self.date];
NSInteger month = [comp month];
NSString *monthStr;
if (month < 10) {
monthStr = [NSString stringWithFormat:@"0%li",month];
}else{
monthStr = [NSString stringWithFormat:@"%li",month];
}
NSString *dateStr = [NSString stringWithFormat:@"%li-%@-%@", [comp year],monthStr, dayStr];
if (self.calendarBlock) {
self.calendarBlock(dateStr, dayBtn);
}
}
#pragma mark - date button style
- (void)setStyle_BeyondThisMonth:(UIButton *)btn
{
btn.enabled = NO;
if (self.isShowOnlyMonthDays) {
[btn setTitleColor:[UIColor lightGrayColor] forState:UIControlStateNormal];
} else {
[btn setTitleColor:[UIColor clearColor] forState:UIControlStateNormal];
}
}
- (void)setStyle_Today:(UIButton *)btn
{
btn.selected = YES;
self.defaultBtn = btn;
[btn setTitleColor:[UIColor lightGrayColor] forState:UIControlStateNormal];
[btn setTitleColor:CommanColor forState:UIControlStateSelected];
btn.layer.borderColor = [self.dateColor CGColor];
}
- (void)setStyle_AfterToday:(UIButton *)btn
{
[btn setTitleColor:CommanColor forState:UIControlStateSelected];
[btn setTitleColor:[UIColor lightGrayColor] forState:UIControlStateNormal];
for (NSString *str in self.allDaysArr) {
NSString *btnTitle = btn.titleLabel.text;
NSInteger btnInt = [btnTitle integerValue];
if (btnInt < 10) btnTitle = [NSString stringWithFormat:@"0%@",btnTitle];
if ([str isEqualToString:btnTitle]) {
btn.backgroundColor = self.allDaysColor;
}
}
for (NSString *str in self.otherDaysArr) {
NSString *btnTitle = btn.titleLabel.text;
NSInteger btnInt = [btnTitle integerValue];
if (btnInt < 10) {
btnTitle = [NSString stringWithFormat:@"0%@",btnTitle];
}
if ([str isEqualToString:btnTitle]) {
btn.backgroundColor = self.otherDaysColor;
}
}
for (NSString *str in self.partDaysArr) {
NSString *btnTitle = btn.titleLabel.text;
NSInteger btnInt = [btnTitle integerValue];
if (btnInt < 10) {
btnTitle = [NSString stringWithFormat:@"0%@",btnTitle];
}
if ([str isEqualToString:btnTitle]) {
btn.backgroundColor = self.partDaysColor;
}
}
}
#pragma mark - Lazy loading
- (NSMutableArray *)allDaysArr {
if (!_allDaysArr) {
_allDaysArr = [NSMutableArray array];
}
return _allDaysArr;
}
- (NSMutableArray *)otherDaysArr{
if (!_otherDaysArr) {
_otherDaysArr = [NSMutableArray array];
}
return _otherDaysArr;
}
- (NSMutableArray *)partDaysArr {
if (!_partDaysArr) {
_partDaysArr = [NSMutableArray array];
}
return _partDaysArr;
}
- (UIColor *)headColor {
if (!_headColor) {
_headColor = [UIColor blackColor];
}
return _headColor;
}
- (UIColor *)dateColor {
if (!_dateColor) {
_dateColor = [UIColor colorWithHexString:@"#7448B4" alpha:0.5];
}
return _dateColor;
}
- (UIColor *)weekDaysColor {
if (!_weekDaysColor) {
_weekDaysColor = [UIColor lightGrayColor];
}
return _weekDaysColor;
}
//一个月第一个周末
- (NSInteger)firstWeekdayInThisMonth:(NSDate *)date{
NSCalendar *calendar = [NSCalendar currentCalendar];
[calendar setFirstWeekday:1];//1.Sun. 2.Mon. 3.Thes. 4.Wed. 5.Thur. 6.Fri. 7.Sat.
NSDateComponents *component = [calendar components:(NSCalendarUnitYear | NSCalendarUnitMonth | NSCalendarUnitDay) fromDate:date];
[component setDay:1];
NSDate *firstDayOfMonthDate = [calendar dateFromComponents:component];
NSUInteger firstWeekDay = [calendar ordinalityOfUnit:NSCalendarUnitWeekday inUnit:NSCalendarUnitWeekOfMonth forDate:firstDayOfMonthDate];
return firstWeekDay - 1;
}
//总天数
- (NSInteger)totaldaysInMonth:(NSDate *)date{
NSRange daysInOfMonth = [[NSCalendar currentCalendar] rangeOfUnit:NSCalendarUnitDay inUnit:NSCalendarUnitMonth forDate:date];
return daysInOfMonth.length;
}
#pragma mark - month +/-
- (NSDate *)lastMonth:(NSDate *)date{
NSDateComponents *dateComponents = [[NSDateComponents alloc] init];
dateComponents.month = -1;
NSDate *newDate = [[NSCalendar currentCalendar] dateByAddingComponents:dateComponents toDate:date options:0];
return newDate;
}
- (NSDate*)nextMonth:(NSDate *)date{
NSDateComponents *dateComponents = [[NSDateComponents alloc] init];
dateComponents.month = +1;
NSDate *newDate = [[NSCalendar currentCalendar] dateByAddingComponents:dateComponents toDate:date options:0];
return newDate;
}
#pragma mark - date get: day-month-year
- (NSInteger)day:(NSDate *)date{
NSDateComponents *components = [[NSCalendar currentCalendar] components:(NSCalendarUnitYear | NSCalendarUnitMonth | NSCalendarUnitDay) fromDate:date];
return [components day];
}
- (NSInteger)month:(NSDate *)date{
NSDateComponents *components = [[NSCalendar currentCalendar] components:(NSCalendarUnitYear | NSCalendarUnitMonth | NSCalendarUnitDay) fromDate:date];
return [components month];
}
- (NSInteger)year:(NSDate *)date{
NSDateComponents *components = [[NSCalendar currentCalendar] components:(NSCalendarUnitYear | NSCalendarUnitMonth | NSCalendarUnitDay) fromDate:date];
return [components year];
}
